From a80c1a5aaa1cecd2439ff253a51470abc5f1e5b8 Mon Sep 17 00:00:00 2001
From: dushixiang <helloworld1024@foxmail.com>
Date: Sun, 24 Jan 2021 18:03:22 +0800
Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0=E5=AF=B9=E6=9C=AA=E5=BB=BA?=
 =?UTF-8?q?=E7=AB=8B=E6=88=90=E5=8A=9F=E4=BC=9A=E8=AF=9D=E7=9A=84=E5=A4=84?=
 =?UTF-8?q?=E7=90=86?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 pkg/api/session.go | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/pkg/api/session.go b/pkg/api/session.go
index 94383e6..64f8432 100644
--- a/pkg/api/session.go
+++ b/pkg/api/session.go
@@ -102,6 +102,7 @@ func CloseSessionById(sessionId string, code int, reason string) {
 		_ = tun.Tun.Close()
 		CloseSessionByWebSocket(tun.WebSocket, code, reason)
 	}
+	global.Store.Del(sessionId)
 
 	s, err := model.FindSessionById(sessionId)
 	if err != nil {
@@ -112,7 +113,12 @@ func CloseSessionById(sessionId string, code int, reason string) {
 		return
 	}
 
-	global.Store.Del(sessionId)
+	if s.Status == model.Connecting {
+		// 会话还未建立成功，无需保留数据
+		model.DeleteSessionById(sessionId)
+		return
+	}
+
 	session := model.Session{}
 	session.ID = sessionId
 	session.Status = model.Disconnected