im

business/partner/purchase/im/im_server.go

@@ -3,7 +3,6 @@ package im
 import (
 	"errors"
 	"fmt"
-	"net/http"
 	"time"
 
 	"git.rosy.net.cn/baseapi/utils"
@@ -33,40 +32,19 @@ func Init() {
 	fmt.Printf("服务器启动成功，端口号：%s\n", CommonSetting.HttpPort)
 }
 
-func StartWebSocket(w http.ResponseWriter, r *http.Request) {
+var upgrader = websocket.Upgrader{}
 
-	conn, err := (&websocket.Upgrader{
-		ReadBufferSize:  8182,
-		WriteBufferSize: 8182,
-		// 允许所有CORS跨域请求
-		CheckOrigin: func(r *http.Request) bool {
-			return true
-		},
-	}).Upgrade(w, r, nil)
-
-	if err != nil {
-		globals.SugarLogger.Errorf("upgrade error: %v", err)
-		http.NotFound(w, r)
-		return
-	}
+func StartWebSocket(conn *websocket.Conn, clientID string, err error) {
 
 	//设置读取消息大小上线
 	conn.SetReadLimit(maxMessageSize)
 
-	clientId := ""
-	temp := r.Header["Clientid"]
-	if temp[0] != "" {
-		clientId = temp[0]
-	} else {
-		clientId = "defaultClientIDJXCS"
-	}
-
-	clientSocket := NewClient(clientId, conn)
+	clientSocket := NewClient(clientID, conn)
 
 	//读取客户端消息
 	clientSocket.Read()
 
-	if err = ConnRender(conn, renderData{ClientId: clientId}); err != nil {
+	if err = ConnRender(conn, renderData{ClientId: clientID}); err != nil {
 		_ = conn.Close()
 		return
 	}