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

Golang vs. Python：並發和多線程Apr 17, 2025 am 12:20 AM

Golang更適合高並發任務，而Python在靈活性上更有優勢。 1.Golang通過goroutine和channel高效處理並發。 2.Python依賴threading和asyncio，受GIL影響，但提供多種並發方式。選擇應基於具體需求。

Golang和C：性能的權衡Apr 17, 2025 am 12:18 AM

Golang和C 在性能上的差異主要體現在內存管理、編譯優化和運行時效率等方面。 1)Golang的垃圾回收機制方便但可能影響性能，2)C 的手動內存管理和編譯器優化在遞歸計算中表現更為高效。

Golang vs. Python：申請和用例Apr 17, 2025 am 12:17 AM

selectgolangforhighpperformanceandcorrency，ifealforBackendServicesSandNetwork程序; selectpypypythonforrapiddevelopment，dataScience和machinelearningDuetoitsverserverserverserversator versator anderticality andextility andextentensivelibraries。

Golang vs. Python：主要差異和相似之處Apr 17, 2025 am 12:15 AM

Golang和Python各有优势：Golang适合高性能和并发编程，Python适用于数据科学和Web开发。Golang以其并发模型和高效性能著称，Python则以简洁语法和丰富库生态系统著称。

Golang vs. Python：易於使用和學習曲線Apr 17, 2025 am 12:12 AM

Golang和Python分別在哪些方面更易用和學習曲線更平緩？ Golang更適合高並發和高性能需求，學習曲線對有C語言背景的開發者較平緩。 Python更適合數據科學和快速原型設計，學習曲線對初學者非常平緩。

表演競賽：Golang vs.CApr 16, 2025 am 12:07 AM

Golang和C 在性能競賽中的表現各有優勢：1)Golang適合高並發和快速開發，2)C 提供更高性能和細粒度控制。選擇應基於項目需求和團隊技術棧。

Golang vs.C：代碼示例和績效分析Apr 15, 2025 am 12:03 AM

Golang適合快速開發和並發編程，而C 更適合需要極致性能和底層控制的項目。 1)Golang的並發模型通過goroutine和channel簡化並發編程。 2)C 的模板編程提供泛型代碼和性能優化。 3)Golang的垃圾回收方便但可能影響性能，C 的內存管理複雜但控制精細。

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

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

See all articles