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",