-
Notifications
You must be signed in to change notification settings - Fork 2
/
sdk_registrar.go
49 lines (42 loc) · 1.27 KB
/
sdk_registrar.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
package sdk
import (
"github.com/configcat/configcat-proxy/config"
"github.com/configcat/configcat-proxy/diag/metrics"
"github.com/configcat/configcat-proxy/diag/status"
"github.com/configcat/configcat-proxy/log"
configcat "github.com/configcat/go-sdk/v9"
)
type Registrar interface {
GetSdkOrNil(sdkId string) Client
GetAll() map[string]Client
Close()
}
type registrar struct {
sdkClients map[string]Client
}
func NewRegistrar(conf *config.Config, metricsReporter metrics.Reporter, statusReporter status.Reporter, externalCache configcat.ConfigCache, log log.Logger) Registrar {
sdkClients := make(map[string]Client, len(conf.SDKs))
for key, sdkConf := range conf.SDKs {
sdkClients[key] = NewClient(&Context{
SDKConf: sdkConf,
MetricsReporter: metricsReporter,
StatusReporter: statusReporter,
ProxyConf: &conf.HttpProxy,
GlobalDefaultAttrs: conf.DefaultAttrs,
SdkId: key,
ExternalCache: externalCache,
}, log)
}
return ®istrar{sdkClients: sdkClients}
}
func (r *registrar) GetSdkOrNil(sdkId string) Client {
return r.sdkClients[sdkId]
}
func (r *registrar) GetAll() map[string]Client {
return r.sdkClients
}
func (r *registrar) Close() {
for _, sdkClient := range r.sdkClients {
sdkClient.Close()
}
}