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

学习Golang要努力，但是不要急！今天的这篇文章《在 Go 中向特定客户端发送 Websocket 消息（使用 Gorilla）》将会介绍到golang等等知识点，如果你想深入学习Golang，可以关注我！我会持续更新相关文章的，希望对大家都能有所帮助！

问题内容

我对 Go 很陌生，发现自己使用套接字作为我的第一个项目。这是一个多余的问题，但我无法理解如何将 websocket 更新发送到 Go 中的特定客户端（使用 Gorilla）。

我要解决的广泛问题是 - 使用 websockets 和像 ES/Lucene 这样的搜索引擎构建预输入。我在我的搜索引擎上维护了一堆索引，并且围绕它有一个 Go 包装器。当我开始在 Go 中使用 websockets 时，我发现了几乎所有显示广播机制的示例。当我试图深入研究这一点并尝试根据此线程和此[答案中给出的示例修改](https://stackoverflow.com/questions/31598147/how- to-send-to-only-one-client-and-not-all-clients-using-go-and-gorilla- websocke)Gorilla的 github 存储库中给出的示例时，我似乎不明白以及它如何适合[](https://stackoverflow.com/questions/31598147/how- to-send-to-only-one-client-and-not-all-clients-using-go-and-gorilla- websocke)connections``client.go

理想情况下，我希望看到这项工作的方式是 -

• 建立客户端和服务器之间的套接字连接
• 在客户端通过套接字发送输入时，服务器获取它并放入通道（Go 通道）
• 索引包装器检查这个通道，一旦有东西要获取，索引就会被检索并写回套接字

服务器如何唯一标识Client?

我已经使用了 Gorilla 的 Github repo中给出的示例

从我的代码库hub.go中有以下内容

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
}

func newHub() *Hub {
    return &Hub{
        broadcast:  make(chan []byte),
        register:   make(chan *Client),
        unregister: make(chan *Client),
        clients:    make(map[*Client]bool),
        connection: make(map[*Client]bool), // is this alright?
    }
}

func (h *Hub) run() {
    for {
        select {
        case client := 
<p>我不确定我应该添加什么<code>client.go</code></p>
<pre class="brush:go;toolbar:false">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

    // connection --> (what should the connection property be?)
    connection string
}

请注意 - 我将Id在结构中添加一个字段Client。我该如何从这里开始？

正确答案

聊天示例展示了如何实现广播。如果不需要广播，则聊天示例不是应用程序的良好起点。

要将消息发送到特定的 websocket 连接，只需使用NextWriter或WriteMessage写入连接。这些方法不支持并发写入器，因此您可能需要使用互斥锁或 goroutine 来确保单个写入器。

查找特定对象的简单方法*websocket.Connection是将其传递*websocket.Connection给需要它的代码。如果应用程序需要将其他状态与连接相关联，则定义一个类型来保存该状态并传递一个指向该状态的指针：

type Client struct {
    conn *websocket.Conn
    mu sync.Mutex
    ...
}

Hub可以修改为向特定连接发送消息，但如果不需要广播，这是一个迂回的路径。这是如何做到的：

向客户端添加 ID 字段：

 ID idType // replace idType with int, string, or whatever you want to use

将 Gorilla hub 字段从 更改connections map[*connection]bool为connections map[idType]*connection。

定义一个包含消息数据和目标客户端 ID 的消息类型：

type message struct {
   ID idtype
   data []byte
}

将集线器广播字段替换为：

   send chan message

将集线器 for 循环更改为：

for {
    select {
    case client := 
<p><code>message</code>通过使用适当的 ID创建一个向特定客户端发送消息：</p>
<pre class="brush:go;toolbar:false">   hub.send <p>以上就是《在 Go 中向特定客户端发送 Websocket 消息（使用 Gorilla）》的详细内容，更多关于golang的资料请关注golang学习网公众号！</p>