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" "github.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 双向通信,为您的应用程序添加实时交互功能。
以上是Go WebSocket 如何实现双向通信?的详细内容。更多信息请关注PHP中文网其他相关文章!