Завершение процесса redigo - PullRequest
Новая
       34

Завершение процесса redigo

0 голосов
/ 26 апреля 2018

Я работаю на сервере чата и использую redigo для публикации сообщений через множество подключений через веб-сокеты. У меня есть подпрограмма go, которая (для каждого пользователя) открывает соединение с Redis, подписывается на некоторые каналы и ждет сообщения. Ожидание этого сообщения - 

msg := psc.Receive()

является блокирующей операцией, поскольку любой код после функции приема не будет работать до тех пор, пока не поступит сообщение. Однако я хотел бы иметь возможность отслеживать контекст - 

ctx.Done()

или тому подобное. Это достижимо, используя select, или есть более идиоматический способ написания этого? Я пытаюсь сделать это, так как в текущей реализации эта функция запускается как процедура go, и если веб-сеанс завершается, я хотел бы завершить соединение 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
        }
    }
}
...