mirror of
https://github.com/lkmio/lkm.git
synced 2025-11-03 11:31:00 +08:00
34 lines
781 B
Go
34 lines
781 B
Go
package rtmp
|
|
|
|
import (
|
|
"github.com/yangjiechina/avformat/librtmp"
|
|
"github.com/yangjiechina/avformat/utils"
|
|
"github.com/yangjiechina/lkm/stream"
|
|
"net"
|
|
)
|
|
|
|
type Sink struct {
|
|
stream.BaseSink
|
|
stack *librtmp.Stack
|
|
}
|
|
|
|
func (s *Sink) Start() {
|
|
_ = s.stack.SendStreamBeginChunk(s.Conn)
|
|
}
|
|
|
|
func (s *Sink) Flush() {
|
|
_ = s.stack.SendStreamEOFChunk(s.Conn)
|
|
}
|
|
|
|
func (s *Sink) Close() {
|
|
s.stack = nil
|
|
s.BaseSink.Close()
|
|
}
|
|
|
|
func NewSink(id stream.SinkId, sourceId string, conn net.Conn, stack *librtmp.Stack) stream.Sink {
|
|
return &Sink{
|
|
BaseSink: stream.BaseSink{Id_: id, SourceId_: sourceId, State_: stream.SessionStateCreate, Protocol_: stream.ProtocolRtmp, Conn: conn, DesiredAudioCodecId_: utils.AVCodecIdNONE, DesiredVideoCodecId_: utils.AVCodecIdNONE},
|
|
stack: stack,
|
|
}
|
|
}
|