func (c *Cluster) newPool(addr string, clearThrottle bool) (*pool.Pool, error) {
if clearThrottle {
delete(c.poolThrottles, addr)
} else if throttle, ok := c.poolThrottles[addr]; ok {
select {
case <-throttle:
delete(c.poolThrottles, addr)
default:
return nil, fmt.Errorf("newPool(%s) throttled", addr)
}
}
df := func(network, addr string) (*redis.Client, error) {
return redis.DialTimeout(network, addr, c.o.Timeout)
}
p, err := pool.NewCustom("tcp", addr, c.o.PoolSize, df)
if err != nil {
c.poolThrottles[addr] = time.After(c.o.PoolThrottle)
return nil, err
}
return p, err
}
func TestPSubscribe(t *testing.T) {
pub, err := redis.DialTimeout("tcp", "localhost:6379", time.Duration(10)*time.Second)
if err != nil {
t.Fatal(err)
}
client, err := redis.DialTimeout("tcp", "localhost:6379", time.Duration(10)*time.Second)
if err != nil {
t.Fatal(err)
}
sub := NewSubClient(client)
pattern := "patternThen*"
message := "Hello, World!"
sr := sub.PSubscribe(pattern)
if sr.Err != nil {
t.Fatal(sr.Err)
}
if sr.Type != Subscribe {
t.Fatal("Did not receive a subscribe reply")
}
if sr.SubCount != 1 {
t.Fatal(fmt.Sprintf("Unexpected subscription count, Expected: 0, Found: %d", sr.SubCount))
}
r := pub.Cmd("PUBLISH", "patternThenHello", message)
if r.Err != nil {
t.Fatal(r.Err)
}
subChan := make(chan *SubResp)
go func() {
subChan <- sub.Receive()
}()
select {
case sr = <-subChan:
case <-time.After(time.Duration(10) * time.Second):
t.Fatal("Took too long to Receive message")
}
if sr.Err != nil {
t.Fatal(sr.Err)
}
if sr.Type != Message {
t.Fatal("Did not receive a message reply")
}
if sr.Pattern != pattern {
t.Fatal(fmt.Sprintf("Did not recieve expected pattern '%s', instead got: '%s'", pattern, sr.Pattern))
}
if sr.Message != message {
t.Fatal(fmt.Sprintf("Did not recieve expected message '%s', instead got: '%s'", message, sr.Message))
}
sr = sub.PUnsubscribe(pattern)
if sr.Err != nil {
t.Fatal(sr.Err)
}
if sr.Type != Unsubscribe {
t.Fatal("Did not receive a unsubscribe reply")
}
if sr.SubCount != 0 {
t.Fatal(fmt.Sprintf("Unexpected subscription count, Expected: 0, Found: %d", sr.SubCount))
}
}