mirror of
https://github.com/langhuihui/monibuca.git
synced 2025-12-24 13:48:04 +08:00
8a9fffb987
- Refactor frame converter implementation - Update mp4 track to use ICodex - General refactoring and code improvements 🤖 Generated with [Claude Code](https://claude.ai/code) Co-Authored-By: Claude <noreply@anthropic.com>
75 lines
1.8 KiB
Go
75 lines
1.8 KiB
Go
package rtmp
|
|
|
|
import (
|
|
"errors"
|
|
"runtime"
|
|
|
|
"m7s.live/v5/pkg"
|
|
|
|
"m7s.live/v5"
|
|
)
|
|
|
|
type Sender struct {
|
|
*NetConnection
|
|
ChunkHeader
|
|
errContinue bool
|
|
lastAbs uint32
|
|
}
|
|
|
|
func (av *Sender) HandleAudio(frame *AudioFrame) (err error) {
|
|
return av.SendFrame((*RTMPData)(frame))
|
|
}
|
|
|
|
func (av *Sender) HandleVideo(frame *VideoFrame) (err error) {
|
|
return av.SendFrame((*RTMPData)(frame))
|
|
}
|
|
|
|
func (av *Sender) SendFrame(frame *RTMPData) (err error) {
|
|
// seq := frame.Sequence
|
|
payloadLen := frame.Size
|
|
if av.errContinue {
|
|
defer func() {
|
|
if err != nil {
|
|
err = pkg.ErrInterrupt
|
|
}
|
|
}()
|
|
}
|
|
if payloadLen == 0 {
|
|
err = errors.New("payload is empty")
|
|
// av.Error("payload is empty", zap.Error(err))
|
|
return err
|
|
}
|
|
if av.writeSeqNum > av.bandwidth {
|
|
av.totalWrite += av.writeSeqNum
|
|
av.writeSeqNum = 0
|
|
av.SendMessage(RTMP_MSG_ACK, Uint32Message(av.totalWrite))
|
|
av.SendStreamID(RTMP_USER_PING_REQUEST, 0)
|
|
}
|
|
av.MessageLength = uint32(payloadLen)
|
|
for !av.writing.CompareAndSwap(false, true) {
|
|
runtime.Gosched()
|
|
}
|
|
defer av.writing.Store(false)
|
|
// 第一次是发送关键帧,需要完整的消息头(Chunk Basic Header(1) + Chunk Message Header(11) + Extended Timestamp(4)(可能会要包括))
|
|
// 后面开始,就是直接发送音视频数据,那么直接发送,不需要完整的块(Chunk Basic Header(1) + Chunk Message Header(7))
|
|
// 当Chunk Type为0时(即Chunk12),
|
|
if av.lastAbs == 0 {
|
|
av.SetTimestamp(1)
|
|
err = av.sendChunk(frame.Memory, &av.ChunkHeader, RTMP_CHUNK_HEAD_12)
|
|
} else {
|
|
av.SetTimestamp(frame.GetTS32() - av.lastAbs)
|
|
err = av.sendChunk(frame.Memory, &av.ChunkHeader, RTMP_CHUNK_HEAD_8)
|
|
}
|
|
av.lastAbs = frame.GetTS32()
|
|
// //数据被覆盖导致序号变了
|
|
// if seq != frame.Sequence {
|
|
// return errors.New("sequence is not equal")
|
|
// }
|
|
return
|
|
}
|
|
|
|
type Receiver struct {
|
|
*m7s.Publisher
|
|
NetStream
|
|
}
|