Golang中实时推送的功臣 - WebSocket

Golang中实时推送的功臣 - WebSocket

HTTP的问题是 一问一答，短连接

• 客户端主动发请求 → 服务端才返回数据
• 服务端不能主动给客户端发消息
• 每次通信都要重新建立连接（像发短信）

WebSocket 就是为了解决这个问题诞生的,基于 TCP 的 双向长连接 通信协议

• 一次连接，永久保持（直到断开）
• 客户端 ↔ 服务端 随时互相发消息
• 服务端可以主动推送消息给客户端
• 像打电话：接通后，你随时说，对方随时说

WebSocket 是一个「实时双向通信」的协议，底层用 TCP，借用 HTTP 建立连接，专门用来做实时功能。

WebSocket 能干嘛？

• 服务端主动推消息（HTTP 做不到！）

• 双向实时通信（毫秒级延迟）

• 长连接，不用反复创建连接

• 轻量，开销比 HTTP 小很多

• 浏览器原生支持（网页直接用）

• 实时推送、聊天、直播

极简示例

package main

import (
	"fmt"
	"net/http"

	"github.com/gorilla/websocket" // go get github.com/gorilla/websocket
)

// 定义 WebSocket 升级器：把 HTTP 连接升级成 WebSocket
var upgrader = websocket.Upgrader{
	// 允许跨域（测试用）
	CheckOrigin: func(r *http.Request) bool {
		return true
	},
}

// 处理 WebSocket 连接
func wsHandler(w http.ResponseWriter, r *http.Request) {
	// 1. 把 HTTP 连接升级为 WebSocket 连接
	conn, err := upgrader.Upgrade(w, r, nil)
	if err != nil {
		fmt.Println("升级失败：", err)
		return
	}
	defer conn.Close()

	fmt.Println("客户端已连接！")

	// 2. 循环接收/发送消息（长连接）
	for {
		// 读取客户端消息
		msgType, msg, err := conn.ReadMessage()
		if err != nil {
			fmt.Println("客户端断开连接")
			break
		}

		fmt.Printf("收到消息：%s\n", string(msg))

		// 服务端主动回消息
		reply := "服务端收到：" + string(msg)
		conn.WriteMessage(msgType, []byte(reply))
	}
}

func main() {
	http.HandleFunc("/ws", wsHandler)
	fmt.Println("WebSocket 服务启动：ws://127.0.0.1:8081/ws")
	http.ListenAndServe(":8081", nil)
	// 测试： wscat -c ws://127.0.0.1:8081/ws
}

核心 API

websocket.Upgrader把 HTTP 连接 → 升级成 WebSocket 长连接（WebSocket 必须先用 HTTP 握手）

var upgrader = websocket.Upgrader{
	CheckOrigin: func(r *http.Request) bool {
		return true // 允许跨域（开发必开）
	},
}

upgrader.Upgrade()执行升级，得到 WebSocket 连接对象 *websocket.Conn

conn, err := upgrader.Upgrade(w, r, nil)

conn.ReadMessage()阻塞读取客户端发来的消息返回：消息类型、消息内容、错误

msgType, msg, err := conn.ReadMessage()

conn.WriteMessage()服务端主动发消息给客户端（核心能力！）

conn.WriteMessage(msgType, []byte("你好客户端"))

conn.ReadJSON() / conn.WriteJSON()直接收发 JSON 结构体（不用手动序列化）

// 接收 JSON
var data map[string]any
conn.ReadJSON(&data)

// 发送 JSON
conn.WriteJSON(map[string]any{
    "code": 200,
    "msg":  "服务端推送",
})