如何在Redlang（redigo）Pubsub中更好地编写Golang中的Receive（）？

Question

psc := redis.PubSubConn{c} 
psc.Subscribe("example") 

func Receive() { 
    for { 
     switch v := psc.Receive().(type) { 
     case redis.Message: 
      fmt.Printf("%s: message: %s\n", v.Channel, v.Data) 
     case redis.Subscription: 
      fmt.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count) 
     case error: 
      return v 
     } 
    } 
} 

在上面的代码中（取自Redigo doc），如果连接丢失，所有订阅也会丢失。什么是更好的方式来恢复丢失的连接和重新订阅。

Answer 1

使用两个嵌套循环。外部循环获取连接，设置订阅，然后调用内部循环接收消息。执行内部循环，直到连接出现永久性错误。

for { 
    // Get a connection from a pool 
    c := pool.Get() 
    psc := redis.PubSubConn{c} 

    // Set up subscriptions 
    psc.Subscribe("example")) 

    // While not a permanent error on the connection. 
    for c.Err() == nil { 
     switch v := psc.Receive().(type) { 
     case redis.Message: 
      fmt.Printf("%s: message: %s\n", v.Channel, v.Data) 
     case redis.Subscription: 
      fmt.Printf("%s: %s %d\n", v.Channel, v.Kind, v.Count) 
     case error: 
      fmt.Printf(err) 
     } 
    } 
    c.Close() 
} 

本示例使用Redigo pool来获取连接。另一种方法是直接拨号连接：

c, err := redis.Dial("tcp", serverAddress)