快捷導(dǎo)航

Golang利用WebSocket實(shí)現(xiàn)實(shí)時(shí)推送功能

更新時(shí)間：2026年03月15日 09:05:14 作者：我叫黑大帥

WebSocket 是一個(gè)實(shí)時(shí)雙向通信的協(xié)議,底層用 TCP,借用 HTTP 建立連接,專門用來做實(shí)時(shí)功能,下面小編就和大家詳細(xì)介紹一下Golang如何利用WebSocket實(shí)現(xiàn)實(shí)時(shí)推送功能吧

WebSocket 能干嘛

服務(wù)端主動推消息（HTTP 做不到?。?/li>
雙向?qū)崟r(shí)通信（毫秒級延遲）
長連接，不用反復(fù)創(chuàng)建連接
輕量，開銷比 HTTP 小很多
瀏覽器原生支持（網(wǎng)頁直接用）
實(shí)時(shí)推送、聊天、直播

極簡示例

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. 循環(huán)接收/發(fā)送消息（長連接）
	for {
		// 讀取客戶端消息
		msgType, msg, err := conn.ReadMessage()
		if err != nil {
			fmt.Println("客戶端斷開連接")
			break
		}

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

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

func main() {
	http.HandleFunc("/ws", wsHandler)
	fmt.Println("WebSocket 服務(wù)啟動：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 // 允許跨域（開發(fā)必開）
	},
}

upgrader.Upgrade()執(zhí)行升級，得到 WebSocket 連接對象*websocket.Conn

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

conn.ReadMessage()阻塞讀取客戶端發(fā)來的消息返回：消息類型、消息內(nèi)容、錯(cuò)誤

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

conn.WriteMessage()服務(wù)端主動發(fā)消息給客戶端（核心能力?。?/p>

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

conn.ReadJSON()/conn.WriteJSON()直接收發(fā) JSON 結(jié)構(gòu)體（不用手動序列化）

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

// 發(fā)送 JSON
conn.WriteJSON(map[string]any{
    "code": 200,
    "msg":  "服務(wù)端推送",
})

到此這篇關(guān)于Golang利用WebSocket實(shí)現(xiàn)實(shí)時(shí)推送功能的文章就介紹到這了,更多相關(guān)Golang WebSocket實(shí)時(shí)推送內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: