30 lines
360 B
Go
30 lines
360 B
Go
|
package redis
|
||
|
|
|
||
|
|
import (
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"github.com/gomodule/redigo/redis"
|
||
|
|
)
|
||
|
|
|
||
|
|
type pool struct {
|
||
|
|
sync.Mutex
|
||
|
|
i int
|
||
|
|
addrs []string
|
||
|
|
}
|
||
|
|
|
||
|
|
func (p *pool) Get() redis.Conn {
|
||
|
|
for i := 0; i < 3; i++ {
|
||
|
|
p.Lock()
|
||
|
|
addr := p.addrs[p.i%len(p.addrs)]
|
||
|
|
p.i++
|
||
|
|
p.Unlock()
|
||
|
|
|
||
|
|
c, err := redis.Dial("tcp", addr)
|
||
|
|
if err != nil {
|
||
|
|
continue
|
||
|
|
}
|
||
|
|
return c
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|