From 37cf01e1821c1526fb5880f91f5cfd7cc7bb7629 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8B=8F=E5=B0=B9=E5=B2=9A?= <770236076@qq.com>
Date: Thu, 31 Dec 2020 09:51:54 +0800
Subject: [PATCH] aa

---
 business/jxstore/event/event.go | 1 +
 controllers/event_controller.go | 3 +--
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/business/jxstore/event/event.go b/business/jxstore/event/event.go
index e7a751a7c..499a872bc 100644
--- a/business/jxstore/event/event.go
+++ b/business/jxstore/event/event.go
@@ -169,6 +169,7 @@ func ImMessage(userID string, ws *websocket.Conn) (err error) {
 			})
 		}
 	}
+	ws.Close()
 	return err
 }
 
diff --git a/controllers/event_controller.go b/controllers/event_controller.go
index ed895d7ef..54e7e1339 100644
--- a/controllers/event_controller.go
+++ b/controllers/event_controller.go
@@ -57,7 +57,6 @@ func (c *EventController) TestWebsocket() {
 	if err != nil {
 		log.Fatal(err)
 	}
-	defer ws.Close()
 	var (
 		userID = c.GetString("userID")
 	)
@@ -76,7 +75,7 @@ func (c *EventController) TestWebsocket() {
 		globals.SugarLogger.Debugf("TestWebsocket connection...")
 	}
 	c.EnableRender = false //Beego不启用渲染
-	event.ImMessage(userID, ws)
+	go event.ImMessage(userID, ws)
 }
 
 // @Title 查询聊天记录