构建Go语言WebSocket服务器的实用指南

使用 Go 实现一个 WebSocket 服务器可以通过 golang.org/x/net/websocket 包或更流行的 github.com/gorilla/websocket 包来完成。下面是一个使用 gorilla/websocket 包的示例，这个包功能强大且易于使用。

首先，确保你已经安装了 gorilla/websocket 包。你可以使用以下命令安装它：

go get github.com/gorilla/websocket

然后，你可以编写一个简单的 WebSocket 服务器。以下是一个完整的示例代码：

package main

import (
	"fmt"
	"log"
	"net/http"

	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
	ReadBufferSize:  1024,
	WriteBufferSize: 1024,
}

func echo(w http.ResponseWriter, r *http.Request) {
	// 升级 HTTP 请求到 WebSocket 连接
	ws, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		log.Println(err)
		return
	}
	defer ws.Close()

	for {
		// 从客户端读取消息
		messageType, p, err := ws.ReadMessage()
		if err != nil {
			log.Println(err)
			return
		}
		fmt.Printf("Received message: %s\n", p)

		// 向客户端发送消息
		err = ws.WriteMessage(messageType, p)
		if err != nil {
			log.Println(err)
			return
		}
	}
}

func main() {
	http.HandleFunc("/ws", echo)

	fmt.Println("Starting WebSocket server on ws://localhost:8080/ws")
	if err := http.ListenAndServe(":8080", nil); err != nil {
		log.Fatal("ListenAndServe: ", err)
	}
}

解释

1. 导入必要的包：
   • fmt 和 log 用于打印和记录日志。
   • net/http 用于处理 HTTP 请求。
   • github.com/gorilla/websocket 提供了 WebSocket 的实现。
2. 定义 Upgrader：
   • websocket.Upgrader 用于将 HTTP 升级到 WebSocket 连接。
   • 你可以设置 ReadBufferSize 和 WriteBufferSize 来优化性能。
3. 定义 echo 函数：
   • echo 函数处理 WebSocket 连接。
   • 使用 upgrader.Upgrade 将 HTTP 请求升级到 WebSocket 连接。
   • 使用 ws.ReadMessage 从客户端读取消息。
   • 使用 ws.WriteMessage 向客户端发送消息。
   • 使用 defer ws.Close() 确保在函数返回时关闭 WebSocket 连接。
4. 在 main 函数中设置路由并启动服务器：
   • 使用 http.HandleFunc 将 /ws 路径映射到 echo 函数。
   • 使用 http.ListenAndServe 启动 HTTP 服务器，监听 localhost:8080。

测试

运行服务器：

go run main.go

使用 WebSocket 客户端（如浏览器控制台、Postman 或 WebSocket 测试工具）连接到 ws://localhost:8080/ws。

发送消息，服务器将回显你发送的消息。

这个示例展示了如何使用 Go 和 Gorilla WebSocket 包创建一个简单的 WebSocket 服务器。你可以根据需要扩展此示例，例如添加身份验证、处理不同的消息类型、广播消息到所有连接的客户端等。