From ea6ae2eb0e035320bde28c68736b42cbe1a8d3e1 Mon Sep 17 00:00:00 2001
From: better0332 <better0332@163.com>
Date: Fri, 12 May 2023 15:55:54 +0800
Subject: [PATCH] net: move OnConnClose before Swap (#277)

---
 pkg/proxy/backend/backend_conn_mgr.go | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pkg/proxy/backend/backend_conn_mgr.go b/pkg/proxy/backend/backend_conn_mgr.go
index eb26313a..327c8379 100644
--- a/pkg/proxy/backend/backend_conn_mgr.go
+++ b/pkg/proxy/backend/backend_conn_mgr.go
@@ -633,6 +633,8 @@ func (mgr *BackendConnManager) Close() error {
 	}
 	mgr.wg.Wait()
 
+	handErr := mgr.handshakeHandler.OnConnClose(mgr)
+
 	var connErr error
 	var addr string
 	mgr.processLock.Lock()
@@ -642,8 +644,6 @@ func (mgr *BackendConnManager) Close() error {
 	}
 	mgr.processLock.Unlock()
 
-	handErr := mgr.handshakeHandler.OnConnClose(mgr)
-
 	eventReceiver := mgr.getEventReceiver()
 	if eventReceiver != nil {
 		// Notify the receiver if there's any event.