/
adapterfactory.go
68 lines (52 loc) · 1.39 KB
/
adapterfactory.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
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
package server
import (
"net"
"strconv"
"github.com/go-redis/redis/v7"
"github.com/juju/errors"
"github.com/peer-calls/peer-calls/server/identifiers"
"github.com/peer-calls/peer-calls/server/logger"
)
type AdapterFactory struct {
pubClient *redis.Client
subClient *redis.Client
NewAdapter func(room identifiers.RoomID) Adapter
}
func NewAdapterFactory(log logger.Logger, c StoreConfig) *AdapterFactory {
log = log.WithNamespaceAppended("adapterfactory")
f := AdapterFactory{}
switch c.Type {
case StoreTypeRedis:
addr := net.JoinHostPort(c.Redis.Host, strconv.Itoa(c.Redis.Port))
prefix := c.Redis.Prefix
log.Info("Using RedisAdapter", logger.Ctx{
"remote_addr": addr,
"prefix": prefix,
})
f.pubClient = redis.NewClient(&redis.Options{
Addr: addr,
})
f.subClient = redis.NewClient(&redis.Options{
Addr: addr,
})
f.NewAdapter = func(room identifiers.RoomID) Adapter {
return NewRedisAdapter(log, f.pubClient, f.subClient, prefix, room)
}
default:
log.Info("Using MemoryAdapter", nil)
f.NewAdapter = func(room identifiers.RoomID) Adapter {
return NewMemoryAdapter(room)
}
}
return &f
}
func (a *AdapterFactory) Close() (err error) {
var errs MultiErrorHandler
if a.pubClient != nil {
errs.Add(errors.Trace(a.pubClient.Close()))
}
if a.subClient != nil {
errs.Add(errors.Trace(a.subClient.Close()))
}
return errors.Trace(errs.Err())
}