SFU id and parse media info before remove request.

Former-commit-id: e8e744c
OrlandoCo · May 14, 2020 · ff9d6c7 · ff9d6c7
1 parent 710bcbe
commit ff9d6c7
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/cmd/sfu/main.go b/cmd/sfu/main.go
@@ -80,6 +80,6 @@ func main() {
 
 	rpcID := serviceNode.GetRPCChannel()
 	eventID := serviceNode.GetEventChannel()
-	sfu.Init(conf.Global.Dc, serviceNode.NodeInfo().ID, rpcID, eventID, conf.Nats.URL)
+	sfu.Init(conf.Global.Dc, serviceNode.NodeInfo().Info["id"], rpcID, eventID, conf.Nats.URL)
 	select {}
 }
diff --git a/pkg/node/sfu/init.go b/pkg/node/sfu/init.go
@@ -29,8 +29,13 @@ func Init(dcID, nodeID, rpcID, eventID, natsURL string) {
 func checkRTC() {
 	log.Infof("SFU.checkRTC start")
 	go func() {
-		for mid := range rtc.CleanChannel {
-			broadcaster.Say(proto.SFUStreamRemove, util.Map("mid", mid))
+		for mInfo := range rtc.CleanChannel {
+			mediaInfo, err := proto.ParseMediaInfo(mInfo)
+			if err != nil {
+				log.Errorf("Error parsing media info %v", mInfo)
+				continue
+			}
+			broadcaster.Say(proto.SFUStreamRemove, util.Map("mid", mediaInfo.MID))
 		}
 	}()
 }