Imp: new udp session

b4bd367f · AlexStocks · watermelo · 4d0b3662 · b4bd367f
Commit b4bd367f authored Apr 24, 2020 by AlexStocks Committed by watermelo Jul 18, 2020
Show whitespace changes
Inline Side-by-side

Showing with 16 additions and 8 deletions

server.go server.go +16 -8

No files found.
--- a/server.go
+++ b/server.go
@@ -280,15 +280,23 @@ func (s *server) runTcpEventLoop(newSession NewSessionCallback) {
 }
 func (s *server) runUDPEventLoop(newSession NewSessionCallback) {
+	s.wg.Add(1)
+	go func() {
+		defer s.wg.Done()
 		var (
+			err  error
+			conn *net.UDPConn
 			ss   Session
 		)
-	ss = newUDPSession(s.pktListener.(*net.UDPConn), s)
+		conn = s.pktListener.(*net.UDPConn)
-	if err := newSession(ss); err != nil {
+		ss = newUDPSession(conn, s)
+		if err = newSession(ss); err != nil {
+			conn.Close()
 			panic(err.Error())
 		}
 		ss.(*session).run()
+	}()
 }
 type wsHandler struct {