WebSocket 服务端

概念说明

WebSocket 是一种在单个 TCP 连接上进行双向通信的协议。
浏览器和服务端建立连接后，双方都可以主动发送消息。

Go 标准库没有直接提供 WebSocket 升级封装。
下面示例使用常见第三方包 github.com/gorilla/websocket。

安装依赖

1
2
go mod init go-network-practice
go get github.com/gorilla/websocket

如果你已经在现成的 Go 模块里练习，可以跳过 go mod init。

语法/规则

WebSocket 服务端通常先通过 HTTP 接收请求。
使用 Upgrader.Upgrade 把 HTTP 连接升级为 WebSocket 连接。
使用 ReadMessage 读取客户端消息。
使用 WriteMessage 写回服务端消息。
连接结束后要关闭。
生产环境不要随意放开 CheckOrigin。

WebSocket 服务端示例

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true // 示例中直接放行，生产环境不要这样写
	},
}

func wsHandler(w http.ResponseWriter, r *http.Request) {
	conn, err := upgrader.Upgrade(w, r, nil) // 先把 HTTP 请求升级成 WebSocket 连接
	if err != nil {
		fmt.Println("upgrade error:", err)
		return
	}
	defer conn.Close()

	for {
		messageType, message, err := conn.ReadMessage() // 读取客户端消息
		if err != nil {
			fmt.Println("read error:", err)
			return
		}

		fmt.Println("server receive:", string(message))
		err = conn.WriteMessage(messageType, []byte("echo: "+string(message))) // 回显消息给客户端
		if err != nil {
			fmt.Println("write error:", err)
			return
		}
	}
}

func main() {
	http.HandleFunc("/ws", wsHandler)
	fmt.Println("websocket server listen on :8080")

	err := http.ListenAndServe(":8080", nil)
	if err != nil {
		panic(err)
	}
}

连接地址：

1
ws://127.0.0.1:8080/ws

常见错误

没有先执行 HTTP 升级，就按普通 HTTP 请求处理 WebSocket 消息。
生产环境直接 CheckOrigin 返回 true，可能带来跨站风险。
读取或写入出错后不关闭连接，导致资源泄漏。
忘记安装第三方依赖，导致编译时找不到包。

章节

17.7 WebSocket 服务端

WebSocket 服务端

概念说明

安装依赖

语法/规则

WebSocket 服务端示例

常见错误