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