Code refactoring
This commit is contained in:
@@ -12,6 +12,7 @@ import (
|
|||||||
"github.com/AlexxIT/go2rtc/pkg/core"
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
||||||
"github.com/AlexxIT/go2rtc/pkg/h264"
|
"github.com/AlexxIT/go2rtc/pkg/h264"
|
||||||
"github.com/AlexxIT/go2rtc/pkg/h264/annexb"
|
"github.com/AlexxIT/go2rtc/pkg/h264/annexb"
|
||||||
|
"github.com/AlexxIT/go2rtc/pkg/multipart"
|
||||||
"github.com/AlexxIT/go2rtc/pkg/tcp"
|
"github.com/AlexxIT/go2rtc/pkg/tcp"
|
||||||
"github.com/pion/rtp"
|
"github.com/pion/rtp"
|
||||||
)
|
)
|
||||||
@@ -69,7 +70,7 @@ func (c *Producer) Start() error {
|
|||||||
}
|
}
|
||||||
|
|
||||||
for {
|
for {
|
||||||
header, body, err := tcp.NextMultipart(c.reader)
|
header, body, err := multipart.Next(c.reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
@@ -130,7 +131,7 @@ func (c *Producer) probe() error {
|
|||||||
timeout := time.Now().Add(core.ProbeTimeout)
|
timeout := time.Now().Add(core.ProbeTimeout)
|
||||||
|
|
||||||
for (waitVideo || waitAudio) && time.Now().Before(timeout) {
|
for (waitVideo || waitAudio) && time.Now().Before(timeout) {
|
||||||
header, body, err := tcp.NextMultipart(c.reader)
|
header, body, err := multipart.Next(c.reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -1,4 +1,4 @@
|
|||||||
package tcp
|
package multipart
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
"bufio"
|
||||||
@@ -10,7 +10,7 @@ import (
|
|||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
func NextMultipart(rd *bufio.Reader) (http.Header, []byte, error) {
|
func Next(rd *bufio.Reader) (http.Header, []byte, error) {
|
||||||
for {
|
for {
|
||||||
// search next boundary and skip empty lines
|
// search next boundary and skip empty lines
|
||||||
s, err := rd.ReadString('\n')
|
s, err := rd.ReadString('\n')
|
||||||
@@ -6,7 +6,6 @@ import (
|
|||||||
"io"
|
"io"
|
||||||
|
|
||||||
"github.com/AlexxIT/go2rtc/pkg/core"
|
"github.com/AlexxIT/go2rtc/pkg/core"
|
||||||
"github.com/AlexxIT/go2rtc/pkg/tcp"
|
|
||||||
"github.com/pion/rtp"
|
"github.com/pion/rtp"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -46,7 +45,7 @@ func (c *Producer) Start() error {
|
|||||||
mjpeg := c.Receivers[0]
|
mjpeg := c.Receivers[0]
|
||||||
|
|
||||||
for {
|
for {
|
||||||
_, body, err := tcp.NextMultipart(c.reader)
|
_, body, err := Next(c.reader)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return err
|
return err
|
||||||
}
|
}
|
||||||
|
|||||||
Reference in New Issue
Block a user