ws

controllers/im.go

@@ -3,6 +3,8 @@ package controllers
 import (
 	"encoding/json"
 
+	"git.rosy.net.cn/jx-callback/globals"
+
 	"git.rosy.net.cn/jx-callback/business/partner/purchase/im"
 	"github.com/astaxie/beego/server/web"
 )
@@ -16,18 +18,19 @@ type IMController struct {
 // @Success 200 {object} controllers.CallResult
 // @Failure 200 {object} controllers.CallResult
 // @router /StartWebSocket [get]
-//func (c *IMController) StartWebSocket() {
+func (c *IMController) StartWebSocket() {
-//	ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
+	ws, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
-//	if err != nil {
+	if err != nil {
-//		globals.SugarLogger.Errorf("upgrade error: %v", err)
+		globals.SugarLogger.Errorf("upgrade error: %v", err)
-//		return
+		return
-//	}
+	}
-//	defer ws.Close()
+	defer ws.Close()
-//
+
-//	clientID := c.GetString("clientID")
+	clientID := c.GetString("clientID")
-//	globals.SugarLogger.Debugf("clientID=%s", clientID)
+	globals.SugarLogger.Debugf("clientID=%s", clientID)
-//	im.StartWebSocket(ws, clientID, err)
+
-//}
+	im.StartWebSocket(ws, clientID, err)
+}
 
 // @Title IM获取门店用户聊天列表
 // @Description IM获取门店用户聊天列表

main.go

@@ -97,7 +97,7 @@ func Init() {
 	auto_delivery.Init() // 初始化骑手列表
 
 	go im.Init() //初始化ws连接
-	go http.HandleFunc("/v2/im/StartWebSocket", im.Run)
+	//go http.HandleFunc("/v2/im/StartWebSocket", im.Run)
 
 	//test
 	//mux := http.NewServeMux()

routers/commentsRouter_controllers.go

@@ -4356,14 +4356,14 @@ func init() {
 			MethodParams:     param.Make(),
 			Filters:          nil,
 			Params:           nil})
-	//web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"],
+	web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:IMController"],
-	//	web.ControllerComments{
+		web.ControllerComments{
-	//		Method:           "StartWebSocket",
+			Method:           "StartWebSocket",
-	//		Router:           `/StartWebSocket`,
+			Router:           `/StartWebSocket`,
-	//		AllowHTTPMethods: []string{"get"},
+			AllowHTTPMethods: []string{"get"},
-	//		MethodParams:     param.Make(),
+			MethodParams:     param.Make(),
-	//		Filters:          nil,
+			Filters:          nil,
-	//		Params:           nil})
+			Params:           nil})
 
 	//web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FnController"] = append(web.GlobalControllerRouter["git.rosy.net.cn/jx-callback/controllers:FnController"],
 	//	web.ControllerComments{