mirror of
https://github.com/bluenviron/mediamtx
synced 2025-02-02 20:53:02 +00:00
9e6abc6e9f
* rtmp: improve MsgCommandAMF0 * rtmp: fix MsgSetPeerBandwidth * rtmp: add message tests * rtmp: replace implementation with new one * rtmp: rename handshake functions * rtmp: avoid calling useless function * rtmp: use time.Duration for PTSDelta * rtmp: fix decoding chunks with relevant size * rtmp: rewrite implementation of rtmp connection * rtmp: fix tests * rtmp: improve error message * rtmp: replace h264 config implementation * link against github.com/notedit/rtmp * normalize MessageStreamID * rtmp: make acknowledge optional * rtmp: fix decoding of chunk2 + chunk3 * avoid using encoding/binary
47 lines
916 B
Go
47 lines
916 B
Go
package message
|
|
|
|
import (
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/bytecounter"
|
|
"github.com/aler9/rtsp-simple-server/internal/rtmp/rawmessage"
|
|
)
|
|
|
|
// Writer is a message writer.
|
|
type Writer struct {
|
|
w *rawmessage.Writer
|
|
}
|
|
|
|
// NewWriter allocates a Writer.
|
|
func NewWriter(w *bytecounter.Writer, checkAcknowledge bool) *Writer {
|
|
return &Writer{
|
|
w: rawmessage.NewWriter(w, checkAcknowledge),
|
|
}
|
|
}
|
|
|
|
// SetAcknowledgeValue sets the value of the last received acknowledge.
|
|
func (w *Writer) SetAcknowledgeValue(v uint32) {
|
|
w.w.SetAcknowledgeValue(v)
|
|
}
|
|
|
|
// Write writes a message.
|
|
func (w *Writer) Write(msg Message) error {
|
|
raw, err := msg.Marshal()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
err = w.w.Write(raw)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
switch tmsg := msg.(type) {
|
|
case *MsgSetChunkSize:
|
|
w.w.SetChunkSize(tmsg.Value)
|
|
|
|
case *MsgSetWindowAckSize:
|
|
w.w.SetWindowAckSize(tmsg.Value)
|
|
}
|
|
|
|
return nil
|
|
}
|