net: move OnConnClose before Swap (#277)

pingcap · May 12, 2023 · ea6ae2e · ea6ae2e
1 parent d0e6292
commit ea6ae2e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.