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

哈喽！今天心血来潮给大家带来了《在 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()
}

正确答案

正如 selbie 所指出的，这里要做的正常事情是期望客户端通过 WebSocket 发送一条特殊的第一条消息。如果该消息未收到或不满足要求，服务器将结束 WebSocket 连接。

使用 socket.onopen 对于此任务非常常见。

更多关于为什么不能放置标题的信息： HTTP headers in Websockets client API

以上就是本文的全部内容了，是否有顺利帮助你解决问题？若是能给你带来学习上的帮助，请大家多多支持golang学习网！更多关于Golang的相关知识，也可关注golang学习网公众号。