From 9b9a976663ba3a08039334cfe63afee9157f85d2 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=BC=A0=E5=BE=B7=E6=BB=A1?= <go_developer@163.com>
Date: Sun, 18 Apr 2021 01:02:17 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8D=87=E7=BA=A7connectionID=E7=94=9F?=
 =?UTF-8?q?=E6=88=90=E8=A7=84=E5=88=99?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 context/context.go | 7 ++++---
 1 file changed, 4 insertions(+), 3 deletions(-)

diff --git a/context/context.go b/context/context.go
index 9e472c9..08bb5bc 100644
--- a/context/context.go
+++ b/context/context.go
@@ -45,7 +45,7 @@ func NewContext(ginCtx *gin.Context, flag string, session *melody.Session) *WSCo
 	d, _ := easymap.NewSegment(128, true)
 	l, _ := easylock.NewSegment(128)
 	return &WSContext{
-		ConnectionID: generateConnectionID(),
+		ConnectionID: generateConnectionID(flag),
 		Flag:         flag,
 		GinCtx:       ginCtx,
 		Session:      session,
@@ -60,8 +60,9 @@ func NewContext(ginCtx *gin.Context, flag string, session *melody.Session) *WSCo
 // Author : go_developer@163.com<张德满>
 //
 // Date : 5:18 下午 2021/4/8
-func generateConnectionID() string {
-	return fmt.Sprintf("%s-%s-%s",
+func generateConnectionID(flag string) string {
+	return fmt.Sprintf("%s-%s-%s-%s",
+		flag,
 		strings.ReplaceAll(util.GetHostIP(), ".", ""),
 		time.Now().Format("20060102150405"),
 		util.Md5(fmt.Sprintf("%d", time.Now().UnixNano())+util.GenRandomString("", 64)))