X-Git-Url: https://gerrit.automotivelinux.org/gerrit/gitweb?a=blobdiff_plain;f=lib%2Fagent%2Fwebserver.go;h=b75abd43d050f05d45a5e04ead08117c08688827;hb=0a5ab2ca1ab89e6e667b7f9c7bdffdff036ca9b7;hp=4b2e024ac40a2fcda0936acc69118df14acc6609;hpb=2d90eac319979dba64371258b30e61e77a15db7d;p=src%2Fxds%2Fxds-agent.git

diff --git a/lib/agent/webserver.go b/lib/agent/webserver.go
index 4b2e024..b75abd4 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,12 +228,13 @@ 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.sessions.UpdateIOSocket(sess.ID, nil)
+			s.events.UnRegister(apiv1.EVTAll, sess.ID)
 		})
 	})