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

diff --git a/lib/agent/webserver.go b/lib/agent/webserver.go
index b75abd4..2f24c6c 100644
--- a/lib/agent/webserver.go
+++ b/lib/agent/webserver.go
@@ -1,3 +1,20 @@
+/*
+ * Copyright (C) 2017-2018 "IoT.bzh"
+ * Author Sebastien Douheret <sebastien@iot.bzh>
+ *
+ * Licensed under the Apache License, Version 2.0 (the "License");
+ * you may not use this file except in compliance with the License.
+ * You may obtain a copy of the License at
+ *
+ *   http://www.apache.org/licenses/LICENSE-2.0
+ *
+ * Unless required by applicable law or agreed to in writing, software
+ * distributed under the License is distributed on an "AS IS" BASIS,
+ * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+ * See the License for the specific language governing permissions and
+ * limitations under the License.
+ */
+
 package agent
 
 import (
@@ -7,11 +24,12 @@ import (
 	"os"
 	"path"
 
+	socketio "github.com/googollee/go-socket.io"
+
+	"gerrit.automotivelinux.org/gerrit/src/xds/xds-agent/lib/xaapiv1"
 	"github.com/Sirupsen/logrus"
 	"github.com/gin-contrib/static"
 	"github.com/gin-gonic/gin"
-	"github.com/googollee/go-socket.io"
-	"github.com/iotbzh/xds-agent/lib/apiv1"
 )
 
 // WebServer .
@@ -76,7 +94,7 @@ func (s *WebServer) Serve() error {
 		for _, svrCfg := range s.Config.FileConf.ServersConf {
 			if svr, err := s.api.AddXdsServer(svrCfg); err != nil {
 				// Just log error, don't consider as critical
-				s.Log.Infof("Cannot connect to XDS Server url=%s: %v", svr.BaseURL, err.Error())
+				s.Log.Errorf("Cannot connect to XDS Server url=%s: %v", svr.BaseURL, err.Error())
 			}
 		}
 	}()
@@ -233,8 +251,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.events.UnRegister(xaapiv1.EVTAll, sess.ID)
 			s.sessions.UpdateIOSocket(sess.ID, nil)
-			s.events.UnRegister(apiv1.EVTAll, sess.ID)
 		})
 	})