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

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

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

使用Go語言創(chuàng)建WebSocket服務器可以利用現(xiàn)有的庫來簡化開發(fā)過程。gorilla/websocket 是一個非常流行且功能強大的庫，適用于Go語言的WebSocket應用。下面是一個詳細的步驟指南，介紹如何使用 gorilla/websocket 創(chuàng)建一個WebSocket服務器，并實現(xiàn)商品信息的實時廣播。

安裝依賴

首先，你需要安裝 gorilla/websocket 庫。可以通過以下命令安裝：

go get github.com/gorilla/websocket

創(chuàng)建WebSocket服務器

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

1. 導入必要的包

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

2. 設置WebSocket升級器

WebSocket連接需要通過HTTP握手來建立。gorilla/websocket 提供了一個 Upgrader 結構體來處理這個過程。

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

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

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

func serveWs(w http.ResponseWriter, r *http.Request) {
    // 升級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. 注冊路由并啟動HTTP服務器

最后，我們需要注冊路由并啟動HTTP服務器。

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服務器代碼：

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)
    }
}