forked from letsfire/redigo
/
alone.go
44 lines (37 loc) · 876 Bytes
/
alone.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
package alone
import (
"github.com/gomodule/redigo/redis"
"github.com/letsfire/redigo/mode"
)
type standAloneMode struct {
pool *redis.Pool
}
func (sam *standAloneMode) GetConn() redis.Conn {
return sam.pool.Get()
}
func (sam *standAloneMode) NewConn() (redis.Conn, error) {
return sam.pool.Dial()
}
func (sam *standAloneMode) String() string {
return "alone"
}
var _ mode.IMode = &standAloneMode{}
func New(optFuncs ...OptFunc) *standAloneMode {
opts := options{
addr: "127.0.0.1:6379",
dialOpts: mode.DefaultDialOpts(),
poolOpts: mode.DefaultPoolOpts(),
}
for _, optFunc := range optFuncs {
optFunc(&opts)
}
pool := &redis.Pool{
Dial: func() (conn redis.Conn, e error) {
return redis.Dial("tcp", opts.addr, opts.dialOpts...)
},
}
for _, poolOptFunc := range opts.poolOpts {
poolOptFunc(pool)
}
return &standAloneMode{pool: pool}
}