在 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 {