当尝试向一个通道发送数据同时处理另一个通道时，Gorilla Websocket 示例会挂起吗？

来到golang学习网的大家，相信都是编程学习爱好者，希望在这里学习Golang相关编程知识。下面本篇文章就来带大家聊聊《当尝试向一个通道发送数据同时处理另一个通道时，Gorilla Websocket 示例会挂起吗？》，介绍一下，希望对大家的知识积累有所帮助，助力实战开发！

我正在关注 gorilla websocket 库的聊天客户端/服务器示例。

https://github.com/gorilla/websocket/blob/master/examples/chat/hub.go#l36

我尝试修改代码以在新客户端连接时通知其他客户端，如下所示：

for {
        select {
        case client := <-h.register:
            h.clients[client] = true

            // My addition. Hangs after this (no further register/unregister events are processed):
            h.broadcast <- []byte("Another client connected!")
        case client := <-h.unregister:
            if _, ok := h.clients[client]; ok {
                delete(h.clients, client)
                close(client.send)
            }
        case message := <-h.broadcast:
            for client := range h.clients {
                select {
                case client.send <- message:
                default:
                    close(client.send)
                    delete(h.clients, client)
                }
            }
        }
    }
}

我的理解是，在外部 for 循环的下一次迭代中，广播通道应该接收该数据并遵循 message 情况中的逻辑，但它只是挂起。

解决方案

集线器的 broadcast 通道未缓冲。无缓冲通道上的通信等待就绪的发送者和就绪的接收者。 hub goroutine 会阻塞，因为该 goroutine 无法同时准备好发送和接收。

将通道从无缓冲通道更改为缓冲通道并不能解决问题。考虑缓冲区容量为1的情况：

return &hub{
    broadcast:  make(chan []byte, 1),
    ...
}

按照这个时间表：

1 clienta: client.hub.register <- client 
2 clientb: c.hub.broadcast <- message
3 hub:     case client := <-h.register:
4 hub:     h.broadcast <- []byte("another client connected!")

集线器在#4处阻塞，因为#2处的通道已满。将通道容量增加到两个或更多并不能解决问题，因为任意数量的客户端都可以在另一个客户端注册时广播消息。

要解决此问题，请将广播代码移至一个函数，然后从 select 中的两种情况调用该函数：

// sendAll sends message to all registered clients.
// This method must only be called by Hub.run.
func (h *Hub) sendAll(message []byte) {
    for client := range h.clients {
        select {
        case client.send <- message:
        default:
            close(client.send)
            delete(h.clients, client)
        }
    }
}

func (h *Hub) run() {
    for {
        select {
        case client := <-h.register:
            h.clients[client] = true
            h.sendAll([]byte("Another client connected!"))
        case client := <-h.unregister:
            if _, ok := h.clients[client]; ok {
                delete(h.clients, client)
                close(client.send)
            }
        case message := <-h.broadcast:
            h.sendAll(message)
        }
    }
}

你的通道是无缓冲的，这意味着每个读/写都会阻塞，直到另一个 goroutine 在同一通道上执行相反的操作。

当您尝试写入 h.broadcast 时，goroutine 会停止，等待读取器。但是同一个 goroutine 应该充当该通道的读取器，但这种情况永远不会发生，因为 goroutine 被写入阻塞了。从而导致程序死锁。

本篇关于《当尝试向一个通道发送数据同时处理另一个通道时，Gorilla Websocket 示例会挂起吗？》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！