From 26daf922613ee530e68e655a9bbf6d9afead0331 Mon Sep 17 00:00:00 2001
From: qiminbao <qaz001122>
Date: Mon, 29 Sep 2025 09:36:35 +0800
Subject: [PATCH] =?UTF-8?q?=E6=96=AD=E5=BC=80=E8=BF=9E=E6=8E=A5=E5=8F=96?=
 =?UTF-8?q?=E6=B6=88=E5=AE=9A=E6=97=B6=E4=BB=BB=E5=8A=A1?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .../agent/client/netty/handler/TCPListenHandler.java        | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/src/main/java/com/tongran/agent/client/netty/handler/TCPListenHandler.java b/src/main/java/com/tongran/agent/client/netty/handler/TCPListenHandler.java
index d364383..fd4d386 100644
--- a/src/main/java/com/tongran/agent/client/netty/handler/TCPListenHandler.java
+++ b/src/main/java/com/tongran/agent/client/netty/handler/TCPListenHandler.java
@@ -3,6 +3,7 @@ package com.tongran.agent.client.netty.handler;
 import com.tongran.agent.client.core.config.GlobalConfig;
 import com.tongran.agent.client.core.session.SessionManager;
 import com.tongran.agent.client.scheduler.service.AppInitializer;
+import com.tongran.agent.client.service.AgentService;
 import com.tongran.agent.client.utils.AssertLog;
 import io.netty.channel.ChannelHandler;
 import io.netty.channel.ChannelHandlerContext;
@@ -26,6 +27,9 @@ public class TCPListenHandler extends ChannelInboundHandlerAdapter {
 
 	private final SessionManager sessionManager;
 
+	@Resource
+	private AgentService agentService;
+
 	public TCPListenHandler() {
 		this.sessionManager = SessionManager.getInstance();
 	}
@@ -41,6 +45,7 @@ public class TCPListenHandler extends ChannelInboundHandlerAdapter {
 		sessionManager.remove(ctx.channel());
 		try {
 			GlobalConfig.isCollect = false;
+			agentService.cancelCollect();
 			appInitializer.run();
 		} catch (Exception e) {
 			e.printStackTrace();
@@ -51,6 +56,7 @@ public class TCPListenHandler extends ChannelInboundHandlerAdapter {
 	public void exceptionCaught(ChannelHandlerContext ctx, Throwable e) {
 		if (e instanceof IOException) {
 			AssertLog.info("<<<<<<[终端断开连接]{} {}", sessionManager.client(ctx), e.getMessage());
+			agentService.cancelCollect();
 		} else {
 			AssertLog.info(">>>>>>[消息处理异常]" + sessionManager.client(ctx), e);
 		}