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)))