go 语言内置了对 websocket 的支持，可通过以下步骤实现双向通信：创建 websocket 服务端：创建 http 处理程序函数。创建 upgrader 结构体升级 http 请求。启动 websocket 服务端。创建 websocket 客户端：创建 http 请求并设置 websocket 头。创建 dialer 结构体。使用 dial 方法建立连接。
Go WebSocket：实现双向通信
WebSocket 是一种双向通信协议，允许客户端与服务器在单个持久的 TCP 连接上进行实时数据交换。Go 语言为开发者提供了针对 WebSocket 的内置支持，使其实现双向通信变得简单直接。
创建 WebSocket 服务端
创建 WebSocket 服务端需要三步：
- 使用 net/http 包创建 http.Handler 函数，该函数处理 WebSocket 连接。
- 创建 websocket.Upgrader 结构体，该结构体将 HTTP 请求升级为 WebSocket 连接。
- 使用 http.ServeHTTP 方法启动 WebSocket 服务端。
package main
import (
"log"
"net/http"
"<a style='color:#f60; text-decoration:underline;' href="https://www.codesou.cn/" target="_blank">git</a>hub.com/gorilla/websocket"
)
func main() {
// HTTP 处理程序
handler := func(w http.ResponseWriter, r *http.Request) {
upgrader := websocket.Upgrader{}
conn, err := upgrader.Upgrade(w, r, nil)
if err != nil {
log.Fatal(err)
}
defer conn.Close()
for {
messageType, message, err := conn.ReadMessage()
if err != nil {
log.Fatal(err)
}
if messageType == websocket.CloseMessage {
log.Println("Client closed connection")
return
}
log.Printf("Received from client: %s", string(message))
err = conn.WriteMessage(messageType, []byte("Echoed: "+string(message)))
if err != nil {
log.Fatal(err)
}
}
}
// 启动 WebSocket 服务端
http.HandleFunc("/websocket", handler)
err := http.ListenAndServe(":8080", nil)
if err != nil {
log.Fatal(err)
}
}
创建 WebSocket 客户端
创建 WebSocket 客户端同样需要三步：
- 使用 net/http 包创建 http.Request，并设置 “Upgrade: WebSocket” 头。
- 创建 websocket.Dialer 结构体，该结构体用于建立 WebSocket 连接。
- 使用 Dial 方法建立连接。
package main
import (
"fmt"
"log"
"github.com/gorilla/websocket"
)
func main() {
dialers := websocket.Dialer{
ReadBufferSize: 1024,
WriteBufferSize: 1024,
}
conn, _, err := dialers.Dial("ws://localhost:8080/websocket", nil)
if err != nil {
log.Fatal(err)
}
defer conn.Close()
for i := 0; i < 10; i++ {
message := fmt.Sprintf("Hello from client %d", i)
err := conn.WriteMessage(websocket.TextMessage, []byte(message))
if err != nil {
log.Fatal(err)
}
_, message, err = conn.ReadMessage()
if err != nil {
log.Fatal(err)
}
log.Printf("Received echo: %s", string(message))
}
}
通过这些步骤，您可以轻松地在 Go 中实现 WebSocket 双向通信，为您的应用程序添加实时交互功能。
