gilles/go2rtc
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix stream info for publishing RTMP

Browse Source
This commit is contained in:
Alex X
2024-06-19 06:52:49 +03:00
parent a4885c2c3a
commit eaae7aee39
2 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
internal/rtmp/rtmp.go
+1 -1
View File
@@ -133,7 +133,7 @@ func streamsHandle(url string) (core.Producer, error) {
func streamsConsumerHandle(url string) (core.Consumer, func(), error) { func streamsConsumerHandle(url string) (core.Consumer, func(), error) {
cons := flv.NewConsumer() cons := flv.NewConsumer()
run := func() { run := func() {
wr, err := rtmp.DialPublish(url) wr, err := rtmp.DialPublish(url, cons)
if err != nil { if err != nil {
return return
} }
pkg/rtmp/client.go
+6 -1
View File
@@ -35,7 +35,7 @@ func DialPlay(rawURL string) (*flv.Producer, error) {
return client.Producer() return client.Producer()
} }
func DialPublish(rawURL string) (io.Writer, error) { func DialPublish(rawURL string, cons *flv.Consumer) (io.Writer, error) {
u, err := url.Parse(rawURL) u, err := url.Parse(rawURL)
if err != nil { if err != nil {
return nil, err return nil, err
@@ -55,6 +55,11 @@ func DialPublish(rawURL string) (io.Writer, error) {
return nil, err return nil, err
} }
cons.FormatName = "rtmp"
cons.Protocol = "rtmp"
cons.RemoteAddr = conn.RemoteAddr().String()
cons.URL = rawURL
return client, nil return client, nil
} }