-
Notifications
You must be signed in to change notification settings - Fork 1.5k
/
conn_pools.go
41 lines (34 loc) · 1.13 KB
/
conn_pools.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
package sender
import (
backend "github.com/open-falcon/falcon-plus/common/backend_pool"
"github.com/open-falcon/falcon-plus/modules/transfer/g"
nset "github.com/toolkits/container/set"
)
func initConnPools() {
cfg := g.Config()
// judge
judgeInstances := nset.NewStringSet()
for _, instance := range cfg.Judge.Cluster {
judgeInstances.Add(instance)
}
JudgeConnPools = backend.CreateSafeRpcConnPools(cfg.Judge.MaxConns, cfg.Judge.MaxIdle,
cfg.Judge.ConnTimeout, cfg.Judge.CallTimeout, judgeInstances.ToSlice())
// tsdb
if cfg.Tsdb.Enabled {
TsdbConnPoolHelper = backend.NewTsdbConnPoolHelper(cfg.Tsdb.Address, cfg.Tsdb.MaxConns, cfg.Tsdb.MaxIdle, cfg.Tsdb.ConnTimeout, cfg.Tsdb.CallTimeout)
}
// graph
graphInstances := nset.NewSafeSet()
for _, nitem := range cfg.Graph.ClusterList {
for _, addr := range nitem.Addrs {
graphInstances.Add(addr)
}
}
GraphConnPools = backend.CreateSafeRpcConnPools(cfg.Graph.MaxConns, cfg.Graph.MaxIdle,
cfg.Graph.ConnTimeout, cfg.Graph.CallTimeout, graphInstances.ToSlice())
}
func DestroyConnPools() {
JudgeConnPools.Destroy()
GraphConnPools.Destroy()
TsdbConnPoolHelper.Destroy()
}