apps/plugin-rtmp
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-rtmp.git synced 2025-10-05 23:47:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

适配引擎变更

Browse Source
This commit is contained in:
dexter
2022-02-07 17:29:36 +08:00
parent c1fedbc529
commit 3c227c4acb
2 changed files with 26 additions and 31 deletions
Download Patch File Download Diff File Expand all files Collapse all files

39
main.go
View File

@@ -10,43 +10,40 @@ import (
)
type RTMPConfig struct {
Publish PublishConfig
Subscribe SubscribeConfig
ListenAddr string
ChunkSize int
Publish PublishConfig
Subscribe SubscribeConfig
TCPConfig
ChunkSize int
context.Context
cancel context.CancelFunc
}
var config = &RTMPConfig{
Publish: DefaultPublishConfig,
Subscribe: DefaultSubscribeConfig,
ChunkSize: 4096,
ListenAddr: ":1935",
Publish: DefaultPublishConfig,
Subscribe: DefaultSubscribeConfig,
ChunkSize: 4096,
TCPConfig: TCPConfig{ListenAddr: ":1935"},
}
func (cfg *RTMPConfig) Update(override map[string]any) {
if cfg.cancel == nil || (override != nil && override["ListenAddr"] != nil) {
start()
}
}
func init() {
InstallPlugin(config)
}
func start() {
func (cfg *RTMPConfig) Update(override Config) {
override.Unmarshal(cfg)
if config.cancel == nil {
util.Print(Green("server rtmp start at"), BrightBlue(config.ListenAddr))
} else {
} else if override.Has("ListenAddr") {
config.cancel()
util.Print(Green("server rtmp restart at"), BrightBlue(config.ListenAddr))
} else {
return
}
config.Context, config.cancel = context.WithCancel(Ctx)
err := util.ListenTCP(config.ListenAddr, config)
err := cfg.Listen(cfg)
if err == context.Canceled {
log.Println(err)
} else {
log.Fatal(err)
}
}
func init() {
InstallPlugin(config)
}

18
netStream.go
View File

@@ -14,7 +14,7 @@ import (
var gstreamid = uint32(64)
func (cfg *RTMPConfig) Process(conn *net.TCPConn) {
func (cfg *RTMPConfig) ServeTCP(conn *net.TCPConn) {
nc := NetConnection{
TCPConn: conn,
Reader: bufio.NewReader(conn),
@@ -132,28 +132,26 @@ func (cfg *RTMPConfig) Process(conn *net.TCPConn) {
err = nc.SendStreamID(RTMP_USER_STREAM_BEGIN)
err = nc.SendCommand(SEND_PLAY_RESPONSE_MESSAGE, newPlayResponseMessageData(nc.streamID, NetStream_Play_Reset, Level_Status))
err = nc.SendCommand(SEND_PLAY_RESPONSE_MESSAGE, newPlayResponseMessageData(nc.streamID, NetStream_Play_Start, Level_Status))
vt, at := subscriber.WaitVideoTrack("h264", "h265"), subscriber.WaitAudioTrack("aac", "pcma", "pcmu")
vt, at := subscriber.WaitVideoTrack(), subscriber.WaitAudioTrack()
if vt != nil {
frame := vt.DecoderConfiguration
err = nc.sendAVMessage(0, frame.AVCC, false, true)
subscriber.OnVideo = func(frame *engine.VideoFrame) bool {
err = nc.sendAVMessage(frame.DeltaTime, frame.AVCC, false, false)
return err == nil
subscriber.OnVideo = func(frame *engine.VideoFrame) error {
return nc.sendAVMessage(frame.DeltaTime, frame.AVCC, false, false)
}
}
if at != nil {
subscriber.OnAudio = func(frame *engine.AudioFrame) bool {
subscriber.OnAudio = func(frame *engine.AudioFrame) (err error) {
if at.CodecID == codec.CodecID_AAC {
frame := at.DecoderConfiguration
err = nc.sendAVMessage(0, frame.AVCC, true, true)
} else {
err = nc.sendAVMessage(0, frame.AVCC, true, true)
}
subscriber.OnAudio = func(frame *engine.AudioFrame) bool {
err = nc.sendAVMessage(frame.DeltaTime, frame.AVCC, true, false)
return err == nil
subscriber.OnAudio = func(frame *engine.AudioFrame) error {
return nc.sendAVMessage(frame.DeltaTime, frame.AVCC, true, false)
}
return err == nil
return
}
}
go subscriber.Play(at, vt)