Imp: new udp session

d6fb3362 · AlexStocks · 5d412f1b · d6fb3362
Commit d6fb3362 authored Apr 24, 2020 by AlexStocks
Hide 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) {
-	var (
+	s.wg.Add(1)
-		ss Session
+	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)
-		panic(err.Error())
+		if err = newSession(ss); err != nil {
-	}
+			conn.Close()
-	ss.(*session).run()
+			panic(err.Error())
+		}
+		ss.(*session).run()
+	}()
 }
 type wsHandler struct {