X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=lib%2Fagent%2Fwebserver.go;h=13572b41c629a937f2fdb418644cf748c0d7d750;hb=c51d5034d527578da70bdd41b9ce13f28455c598;hp=4b2e024ac40a2fcda0936acc69118df14acc6609;hpb=4695555e178bcabe54c5bf82117c9c4cef5440b5;p=src%2Fxds%2Fxds-agent.git

diff --git a/lib/agent/webserver.go b/lib/agent/webserver.go
index 4b2e024..13572b4 100644
--- a/lib/agent/webserver.go
+++ b/lib/agent/webserver.go
@@ -11,6 +11,7 @@ import (
 	"github.com/gin-contrib/static"
 	"github.com/gin-gonic/gin"
 	"github.com/googollee/go-socket.io"
+	"github.com/iotbzh/xds-agent/lib/apiv1"
 )
 
 // WebServer .
@@ -227,11 +228,12 @@ func (s *WebServer) socketHandler(c *gin.Context) {
 	}
 
 	s.sIOServer.On("connection", func(so socketio.Socket) {
-		s.Log.Debugf("WS Connected (SID=%v)", so.Id())
+		s.Log.Debugf("WS Connected (WSID=%s, SID=%s)", so.Id(), sess.ID)
 		s.sessions.UpdateIOSocket(sess.ID, &so)
 
 		so.On("disconnection", func() {
-			s.Log.Debugf("WS disconnected (SID=%v)", so.Id())
+			s.Log.Debugf("WS disconnected (WSID=%s, SID=%s)", so.Id(), sess.ID)
+			s.events.UnRegister(apiv1.EVTAll, sess.ID)
 			s.sessions.UpdateIOSocket(sess.ID, nil)
 		})
 	})