Added events decoder helpers.

[src/xds/xds-agent.git] / lib / agent / webserver.go

diff --git a/lib/agent/webserver.go b/lib/agent/webserver.go
index b75abd4..13572b4 100644 (file)
--- a/lib/agent/webserver.go
+++ b/lib/agent/webserver.go
@@ -233,8 +233,8 @@ func (s *WebServer) socketHandler(c *gin.Context) {
 
                so.On("disconnection", func() {
                        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)
+                       s.sessions.UpdateIOSocket(sess.ID, nil)
                })
        })