golang WebSocket範例：建立一個簡單的聊天應用

Golang WebSocket範例：建立一個簡單的聊天應用程式

#隨著網路的發展，即時通訊越來越受人們的關注。 WebSocket作為一種即時通訊協議，能夠在客戶端和伺服器之間建立持久性的連接，使得雙向即時資料傳輸成為可能。本文將介紹如何使用Golang建立一個簡單的聊天應用，透過WebSocket實現即時訊息的發送和接收。

在開始之前，確保你已經在電腦上正確安裝了Golang開發環境。

首先，我們需要導入"Gorilla Websocket"包，這是Golang中一個常用的WebSocket庫。在終端機中使用以下命令安裝：

go get github.com/gorilla/websocket

接下來，我們建立一個main.go檔案並匯入所需的套件：

package main

import (
    "log"
    "net/http"

    "github.com/gorilla/websocket"
)

在main函數中，我們定義一個WebSocket處理器函數，用於處理來自客戶端的WebSocket連接請求：

func wsHandler(w http.ResponseWriter, r *http.Request) {
    // 升级HTTP连接为WebSocket连接
    conn, err := websocket.Upgrade(w, r, nil, 1024, 1024)
    if err != nil {
        log.Println(err)
        return
    }

    // 处理WebSocket连接
    for {
        // 读取客户端发送的消息
        _, msg, err := conn.ReadMessage()
        if err != nil {
            log.Println(err)
            break
        }

        // 将消息发送给所有连接的客户端
        for client := range clients {
            err := client.WriteMessage(websocket.TextMessage, msg)
            if err != nil {
                log.Println(err)
                client.Close()
                delete(clients, client)
            }
        }
    }

    // 关闭WebSocket连接
    conn.Close()
}

在上面的程式碼中，我們首先將HTTP連接升級為WebSocket連接，然後使用for循環不斷地接收客戶端發送的訊息，並將這些訊息發送給所有連接的客戶端。當有錯誤發生或客戶端斷開連線時，我們會關閉對應的連線。

接下來，我們需要定義一個全域變數clients來保存所有連接的客戶端：

var clients = make(map[*websocket.Conn]bool)

然後，我們修改主函數中的程式碼，將WebSocket處理器函數與"/ws "路徑關聯起來，並啟動一個HTTP伺服器：

func main() {
    http.HandleFunc("/ws", wsHandler)

    err := http.ListenAndServe(":8000", nil)
    if err != nil {
        log.Fatal("ListenAndServe: ", err)
    }
}

最後，我們可以使用瀏覽器或WebSocket客戶端工具來測試我們的聊天應用程式了。在瀏覽器中輸入"http://localhost:8000"，打開控制台，並執行以下JavaScript代碼：

var ws = new WebSocket("ws://localhost:8000/ws");

// 连接建立时触发的事件
ws.onopen = function() {
    console.log("连接成功");
};

// 收到消息时触发的事件
ws.onmessage = function(evt) {
    console.log("收到消息：" + evt.data);
};

// 发送消息
ws.send("Hello, World!");

透過上述程式碼，我們將會收到來自伺服器的“Hello, World!”訊息，並在控制台中輸出。

至此，我們已經成功地建立了一個簡單的聊天應用，使用Golang和WebSocket實現了即時訊息的發送和接收。當然，這只是一個基礎的範例，你可以根據自己的需求進行擴展和最佳化。希望這篇文章對你有幫助，祝你在使用Golang和WebSocket開發即時通訊應用程式時，取得更多的成功！

以上是golang WebSocket範例：建立一個簡單的聊天應用的詳細內容。更多資訊請關注PHP中文網其他相關文章！