From 716f27c4356f7cb335e2f34aa357f1ef57d8c6d8 Mon Sep 17 00:00:00 2001
From: cnderrauber <zengjie9004@gmail.com>
Date: Sat, 1 Oct 2022 09:03:58 +0800
Subject: [PATCH] Close UDPMux when exit (#1064)

* Close UDPMux when exit

* fix test failed
---
 pkg/rtc/config.go          | 2 +-
 pkg/service/roommanager.go | 3 +++
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pkg/rtc/config.go b/pkg/rtc/config.go
index 0d2c97dff..050d54720 100644
--- a/pkg/rtc/config.go
+++ b/pkg/rtc/config.go
@@ -78,7 +78,7 @@ func NewWebRTCConfig(conf *config.Config, externalIP string) (*WebRTCConfig, err
 		rtcConf.PacketBufferSize = 500
 	}
 
-	var udpMux *ice.UDPMuxDefault
+	var udpMux ice.UDPMux
 	var udpMuxConn *net.UDPConn
 	var err error
 	networkTypes := make([]webrtc.NetworkType, 0, 4)
diff --git a/pkg/service/roommanager.go b/pkg/service/roommanager.go
index 754eb3b57..539db782c 100644
--- a/pkg/service/roommanager.go
+++ b/pkg/service/roommanager.go
@@ -196,6 +196,9 @@ func (r *RoomManager) Stop() {
 	}
 
 	if r.rtcConfig != nil {
+		if r.rtcConfig.UDPMux != nil {
+			_ = r.rtcConfig.UDPMux.Close()
+		}
 		if r.rtcConfig.UDPMuxConn != nil {
 			_ = r.rtcConfig.UDPMuxConn.Close()
 		}