修改推流适配引擎

This commit is contained in:
dexter
2022-05-09 22:04:11 +08:00
parent 48c58bab02
commit 2dbfcfe19f

View File

@@ -2,6 +2,7 @@ package rtmp
import ( import (
"bufio" "bufio"
"errors"
"net" "net"
"net/url" "net/url"
"strings" "strings"
@@ -85,13 +86,13 @@ func (pusher *RTMPPusher) Connect() (err error) {
return return
} }
func (pusher *RTMPPusher) Push() { func (pusher *RTMPPusher) Push() error {
pusher.SendMessage(RTMP_MSG_AMF0_COMMAND, &CommandMessage{"createStream", 2}) pusher.SendMessage(RTMP_MSG_AMF0_COMMAND, &CommandMessage{"createStream", 2})
defer pusher.Stop() defer pusher.Stop()
for { for {
msg, err := pusher.RecvMessage() msg, err := pusher.RecvMessage()
if err != nil { if err != nil {
break return err
} }
switch msg.MessageTypeID { switch msg.MessageTypeID {
case RTMP_MSG_AMF0_COMMAND: case RTMP_MSG_AMF0_COMMAND:
@@ -118,7 +119,7 @@ func (pusher *RTMPPusher) Push() {
if response.Infomation["code"] == NetStream_Publish_Start { if response.Infomation["code"] == NetStream_Publish_Start {
go pusher.PlayBlock() go pusher.PlayBlock()
} else { } else {
return return errors.New(response.Infomation["code"].(string))
} }
} }
} }