apps/plugin-webrtc
1
0
Fork 0
mirror of https://github.com/Monibuca/plugin-webrtc.git synced 2025-10-04 22:42:44 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

🐛 FIX: 批量订阅

Browse Source
This commit is contained in:
dexter
2022-11-30 13:17:57 +08:00
parent 244985d683
commit 190a0161ba
2 changed files with 23 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

19
batcher.go
View File

@@ -53,6 +53,7 @@ func (suber *WebRTCBatcher) Start() (err error) {
func (suber *WebRTCBatcher) Signal(msg DataChannelMessage) {
var s Signal
var removeMap = map[string]string{"type": "remove", "streamPath": ""}
// var offer SessionDescription
if err := json.Unmarshal(msg.Data, &s); err != nil {
WebRTCPlugin.Error("Signal", zap.Error(err))
@@ -66,13 +67,27 @@ func (suber *WebRTCBatcher) Signal(msg DataChannelMessage) {
for _, streamPath := range s.StreamList {
sub := &WebRTCBatchSubscriber{}
sub.WebRTCIO = suber.WebRTCIO
if err = WebRTCPlugin.Subscribe(streamPath, sub); err == nil {
if err = WebRTCPlugin.SubscribeExist(streamPath, sub); err == nil {
suber.subscribers = append(suber.subscribers, sub)
go func() {
sub.PlayRTP()
b, _ := json.Marshal(map[string]string{"type": "remove", "streamPath": streamPath})
if sub.audioSender != nil {
suber.RemoveTrack(sub.audioSender)
}
if sub.videoSender != nil {
suber.RemoveTrack(sub.videoSender)
}
if sub.DC != nil {
sub.DC.Close()
}
removeMap["streamPath"] = streamPath
b, _ := json.Marshal(removeMap)
suber.signalChannel.SendText(string(b))
}()
} else {
removeMap["streamPath"] = streamPath
b, _ := json.Marshal(removeMap)
suber.signalChannel.SendText(string(b))
}
}
var answer string

8
subscriber.go
View File

@@ -18,6 +18,8 @@ type WebRTCSubscriber struct {
WebRTCIO
videoTrack *TrackLocalStaticRTP
audioTrack *TrackLocalStaticRTP
videoSender *RTPSender
audioSender *RTPSender
DC *DataChannel
flvHeadCache []byte
}
@@ -44,11 +46,11 @@ func (suber *WebRTCSubscriber) OnEvent(event any) {
if suber.videoTrack == nil {
suber.DC, _ = suber.PeerConnection.CreateDataChannel(suber.Subscriber.Stream.Path, nil)
} else {
rtpSender, _ := suber.PeerConnection.AddTrack(suber.videoTrack)
suber.videoSender, _ = suber.PeerConnection.AddTrack(suber.videoTrack)
go func() {
rtcpBuf := make([]byte, 1500)
for {
if n, _, rtcpErr := rtpSender.Read(rtcpBuf); rtcpErr != nil {
if n, _, rtcpErr := suber.videoSender.Read(rtcpBuf); rtcpErr != nil {
return
} else {
@@ -72,7 +74,7 @@ func (suber *WebRTCSubscriber) OnEvent(event any) {
}
if v.CodecID == codec.CodecID_PCMA || v.CodecID == codec.CodecID_PCMU {
suber.audioTrack, _ = NewTrackLocalStaticRTP(RTPCodecCapability{MimeType: audioMimeType}, v.Name, suber.Subscriber.Stream.Path)
suber.PeerConnection.AddTrack(suber.audioTrack)
suber.audioSender, _ = suber.PeerConnection.AddTrack(suber.audioTrack)
suber.Subscriber.AddTrack(v) //接受这个track
}
case VideoDeConf: