-
Notifications
You must be signed in to change notification settings - Fork 3
/
setup.go
81 lines (71 loc) · 1.8 KB
/
setup.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
69
70
71
72
73
74
75
76
77
78
79
80
81
package etcdKit
import (
"github.com/richelieu-yang/chimera/v2/src/core/fileKit"
"github.com/richelieu-yang/chimera/v2/src/core/strKit"
"github.com/richelieu-yang/chimera/v2/src/log/logrusKit"
"github.com/richelieu-yang/chimera/v2/src/log/zapKit"
"github.com/sirupsen/logrus"
clientv3 "go.etcd.io/etcd/client/v3"
"go.uber.org/zap"
"io"
"sync"
"time"
)
var client *clientv3.Client
var setupOnce sync.Once
func MustSetUp(config *Config) {
if err := setUp(config); err != nil {
logrusKit.DisableQuote(nil)
logrus.Fatalf("%+v", err)
}
}
// setUp
/*
TODO: 可以参考 go-zero 中 registry.go 的 internal.DialClient.
PS:
(1) 如果 Endpoints 无效,会返回error(context.DeadlineExceeded).
*/
func setUp(config *Config) (err error) {
if err = config.Check(); err != nil {
return
}
setupOnce.Do(func() {
/* etcd客户端日志输出 */
var logger *zap.Logger
if strKit.IsNotEmpty(config.LogPath) {
var writer io.Writer
writer, err = fileKit.NewFileInAppendMode(config.LogPath)
if err != nil {
return
}
logger, err = zapKit.NewLogger(writer, zap.InfoLevel)
if err != nil {
return
}
}
v3Config := clientv3.Config{
Endpoints: config.Endpoints,
Logger: logger,
AutoSyncInterval: time.Minute,
DialTimeout: time.Second * 5,
DialKeepAliveTime: time.Second * 5,
DialKeepAliveTimeout: time.Second * 5,
RejectOldCluster: true,
PermitWithoutStream: true,
}
client, err = clientv3.New(v3Config)
})
return
}
// GetClient
/*
PS:
(1) 要使用 KV 的情况下,建议调用 clientv3.NewKV() 以实例化一个用于操作etcd的KV(内置错误重试机制).
(2) 租约相关需要用到 *clientv3.Client实例.
*/
func GetClient() (*clientv3.Client, error) {
if client == nil {
return nil, NotSetupError
}
return client, nil
}