im3

controllers/im.go

@@ -2,9 +2,6 @@ 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"
@@ -22,34 +19,37 @@ var maxMessageSize = int64(8192) // 最大的消息大小
 // @Failure 200 {object} controllers.CallResult
 // @router /StartWebSocket [get]
 func (c *IMController) StartWebSocket() {
-	upgrader.CheckOrigin = func(r *http.Request) bool {
-		return true
-	}
-	conn, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
-	if err != nil {
-		globals.SugarLogger.Errorf("upgrade error: %v", err)
-		return
-	}
-	defer conn.Close()
 
-	clientID := c.GetString("clientID")
-	globals.SugarLogger.Debugf("clientID=%s", clientID)
-
-	//设置读取消息大小上线
-	conn.SetReadLimit(maxMessageSize)
-
-	clientSocket := im.NewClient(clientID, conn)
-
-	//读取客户端消息
-	clientSocket.Read()
-
-	if err = im.ConnRender(conn, im.RetData{Data: clientID}); err != nil {
-		_ = conn.Close()
-		return
-	}
-
-	// 用户连接事件
-	im.Manager.Connect <- clientSocket
+	im.Run(c.Ctx.ResponseWriter, c.Ctx.Request)
+	//
+	//upgrader.CheckOrigin = func(r *http.Request) bool {
+	//	return true
+	//}
+	//conn, err := upgrader.Upgrade(c.Ctx.ResponseWriter, c.Ctx.Request, nil)
+	//if err != nil {
+	//	globals.SugarLogger.Errorf("upgrade error: %v", err)
+	//	return
+	//}
+	//defer conn.Close()
+	//
+	//clientID := c.GetString("clientID")
+	//globals.SugarLogger.Debugf("clientID=%s", clientID)
+	//
+	////设置读取消息大小上线
+	//conn.SetReadLimit(maxMessageSize)
+	//
+	//clientSocket := im.NewClient(clientID, conn)
+	//
+	////读取客户端消息
+	//clientSocket.Read()
+	//
+	//if err = im.ConnRender(conn, im.RetData{Data: clientID}); err != nil {
+	//	_ = conn.Close()
+	//	return
+	//}
+	//
+	//// 用户连接事件
+	//im.Manager.Connect <- clientSocket
 
 }
 

controllers/im_model.go

@@ -1 +0,0 @@
-package controllers