Go WebSocket：轻松构建你的第一个回声服务器

2023-11-23 03:30:49

Go WebSocket：揭秘 WebSocket 的奥秘

WebSocket 是一种全双工通信协议，它允许客户端和服务器之间建立持久连接，实现双向数据传输。与传统的 HTTP 请求-响应模型不同，WebSocket 允许双方在连接建立后随时发送和接收消息。这使得 WebSocket 非常适合构建实时通信应用程序，例如聊天室、在线游戏和协作工具。

构建 Go WebSocket 服务器的步骤：

1. 导入必要的库

import (
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

log：用于记录错误和信息。
net/http：用于创建和处理 HTTP 请求。
gorilla/websocket：一个功能强大的 Go WebSocket 库。

2. 创建 WebSocket 路由

func main() {
	http.HandleFunc("/websocket", handleWebSocket)
	log.Println("Server is listening on port 8080")
	http.ListenAndServe(":8080", nil)
}

main() 函数是程序的入口。
http.HandleFunc() 函数将一个 URL 路由映射到一个处理函数。
handleWebSocket 函数将在客户端连接到 /websocket 路由时被调用。
log.Println() 函数打印信息到控制台。
http.ListenAndServe() 函数启动 HTTP 服务器。

3. 处理 WebSocket 连接

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
	upgrader := websocket.Upgrader{
		ReadBufferSize:  1024,
		WriteBufferSize: 1024,
	}
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println("Error during WebSocket upgrade:", err)
		return
	}
	defer conn.Close()

	for {
		messageType, p, err := conn.ReadMessage()
		if err != nil {
			log.Println("Error during WebSocket read:", err)
			break
		}
		if messageType == websocket.CloseMessage {
			break
		}
		err = conn.WriteMessage(messageType, p)
		if err != nil {
			log.Println("Error during WebSocket write:", err)
			break
		}
	}
}