From 094155d12448bb6bd6b9135302c68ccfe02266c6 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E8=8B=8F=E5=B0=B9=E5=B2=9A?= <770236076@qq.com>
Date: Wed, 19 Aug 2020 15:09:23 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B5=8B=E8=AF=95=E5=8F=91=E4=BF=A1=E6=81=AF?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 controllers/event.go                  | 22 ++++++++++++++++++++++
 routers/commentsRouter_controllers.go |  9 +++++++++
 2 files changed, 31 insertions(+)

diff --git a/controllers/event.go b/controllers/event.go
index 4b7df673c..d7b0ae459 100644
--- a/controllers/event.go
+++ b/controllers/event.go
@@ -3,6 +3,8 @@ package controllers
 import (
 	"time"
 
+	"github.com/gorilla/websocket"
+
 	"git.rosy.net.cn/baseapi/utils"
 
 	"git.rosy.net.cn/jx-callback/business/jxstore/event"
@@ -82,3 +84,23 @@ func (c *EventController) GetWeixinUnlimited() {
 		return retVal, "", err
 	})
 }
+
+// @Title 测试websocket
+// @Description 测试websocket
+// @Param	token			header		string	true	"认证token"
+// @Success 200 {object} controllers.CallResult
+// @Failure 200 {object} controllers.CallResult
+// @router /TestWebsocket [get]
+func (c *EventController) TestWebsocket() {
+	c.callTestWebsocket(func(params *tEventTestWebsocketParams) (retVal interface{}, errCode string, err error) {
+		var upgrader = websocket.Upgrader{}
+		conn, _ := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
+		go func(conn *websocket.Conn) {
+			for {
+				mType, msg, _ := conn.ReadMessage()
+				conn.WriteMessage(mType, msg)
+			}
+		}(conn)
+		return retVal, "", err
+	})
+}
diff --git a/routers/commentsRouter_controllers.go b/routers/commentsRouter_controllers.go
index 0b48de2c5..52f9b0f17 100644
--- a/routers/commentsRouter_controllers.go
+++ b/routers/commentsRouter_controllers.go
@@ -448,6 +448,15 @@ func init() {
             Filters: nil,
             Params: nil})
 
+    beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:EventController"] = append(beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:EventController"],
+        beego.ControllerComments{
+            Method: "TestWebsocket",
+            Router: `/TestWebsocket`,
+            AllowHTTPMethods: []string{"get"},
+            MethodParams: param.Make(),
+            Filters: nil,
+            Params: nil})
+
     beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FinancialController"] = append(beego.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FinancialController"],
         beego.ControllerComments{
             Method: "GetOrdersFinancial",