如何在Go中使用WebSocket？-Golang-PHP中文網

如何在Go中使用WebSocket？

WBOY

發布： 2023-05-11 16:17:07

原創

2412 人瀏覽過

近年來，WebSocket技術已經成為了Web開發中不可或缺的一部分。 WebSocket 是一種在單一 TCP 連線上進行全雙工通訊的協議，它使得客戶端和伺服器之間的通訊更加流暢和有效率。如今，許多現代的網路應用程式都使用了WebSocket技術，例如即時聊天、線上遊戲以及即時數據視覺化等。

Go語言作為一個現代的程式語言，自然也提供了很好的支援WebSocket技術的功能。在本文中，我們將深入探討如何在Go中使用WebSocket技術。

WebSocket在Go語言中的實作

WebSocket 協定是由HTTP升級協定而來，因此在Go語言中使用WebSocket也需要將其與HTTP協定結合。 Go語言已經內建了net/http包，提供了各種HTTP相關的功能。如果我們想要在Go中使用WebSocket，只需導入net/http和github.com/gorilla/websocket庫即可。

下面，我們就來看看如何使用github.com/gorilla/websocket函式庫來實作WebSocket。

建立WebSocket服務端

在Go語言中，建立WebSocket服務端的方式也比較簡單。我們直接看下列程式碼，它展示瞭如何建立一個HTTP伺服器，等待客戶端的WebSocket連線：

package main

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

var upgrader = websocket.Upgrader{
    ReadBufferSize:  1024,
    WriteBufferSize: 1024,
}

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    http.ListenAndServe(":8080", nil)
}

func handleWebSocket(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()

    for {
        messageType, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            return
        }
        log.Printf("Received message: %s
", message)

        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            return
        }
        log.Printf("Sent message: %s
", message)
    }
}

登入後複製

在上面的程式碼中，我們首先建立了一個websocket.Upgrader對象，它負責將傳入的HTTP連線升級為WebSocket連線。然後，我們建立了一個handleWebSocket函數，它會在客戶端進行連線時被呼叫。此函數會先將傳入的HTTP連接升級為WebSocket連接，然後進入循環，不斷地接收來自客戶端的信息，並將接收到的信息回傳給客戶端。其中，conn.ReadMessage()和conn.WriteMessage()分別用於讀取和寫入資訊。

最後，我們將該handleWebSocket函數綁定到/ws路由上，並啟動了HTTP伺服器，以便客戶端可以連接到該WebSocket伺服器。如果一切正常，當我們啟動上述程式碼後，您會看到類似如下輸出：

2021/10/13 15:20:06 http: Server started at http://127.0.0.1:8080

登入後複製

這表示您已經成功地建立了一個WebSocket伺服器，客戶端可以透過ws:// localhost:8080/ws位址來連接該伺服器。

建立WebSocket客戶端

與建立WebSocket服務端類似，我們也可以很容易地在Go語言中建立一個WebSocket客戶端。我們來看下面的程式碼，它描述瞭如何建立一個與WebSocket伺服器建立連線的客戶端：

package main

import (
    "fmt"
    "log"
    "net/url"
    "os"
    "os/signal"
    "syscall"
    "github.com/gorilla/websocket"
)

func main() {
    interrupt := make(chan os.Signal, 1)
    signal.Notify(interrupt, os.Interrupt, syscall.SIGTERM)

    u := url.URL{Scheme: "ws", Host: "localhost:8080", Path: "/ws"}
    log.Printf("Connecting to %s", u.String())

    conn, _, err := websocket.DefaultDialer.Dial(u.String(), nil)
    if err != nil {
        log.Fatal("dial:", err)
    }
    defer conn.Close()

    done := make(chan struct{})
    go func() {
        defer close(done)
        for {
            messageType, message, err := conn.ReadMessage()
            if err != nil {
                log.Println("read:", err)
                return
            }
            log.Printf("Received message: %s
", message)
            if err := conn.WriteMessage(messageType, message); err != nil {
                log.Println("write:", err)
                return
            }
            log.Printf("Sent message: %s
", message)
        }
    }()

    for {
        select {
        case <-done:
            return
        case <-interrupt:
            log.Println("interrupt")

            if err := conn.WriteMessage(websocket.CloseMessage, websocket.FormatCloseMessage(websocket.CloseNormalClosure, "")); err != nil {
                log.Println("write close:", err)
                return
            }
            select {
            case <-done:
            case <-time.After(time.Second):
            }
            return
        }
    }
}

登入後複製

在上面的程式碼中，我們使用websocket.Dial方法來與WebSocket伺服器建立連線。建立連線後，我們再進入一個無限循環，不斷地傳送和接收來自WebSocket伺服器的訊息。其中，conn.ReadMessage()和conn.WriteMessage()用於讀取和寫入資訊。

值得注意的是，在上述程式碼中，我們使用了channels來等待操作完成。如果接收到interrupt訊號，我們將使用websocket.CloseMessage訊息傳送給伺服器，以關閉WebSocket連線。