fix session is missing if token invalid

Author: gy2006

core/src/main/java/com/flowci/core/agent/manager/AgentEventManager.java

@@ -120,6 +120,15 @@ public void afterConnectionClosed(WebSocketSession session, CloseStatus status)
         agentSessionStore.remove(token, session);
     }
 
+    private <T> void writeMessage(WebSocketSession session, ResponseMessage<T> msg) {
+        try {
+            byte[] bytes = objectMapper.writeValueAsBytes(msg);
+            session.sendMessage(new BinaryMessage(bytes));
+        } catch (IOException e) {
+            log.warn(e);
+        }
+    }
+
     private void onConnected(WebSocketSession session, String token, byte[] body) {
         try {
             AgentInit init = objectMapper.readValue(body, AgentInit.class);
@@ -134,7 +143,7 @@ private void onConnected(WebSocketSession session, String token, byte[] body) {
             log.debug("Agent {} is connected with status {}", token, init.getStatus());
         } catch (Exception e) {
             log.warn(e);
-            writeMessage(token, new ResponseMessage<Void>(StatusCode.FATAL, e.getMessage(), null));
+            writeMessage(session, new ResponseMessage<Void>(StatusCode.FATAL, e.getMessage(), null));
         }
     }
 

core/src/main/java/com/flowci/core/agent/service/AgentServiceImpl.java

@@ -354,10 +354,13 @@ public void onConnected(OnConnectedEvent event) {
 
     @EventListener
     public void onDisconnected(OnDisconnectedEvent event) {
-        Agent target = getByToken(event.getToken());
+        try {
+            Agent target = getByToken(event.getToken());
+            update(target, OFFLINE);
+            syncLockNode(target, false);
+        } catch (NotFoundException ignore) {
 
-        update(target, OFFLINE);
-        syncLockNode(target, false);
+        }
     }
 
     @EventListener