golang 建立websocket

WebSocket是一种在Web浏览器和服务器之间进行全双工通信的协议。与传统的HTTP请求/响应协议不同，WebSocket允许服务器主动向客户端推送数据，实现了实时性强、效率高的通信方式。在golang中，我们可以通过使用内置的net/http包和gorilla/websocket包来建立WebSocket连接，实现实时通信。

使用net/http包建立WebSocket连接

在golang中，我们首先需要导入net/http包，并创建一个http处理函数来处理WebSocket连接。下面是一个简单的示例：

```go package main import ( "fmt" "log" "net/http" "github.com/gorilla/websocket" ) func main() { http.HandleFunc("/ws", handleWebSocket) log.Fatal(http.ListenAndServe(":8080", nil)) } func handleWebSocket(w http.ResponseWriter, r *http.Request) { conn, err := websocket.Upgrade(w, r, nil, 1024, 1024) if err != nil { log.Println(err) return } defer conn.Close() for { _, msg, err := conn.ReadMessage() if err != nil { log.Println(err) return } log.Println(string(msg)) reply := []byte("Received: " + string(msg)) if err := conn.WriteMessage(websocket.TextMessage, reply); err != nil { log.Println(err) return } } } ```

创建WebSocket连接

在上述示例中，我们创建了一个handleWebSocket处理函数来处理WebSocket连接。当客户端请求"/ws"路径时，会调用该函数。

在handleWebSocket函数中，我们首先使用websocket.Upgrade函数将HTTP连接升级为WebSocket连接，并返回与客户端的连接对象conn。如果升级失败，则返回相应的错误信息。

读写数据

在建立了WebSocket连接之后，我们可以通过conn.ReadMessage()函数读取客户端发送的消息，通过conn.WriteMessage()函数向客户端发送消息。

在上述示例中，我们通过conn.ReadMessage()读取客户端发送的消息，并使用log.Println()打印在控制台上。然后，对接收到的消息进行回复，使用conn.WriteMessage()函数将回复发送给客户端。

需要注意的是，conn.ReadMessage()和conn.WriteMessage()都是阻塞的操作，因此我们可以将它们放在一个无限循环中，持续地接收和发送消息。

其他操作

除了读写数据外，gorilla/websocket包还提供了其他一些功能，例如消息的序列化和反序列化、设置读写缓冲区大小等。我们可以根据实际需求去使用这些功能。

通过上述步骤，我们就可以在golang中建立WebSocket连接并实现实时通信。WebSocket的出现使得Web应用能够更加高效地处理实时消息和事件，为用户提供更好的用户体验。