快捷導(dǎo)航

如何使用go實(shí)現(xiàn)創(chuàng)建WebSocket服務(wù)器

更新時(shí)間：2024年11月09日 10:44:10 作者：半桶水專家

文章介紹了如何使用Go語(yǔ)言和gorilla/websocket庫(kù)創(chuàng)建一個(gè)簡(jiǎn)單的WebSocket服務(wù)器,并實(shí)現(xiàn)商品信息的實(shí)時(shí)廣播,感興趣的朋友一起看看吧

使用Go語(yǔ)言創(chuàng)建WebSocket服務(wù)器可以利用現(xiàn)有的庫(kù)來(lái)簡(jiǎn)化開(kāi)發(fā)過(guò)程。gorilla/websocket 是一個(gè)非常流行且功能強(qiáng)大的庫(kù)，適用于Go語(yǔ)言的WebSocket應(yīng)用。下面是一個(gè)詳細(xì)的步驟指南，介紹如何使用 gorilla/websocket 創(chuàng)建一個(gè)WebSocket服務(wù)器，并實(shí)現(xiàn)商品信息的實(shí)時(shí)廣播。

安裝依賴

首先，你需要安裝 gorilla/websocket 庫(kù)。可以通過(guò)以下命令安裝：

go get github.com/gorilla/websocket

創(chuàng)建WebSocket服務(wù)器

接下來(lái)，我們將創(chuàng)建一個(gè)簡(jiǎn)單的WebSocket服務(wù)器，該服務(wù)器可以接收客戶端連接，并向所有連接的客戶端廣播商品信息。

1. 導(dǎo)入必要的包

package main
import (
    "fmt"
    "log"
    "net/http"
    "time"
    "github.com/gorilla/websocket"
)

2. 設(shè)置WebSocket升級(jí)器

WebSocket連接需要通過(guò)HTTP握手來(lái)建立。gorilla/websocket 提供了一個(gè) Upgrader 結(jié)構(gòu)體來(lái)處理這個(gè)過(guò)程。

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    // 允許跨域訪問(wèn)
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}

3. 創(chuàng)建WebSocket處理器

定義一個(gè)函數(shù)來(lái)處理WebSocket連接。在這個(gè)函數(shù)中，我們將讀取消息并廣播商品信息。

func serveWs(w http.ResponseWriter, r *http.Request) {
    // 升級(jí)HTTP連接到WebSocket連接
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Failed to set websocket upgrade:", err)
        return
    }
    defer conn.Close()
    // 模擬商品添加事件
    for {
        item := map[string]interface{}{
            "id":    1,
            "name":  "Apple",
            "price": 2.5,
        }
        // 將商品信息發(fā)送給客戶端
        err := conn.WriteJSON(item)
        if err != nil {
            log.Println("Error writing message:", err)
            break
        }
        // 每隔5秒發(fā)送一次
        time.Sleep(5 * time.Second)
    }
}

4. 注冊(cè)路由并啟動(dòng)HTTP服務(wù)器

最后，我們需要注冊(cè)路由并啟動(dòng)HTTP服務(wù)器。

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

完整代碼

將上述代碼片段組合在一起，得到完整的WebSocket服務(wù)器代碼：

package main
import (
    "fmt"
    "log"
    "net/http"
    "time"
    "github.com/gorilla/websocket"
)
var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
    CheckOrigin: func(r *http.Request) bool {
        return true
    },
}
func serveWs(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println("Failed to set websocket upgrade:", err)
        return
    }
    defer conn.Close()
    for {
        item := map[string]interface{}{
            "id":    1,
            "name":  "Apple",
            "price": 2.5,
        }
        err := conn.WriteJSON(item)
        if err != nil {
            log.Println("Error writing message:", err)
            break
        }
        time.Sleep(5 * time.Second)
    }
}
func main() {
    http.HandleFunc("/ws", serveWs)
    fmt.Println("Starting server on :8080")
    if err := http.ListenAndServe(":8080", nil); err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

運(yùn)行服務(wù)器

保存文件并運(yùn)行：

go run main.go

測(cè)試連接

你可以使用任何WebSocket客戶端工具（如瀏覽器的開(kāi)發(fā)者工具）來(lái)測(cè)試連接。連接到 ws://localhost:8080/ws，你應(yīng)該每5秒接收到一次商品信息。

擴(kuò)展功能

多客戶端支持：可以在服務(wù)器端維護(hù)一個(gè)連接列表，以便向所有連接的客戶端廣播消息。
消息類(lèi)型：可以定義不同的消息類(lèi)型，以便客戶端根據(jù)消息類(lèi)型做出相應(yīng)的處理。
錯(cuò)誤處理：添加更詳細(xì)的錯(cuò)誤處理和日志記錄，以便更好地調(diào)試和監(jiān)控。

到此這篇關(guān)于用go實(shí)現(xiàn)創(chuàng)建WebSocket服務(wù)器的文章就介紹到這了,更多相關(guān)go WebSocket服務(wù)器內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: