0

我正在使用聊天服务器并使用 redigo 来跨许多 Web 套接字连接发布消息。我有一个 go 例程（每个用户连接）打开到 redis 的连接，订阅一些频道并等待消息。等待这个消息 -

msg := psc.Receive()

是一个阻塞操作，因为接收函数之后的任何代码在消息进入之前都不会运行。但是我希望能够观察上下文 -

ctx.Done()

之类的。这可以使用选择来实现，还是有更惯用的写法？我正在尝试这样做，因为在当前实现中，此函数作为 go 例程运行，如果 web 会话结束，我想结束 redis 连接并完成 go 例程。

func relayRedisMessages (ctx context.Context, ws *websocket.Conn, rc []string) {

    c, err := redis.Dial("tcp", "localhost:6379")
    if err != nil {
        log.Println(err)
        return
    }
    defer c.Close()

    psc := redis.PubSubConn{c}

    for _, channel := range rc {
        psc.Subscribe(channel)
    }

    for {
        msg := psc.Receive()
        // Write message to websocket
        var msg Message
        json.Unmarshal(v.Data, &msg)
        err = ws.WriteJSON(msg)
        if err != nil {
            log.Println(err)
            break
        }
    }
}
4

0 回答 0