websocket服务器测试

因为一个连接是不允许被多个线程、协程同时写入的，否则数据会乱，所以无论是开协程、线程去分批写入都没有用，没啥技巧，就是客户端的连接id准备好，循环一个个发。
这个是我发广播的代码：

// Broadcast 广播
func Broadcast(param []byte, _ *workerManager.ConnProcessor) {
    //解析业务进程发来的需要广播的数据
    payload := objPool.Broadcast.Get()
    defer objPool.Broadcast.Put(payload)
    if err := proto.Unmarshal(param, payload); err != nil {
        log.Logger.Error().Err(err).Msg("Proto unmarshal netsvrProtocol.Broadcast failed")
        return
    }
    //如果数据是空，直接退出
    if len(payload.Data) == 0 {
        return
    }
    //取出所有的连接
    //加1024的目的是担心获取连接的过程中又有连接进来，导致slice的底层发生扩容，引起内存拷贝
    connections := objPool.ConnSlice.Get(customerManager.Manager.Len() + 1024)
    for _, c := range customerManager.Manager {
        //从客户端连接管理对象中得到所有在线的连接
        c.GetConnections(connections)
    }
    //循环所有的连接，挨个发送出去
    for _, conn := range *connections {
        if err := conn.WriteMessage(websocket.TextMessage, payload.Data); err == nil {
            //发送成功，变更统计数据
            metrics.Registry[metrics.ItemCustomerWriteNumber].Meter.Mark(1)
            metrics.Registry[metrics.ItemCustomerWriteByte].Meter.Mark(int64(len(payload.Data)))
        } else {
            _ = conn.Close()
        }
    }
    objPool.ConnSlice.Put(connections)
}

我找了这两个websocket包，你可以参考一下。
gorilla/websocket
lesismal/nbio

如果还是觉得开发麻烦，想直接使用我这个，可以看看这个代码：func RequestNetSvr(req proto.Message, cmd netsvrProtocol.Cmd, resp proto.Message)
这个是我封装的与netsvr的worker服务器交互的方法，这个代码我刚才又优化了一版，你如果要复制到你的项目下，你得review一下，我刚才提交后又review一波，发现了一些小bug。

你的需求，用我这个项目来实现，大概的思路是：

1. 安装并启动netsvr，注意将配置netsvr.toml的ConnOpenCustomUniqIdKey = ""，改为ConnOpenCustomUniqIdKey = "uniqId"
2. 新增客户端获取接入netsvr的token的接口，服务端获取netsvr的token的代码如下：

   resp := &netsvrProtocol.ConnOpenCustomUniqIdTokenResp{}
   testUtils.RequestNetSvr(nil, netsvrProtocol.Cmd_ConnOpenCustomUniqIdToken, resp)

3. 客户端连接到netsvr成功后，请求服务端，告知服务端，客户端连接成功，服务端应该将该客户端的连接信息存储起来，便于后续使用，客户端连接到netsvr的地址示例：

   //token和uniqId都是服务端给
   new websocket("ws://127.0.0.1:6060/netsvr?uniqId=01xxx&token=o0o0o")

4. 改造导入excel的接口，通过单播到netsvr的方式，将导入进度下发给客户端，代码如下：

   ret := &netsvrProtocol.SingleCast{} //构建单播对象
   ret.UniqId = "0064DF23A97849A100"   //设置接收方
   ret.Data = []byte("code:0,msg:导入进度是10%")       //设置客户端需的数据
   testUtils.RequestNetSvr(ret, netsvrProtocol.Cmd_SingleCast, nil) //发送到网关

5. 客户端编写接收网关下发的导入进度的数据的处理逻辑

注意：以上思路都是基于netsvr只部署了一台机器的情况，如果要考虑部署多台机器的情况，有点复杂。