如何使用Go語言建構WebSocket伺服器

WebSockets是一種先進的網路協議，允許客戶端和伺服器之間在單一TCP連接上進行雙向通訊。 WebSockets支援即時資料傳輸、線上遊戲、聊天室等應用程序，因此備受開發者歡迎。

Go語言是一種高效能、簡潔的程式語言，適合建構WebSockets伺服器。本文將詳細介紹如何使用Go語言建構WebSocket伺服器。

1. 瞭解WebSocket協定

WebSocket協定使用HTTP協定的握手和Upgrade頭，升級為WebSocket連線。 WebSocket連線始終使用回車符和換行符作為訊息結束符。 WebSocket訊息可以是文字或二進位資料。

2. 安裝Gorilla WebSocket庫

Gorilla WebSocket庫是Go語言中最受歡迎的WebSocket庫之一，它提供了WebSocket客戶端和伺服器的實作。您可以使用以下命令安裝Gorilla WebSocket庫：

go get github.com/gorilla/websocket

3. 建立WebSocket伺服器

下面是一個簡單的WebSocket伺服器範例：

package main

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

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

func main() {
    http.HandleFunc("/echo", echoHandler)
    log.Fatal(http.ListenAndServe(":8080", nil))
}

func echoHandler(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)
            break
        }
        log.Printf("Received message: %s
", message)
        err = conn.WriteMessage(messageType, message)
        if err != nil {
            log.Println(err)
            break
        }
    }
}

上面的程式碼中，echoHandler函式處理所有WebSocket連線。它使用Gorilla WebSocket庫提供的Upgrader將HTTP請求升級為WebSocket連線。連線升級後，伺服器使用conn.ReadMessage()從客戶端讀取訊息，並將訊息回顯回客戶端使用conn.WriteMessage()。

4. 運行WebSocket伺服器

現在，運行WebSocket伺服器。在終端機中，轉到Go檔案所在的目錄並執行以下命令：

go run main.go

此時，您應該會看到類似於以下內容的輸出：

2021/08/11 15:32:31 Listening on port 8080...

這意味著您的伺服器正在偵聽連接埠8080上的所有傳入連線。

5. 測試WebSocket伺服器

現在，您已成功建立了WebSocket伺服器，可以使用WebSocket客戶端測試它。建議使用Chrome瀏覽器進行測試，因為Chrome瀏覽器內建了WebSocket客戶端。在Chrome瀏覽器中開啟開發者工具，切換到控制台標籤並輸入以下程式碼：

var ws = new WebSocket("ws://localhost:8080/echo");
ws.onopen = function() {
    console.log("Connected");
    ws.send("Hello, World!");
};
ws.onmessage = function(event) {
    console.log("Received message: " + event.data);
};
ws.onclose = function() {
    console.log("Disconnected");
};

Chrome瀏覽器會與您的伺服器建立WebSocket連接，並傳送「Hello, World!」訊息。您的伺服器將回顯該訊息，並將其發送回客戶端。您應該在控制台中看到類似以下內容的輸出：

Connected
Received message: Hello, World!
Disconnected

6. 總結

#本文介紹如何使用Gorilla WebSocket庫和Go語言建立WebSocket伺服器。現在，您已經知道如何使用WebSocket協定實現即時通訊功能。透過WebSocket，您可以使用單一TCP連線傳輸數據，而無需為每個請求建立新的連線。這可以提高效能和可擴展性，因此值得在應用程式中使用。

以上是如何使用Go語言建構WebSocket伺服器的詳細內容。更多資訊請關注PHP中文網其他相關文章！