如何使用 Gorilla 在 Go 中向特定客户端发送 Websocket 消息？

使用 Gorilla 在 Go 中向特定客户端发送 Websocket 消息

使用 Gorilla 在 Go 中使用 Websocket 时，向单个客户端发送更新可能是一项复杂的任务。本文深入探讨了如何有效实现这一目标的细节。

问题

目标是在客户端和服务器之间建立 Websocket 连接。服务器应该检索来自客户端的输入，使用 Go 通道从索引中获取数据，然后将索引数据写回特定客户端的套接字。挑战在于在服务器代码中唯一标识客户端。

代码

hub.go 中提供的代码管理客户端连接，并包含一个连接映射，该映射最初为每个连接存储了一个布尔值客户。要向特定客户端发送消息，可以修改此映射以存储连接结构。

<code class="go">// hub.go

// ... Existing code ...

type Hub struct {
    // Registered clients.
    clients map[*Client]bool

    // Inbound messages from the clients.
    broadcast chan []byte

    // Register requests from the clients.
    register chan *Client

    // Unregister requests from clients.
    unregister chan *Client

    connections map[string]*connection
}

// ... Existing code ...</code>

在 client.go 中，可以修改连接字段以存储客户端的唯一 ID。

<code class="go">// client.go

// ... Existing code ...

type Client struct {
    // unique ID for each client
    ID string

    // Hub object
    hub *Hub

    // The websocket connection.
    conn *websocket.Conn

    // Buffered channel of outbound messages.
    send chan []byte
}

// ... Existing code ...</code>

Hub 修改

要向特定客户端发送消息，需要修改 Hub 代码以包含一个发送通道，该通道接受包含消息数据和目标 ID 的消息客户端。

<code class="go">// hub.go

// ... Existing code ...

type Hub struct {
    // ... Existing fields ...

    // Send channel for specific clients.
    send chan message
}

// ... Existing code ...</code>

向客户端发送消息

现在可以使用中心的发送通道将消息发送给特定客户端。

<code class="go">// ...
// Send a message to a client with ID "targetID".
hub.send <- message{ID: "targetID", data: messageData}
// ...</code>

结论

按照上述步骤，在 Go 中使用 Gorilla 时可以向特定客户端发送 websocket 消息。然而，在实现此模式之前考虑广播的需求至关重要，因为 Gorilla 存储库中提供的聊天示例不适合没有广播需求的应用程序。

以上是如何使用 Gorilla 在 Go 中向特定客户端发送 Websocket 消息？的详细内容。更多信息请关注PHP中文网其他相关文章！