/
cfg.go
58 lines (47 loc) · 1.2 KB
/
cfg.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
package proxy
import (
"encoding/json"
"flag"
"io/ioutil"
"github.com/fagongzi/log"
)
var (
cfgFile = flag.String("cfg", "./cfg.json", "Configuration file of cell kv proxy, base on json formart.")
)
// Cfg returns cfg for proxy
type Cfg struct {
Addr string `json:"addr"`
AddrNotify string `json:"addrNotify"`
WatcherHeartbeatSec int `json:"watcherHeartbeatSec"`
PDAddrs []string `json:"pdAddrs"`
MaxRetries int `json:"maxRetries"`
RetryDuration int64 `json:"retryDuration"`
WorkerCount uint64 `json:"workerCount"`
SupportCMDs []string `json:"supportCMDs"`
}
// GetCfg get cfg from command
func GetCfg() *Cfg {
data, err := ioutil.ReadFile(*cfgFile)
if err != nil {
log.Fatalf("bootstrap: read configuration file failure, cfg=<%s>, errors:\n %+v",
*cfgFile,
err)
return nil
}
cfg, err := unmarshal(data)
if err != nil {
log.Fatalf("bootstrap: parse configuration file failure, cfg=<%s>, errors:\n %+v",
*cfgFile,
err)
return nil
}
return cfg
}
func unmarshal(data []byte) (*Cfg, error) {
v := &Cfg{}
err := json.Unmarshal(data, v)
if nil != err {
return nil, err
}
return v, nil
}