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

近年來，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連線。