mirror of
https://github.com/lkmio/lkm.git
synced 2025-10-04 14:52:44 +08:00
104 lines
2.5 KiB
Go
104 lines
2.5 KiB
Go
package rtc
|
|
|
|
import (
|
|
"github.com/pion/webrtc/v3"
|
|
"github.com/yangjiechina/avformat/utils"
|
|
"github.com/yangjiechina/live-server/stream"
|
|
)
|
|
|
|
type transStream struct {
|
|
stream.TransStreamImpl
|
|
}
|
|
|
|
func NewTransStream() stream.ITransStream {
|
|
t := &transStream{}
|
|
t.Init()
|
|
return t
|
|
}
|
|
|
|
func (t *transStream) Input(packet utils.AVPacket) error {
|
|
if utils.AVMediaTypeAudio == packet.MediaType() {
|
|
|
|
} else if utils.AVMediaTypeVideo == packet.MediaType() {
|
|
|
|
for _, iSink := range t.Sinks {
|
|
sink_ := iSink.(*sink)
|
|
if sink_.state < webrtc.ICEConnectionStateConnected {
|
|
continue
|
|
}
|
|
|
|
if packet.KeyFrame() {
|
|
extra, err := t.TransStreamImpl.Tracks[packet.Index()].AnnexBExtraData()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
sink_.input(packet.Index(), extra, 0)
|
|
}
|
|
sink_.input(packet.Index(), packet.AnnexBPacketData(), 40)
|
|
}
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func (t *transStream) AddSink(sink_ stream.ISink) error {
|
|
//创建PeerConnection
|
|
var videoTrack *webrtc.TrackLocalStaticSample
|
|
rtcSink := sink_.(*sink)
|
|
rtcSink.setTrackCount(len(t.Tracks))
|
|
connection, err := webrtc.NewPeerConnection(webrtc.Configuration{})
|
|
|
|
connection.OnICECandidate(func(candidate *webrtc.ICECandidate) {
|
|
|
|
})
|
|
|
|
for index, track := range t.Tracks {
|
|
if utils.AVCodecIdH264 != track.CodecId() {
|
|
continue
|
|
}
|
|
|
|
videoTrack, err = webrtc.NewTrackLocalStaticSample(webrtc.RTPCodecCapability{MimeType: webrtc.MimeTypeH264}, "video", "pion")
|
|
if err != nil {
|
|
panic(err)
|
|
}
|
|
|
|
if _, err := connection.AddTransceiverFromTrack(videoTrack, webrtc.RTPTransceiverInit{Direction: webrtc.RTPTransceiverDirectionSendonly}); err != nil {
|
|
return err
|
|
}
|
|
|
|
if _, err = connection.AddTrack(videoTrack); err != nil {
|
|
return err
|
|
}
|
|
|
|
rtcSink.addTrack(index, videoTrack)
|
|
}
|
|
|
|
if err = connection.SetRemoteDescription(webrtc.SessionDescription{Type: webrtc.SDPTypeOffer, SDP: rtcSink.offer}); err != nil {
|
|
return err
|
|
}
|
|
|
|
complete := webrtc.GatheringCompletePromise(connection)
|
|
answer, err := connection.CreateAnswer(nil)
|
|
if err != nil {
|
|
return err
|
|
} else if err = connection.SetLocalDescription(answer); err != nil {
|
|
return err
|
|
}
|
|
<-complete
|
|
|
|
connection.OnICEConnectionStateChange(func(state webrtc.ICEConnectionState) {
|
|
rtcSink.state = state
|
|
if webrtc.ICEConnectionStateDisconnected > state {
|
|
rtcSink.Close()
|
|
}
|
|
})
|
|
|
|
rtcSink.peer = connection
|
|
rtcSink.SendHeader([]byte(connection.LocalDescription().SDP))
|
|
return t.TransStreamImpl.AddSink(sink_)
|
|
}
|
|
|
|
func (t *transStream) WriteHeader() error {
|
|
return nil
|
|
}
|