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

在 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也将在不久后和大家见面，欢迎大家关注。