当前位置：首页 > 文章列表 > Golang > Go问答 > 在 Websocket 握手请求中包含客户端信息

在 Websocket 握手请求中包含客户端信息

来源：stackoverflow 2024-02-28 11:27:23 0浏览收藏

哈喽！今天心血来潮给大家带来了《在 Websocket 握手请求中包含客户端信息》，想必大家应该对Golang都不陌生吧，那么阅读本文就都不会很困难，以下内容主要涉及到，若是你正在学习Golang，千万别错过这篇文章~希望能帮助到你！

问题内容

我目前正在创建一个分布式聊天应用程序。一切工作正常，这意味着可以在客户端和服务器之间发送消息并适当地广播消息。

但是，目前仅将实际消息发送到服务器。我还想添加有关发送消息的用户的信息。

每当我发送新消息时，我都可以添加此信息，但我更希望能够在初始握手期间添加此信息，然后将此信息保存在后端。

我考虑过在 url 中发送一些信息，但由于我只实例化一次 websocket，这似乎不是可行的方法。同样，我考虑过将信息添加为请求的正文，但我了解到通常不建议在 get 请求上添加正文。

所以我的问题是，我是否在尝试做一些我不应该做的事情？我是否应该在发送到服务器的每条新消息上发送有关客户端的信息？

目前，我的客户端如下所示：

const socket = new websocket("ws://localhost:8080/ws");

const connect = (cb) => {
    console.log("attempting connection...")

    socket.onopen = () => {
        console.log("successfully connected");
    }

    socket.onmessage = (msg) => {
        console.log(msg)
        cb(msg);
    }

    socket.onclose = (event) => {
        console.log("socket closed connection: ", event);
    }

    socket.onerror = (error) => {
        console.log("socket error: ", error);
    }
};

const sendmsg = (msg) => {
    console.log("sending msg: ", msg);
    socket.send(msg);
}

后端的初始连接由以下方式处理：

func serveWs(pool *websocket.Pool, w http.ResponseWriter, r *http.Request) {
    fmt.Println("WebSocket Endpoint Hit")

    conn, err := websocket.Upgrade(w, r)

    if err != nil {
        fmt.Fprintf(w, "%+V\n", err)
    }

    client := &websocket.Client{
        Name: "?????", // Obviously, I would like the actual name to be here.
        Conn: conn,
        Pool: pool,
    }

    pool.Register <- client

    client.Read()
}