From 15243b39356b71f3d1d94e88a5ff82e4a50d3b7e Mon Sep 17 00:00:00 2001
From: naison <895703375@qq.com>
Date: Tue, 15 Apr 2025 21:33:41 +0800
Subject: [PATCH] hotfix: remove closed conn from route map (#545)

---
 pkg/core/tcphandler.go | 1 +
 1 file changed, 1 insertion(+)

diff --git a/pkg/core/tcphandler.go b/pkg/core/tcphandler.go
index 16aaf2c5..2705d084 100644
--- a/pkg/core/tcphandler.go
+++ b/pkg/core/tcphandler.go
@@ -101,6 +101,7 @@ func (h *UDPOverTCPHandler) Handle(ctx context.Context, tcpConn net.Conn) {
 func (h *UDPOverTCPHandler) removeFromRouteMapTCP(ctx context.Context, tcpConn net.Conn) {
 	h.routeMapTCP.Range(func(key, value any) bool {
 		if value.(net.Conn) == tcpConn {
+			h.routeMapTCP.Delete(key)
 			plog.G(ctx).Infof("[TCP] Delete to DST: %s by conn %s -> %s from globle route map TCP", key, tcpConn.RemoteAddr(), tcpConn.LocalAddr())
 		}
 		return true