当然可以,以下是一个简单的WebSocket聊天服务的示例:

package main

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

    "github.com/gorilla/websocket"
)

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

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

func main() {
    http.HandleFunc("/ws", handleWebSocket)
    log.Fatal(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
    }

    clients[conn] = true

    for {
        messageType, p, err := conn.ReadMessage()
        if err != nil {
            delete(clients, conn)
            log.Println(err)
            return
        }

        message := string(p)
        fmt.Printf("Received message: %s\n", message)

        for client := range clients {
            if err := client.WriteMessage(messageType, p); err != nil {
                log.Println(err)
                delete(clients, client)
            }
        }
    }
}

这个示例使用了Gorilla WebSocket库,它提供了更方便的WebSocket处理功能。在这个示例中,我们定义了一个全局变量clients,用于存储连接到服务器的所有客户端连接。在handleWebSocket函数中,我们将新连接添加到clients中,并使用一个无限循环来处理从客户端发送的消息。每当接收到一个消息时,我们将其广播到所有其他客户端。

你可以使用以下命令在本地运行这个示例:

go run main.go

然后在浏览器中打开http://localhost:8080,你就可以测试这个WebSocket聊天服务了。

能使用golang帮我写一个简单的ws聊天服务吗

原文地址: https://www.cveoy.top/t/topic/gT3 著作权归作者所有。请勿转载和采集!

免费AI点我,无需注册和登录