浏览代码

add debug log

abhishek9686 1 年之前
父节点
当前提交
052277ade4
共有 1 个文件被更改,包括 3 次插入11 次删除
  1. 3 11
      auth/host_session.go

+ 3 - 11
auth/host_session.go

@@ -131,15 +131,11 @@ func SessionHandler(conn *websocket.Conn) {
 	}
 
 	go func() {
+		defer fmt.Println("#######-----------> Exiting CONN CLOSE watcher")
 		for {
 			msgType, _, err := conn.ReadMessage()
-			if err != nil {
-				return
-			}
-			if msgType == websocket.CloseMessage {
-				if timeout != nil {
-					timeout <- true
-				}
+			if err != nil || msgType == websocket.CloseMessage {
+				netcache.Del(stateStr)
 				return
 			}
 		}
@@ -240,10 +236,6 @@ func SessionHandler(conn *websocket.Conn) {
 		logger.Log(0, "timeout signal recv,exiting oauth socket conn")
 		break
 	}
-	// The entry is not needed anymore, but we will let the producer to close it to avoid panic cases
-	if err = netcache.Del(stateStr); err != nil {
-		logger.Log(0, "failed to remove node SSO cache entry", err.Error())
-	}
 	// Cleanly close the connection by sending a close message and then
 	// waiting (with timeout) for the server to close the connection.
 	if err = conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")); err != nil {