Golang利用WebSocket实现实时推送功能
作者：我叫黑大帅
WebSocket 是一个实时双向通信的协议,底层用 TCP,借用 HTTP 建立连接,专门用来做实时功能,下面小编就和大家详细介绍一下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": "服务端推送",
})
到此这篇关于Golang利用WebSocket实现实时推送功能的文章就介绍到这了,更多相关Golang WebSocket实时推送内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！