golang搭建websocket-Golang-PHP中文網

首頁

後端開發

Golang

golang搭建websocket

WBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWBOYWB

May 22, 2023 pm 07:21 PM

Golang是一種快速編譯、有效率、安全的程式語言，在網路開發方面也有廣泛的應用。其中，使用Golang搭建Websocket是一個非常流行的應用場景。本文將介紹如何使用Golang建構一個Websocket應用程式。

一、什麼是Websocket

Websocket是一種協議，它允許在瀏覽器和伺服器之間建立雙向通訊的連線。傳統的HTTP協定只能由瀏覽器向伺服器發出請求，並接收伺服器回應。但是，使用Websocket可以讓伺服器主動向瀏覽器發送訊息，實現了伺服器與瀏覽器的全雙工通訊。

二、Golang如何實作Websocket

Golang自帶了一個標準函式庫net/http，這個套件提供了一個http.Server結構體，它可以用來建構Web伺服器。除此之外，net/http也提供了一個websocket子包，它為我們提供了建構WebSocket伺服器所需的全部功能。

建立WebSocket連接

要建立WebSocket連接，客戶端需要向伺服器發送一個HTTP請求，並攜帶Upgrade頭部，指示請求升級為WebSocket協定。伺服器端在收到此請求後，需要檢查Upgrade頭部，並根據是否為WebSocket請求來進行相應的處理。

伺服器端可以使用websocket.IsWebSocketUpgrade函數來檢查請求是否為WebSocket請求，並使用websocket.Upgrade函數升級請求。升級成功後，伺服器就可以透過*websocket.Conn類型的連線物件與客戶端進行通訊了。

func handler(w http.ResponseWriter, r *http.Request) {
    conn, err := upgrader.Upgrade(w, r, nil)
    if err != nil {
        log.Println(err)
        return
    }
    defer conn.Close()
    // 使用conn进行通信
}

登入後複製

其中，upgrader是一個websocket.Upgrader類型的變量，透過設定它的屬性可以控制WebSocket連線的處理方式。 Upgrade函數在升級請求時會阻塞，因此要將其放在goroutine中運行。

傳送和接收WebSocket訊息

一旦WebSocket連線建立成功，客戶端和伺服器都可以透過*websocket.Conn物件進行通訊。使用這個物件可以發送和接收WebSocket訊息。

傳送訊息很簡單，只需要呼叫WriteMessage方法：

err := conn.WriteMessage(websocket.TextMessage, []byte("hello"))

登入後複製

接收訊息的方法有三種：

ReadMessage：阻塞讀取，直到有訊息到來
NextReader：阻塞等待下一個訊息到來
NextWriter：傳回一個io.WriteCloser對象，可以使用這個對象發送訊息

一般使用ReadMessage方法接收WebSocket訊息：

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

登入後複製

三、一個簡單的WebSocket應用程式

在這個範例中，我們將建立一個簡單的聊天室應用程式，客戶端發送訊息之後，伺服器將訊息廣播給所有連接的客戶端。

客戶端程式碼：（使用JavaScript）

<!DOCTYPE html>
<html>
<head>
    <title>Chatroom</title>
</head>
<body>
    <input type="text" id="input-message"/>
    <button onclick="sendMessage()">发送</button>
    <ul id="message-list"></ul>

    <script>
        var ws = new WebSocket("ws://localhost:8080/ws");
        ws.onmessage = function(event) {
            var message = event.data;
            var li = document.createElement("li");
            li.appendChild(document.createTextNode(message));
            document.getElementById("message-list").appendChild(li);
        };

        function sendMessage() {
            var message = document.getElementById("input-message").value;
            ws.send(message);
        }
    </script>
</body>
</html>

登入後複製

伺服器端程式碼：

package main

import (
    "fmt"
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

var upgrader = websocket.Upgrader{}

func main() {
    http.HandleFunc("/ws", wsHandler)
    err := http.ListenAndServe(":8080", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

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

    for {
        _, message, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }
        fmt.Printf("recv: %s
", message)

        // 广播给所有客户端
        for _, conn := range conns {
            err := conn.WriteMessage(websocket.TextMessage, message)
            if err != nil {
                log.Println(err)
                break
            }
        }
    }
}

登入後複製

在上面的程式碼中，我們將所有連接的websocket.Conn物件儲存在conns變數中。每當有一個新訊息到來時，伺服器會將訊息廣播給所有客戶端。

總結

使用Golang建立WebSocket應用程式非常簡單，只需要匯入net/http和websocket套件即可，直接使用標準庫提供的API即可完成WebSocket應用程式的開發。希望此文章能夠對Golang Websocket開發有所幫助。

以上是golang搭建websocket的詳細內容。更多資訊請關注PHP中文網其他相關文章！

本網站聲明

本文內容由網友自願投稿，版權歸原作者所有。本站不承擔相應的法律責任。如發現涉嫌抄襲或侵權的內容，請聯絡admin@php.cn

熱AI工具

熱工具

熱門話題

Java教學

1665

CakePHP 教程

1424

Laravel 教程

1322

PHP教程

1270

C# 教程

1250

Related knowledge

Golang vs. Python：性能和可伸縮性 Apr 19, 2025 am 12:18 AM

Golang在性能和可擴展性方面優於Python。 1)Golang的編譯型特性和高效並發模型使其在高並發場景下表現出色。 2)Python作為解釋型語言，執行速度較慢，但通過工具如Cython可優化性能。

Golang和C：並發與原始速度 Apr 21, 2025 am 12:16 AM

Golang在並發性上優於C ，而C 在原始速度上優於Golang。 1)Golang通過goroutine和channel實現高效並發，適合處理大量並發任務。 2)C 通過編譯器優化和標準庫，提供接近硬件的高性能，適合需要極致優化的應用。

Golang的影響：速度，效率和簡單性 Apr 14, 2025 am 12:11 AM

goimpactsdevelopmentpositationality throughspeed，效率和模擬性。 1）速度：gocompilesquicklyandrunseff，IdealforlargeProjects.2）效率：效率：ITScomprehenSevestAndardArdardArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdArdEcceSteral Depentencies，增強的Depleflovelmentimency.3）簡單性。

開始GO：初學者指南 Apr 26, 2025 am 12:21 AM

goisidealforbeginnersandsubableforforcloudnetworkservicesduetoitssimplicity，效率和concurrencyFeatures.1）installgromtheofficialwebsitealwebsiteandverifywith'.2）