im

controllers/im.go

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