EasyRTC编译中Golang 将 http 请求升级为 websocket实现过程分享

Posted 2021-08-10 EasyNVR

在 EasyRTC视频通话项目的编译中，部分操作采用 websocket 连接减少发送请求，增加实时性。一般情况下，会添加新端口的方式，在端口上做 websocket 服务。但是为了减少端口的使用，还有一种可以直接将 http 请求升级为 websokcet 协议。代码如下：

p2pServer := p2p.NewP2PServer(p2p.HandleNewWebSocket)
// 升级 websocket
wsPath := "/ws"
engine.GET(wsPath, p2pServer.UpgradeWebSocket)
engine.POST(wsPath, p2pServer.UpgradeWebSocket)

以上代码代表，如果遇见路由 /ws 就升级为 websokcet 端口。实现代码如下：

func (server *P2PServer) UpgradeWebSocket(c *gin.Context) {
   server.handleWebSocketRequest(c.Writer, c.Request)
}

//WebSocket请求处理
func (server *P2PServer) handleWebSocketRequest(writer http.ResponseWriter, request *http.Request) {
   //返回头
   responseHeader := http.Header{}
   //responseHeader.Add("Sec-WebSocket-Protocol", "protoo")
   //升级为长连接
   socket, err := server.upgrader.Upgrade(writer, request, responseHeader)
   //输出错误日志
   if err != nil {
      gLogger.Panic("p2p 升级为 ws 长连接失败。", zap.Error(err))
      return
   }
   //实例化一个WebSocketConn对象
   wsTransport := ews.NewWebSocketConn(socket)
   //处理具体的请求消息
   server.handleWebSocket(wsTransport, request)
   //WebSocketConn开始读取消息
   wsTransport.ReadMessage()
}

这里主要重写了 writer 和 request 请求，从而实现 websocket 长连接。

EasyRTC管理平台采用基于Java的SSH架构、MCU模式，系统稳定性高，EasyRTC支持会议录播、实时直播，以更流畅的成像和更低的延时给用户带来不同的视频会议体验，真正实现只要能上网，就能参加视频会议的需求。目前EasyRTC正在进行新架构的研发，对于原有的架构，我们部分做了修改和保留，新架构下的EasyRTC也将在不久后和大家见面，欢迎大家关注。