mirror of
https://gitee.com/milvus-io/milvus.git
synced 2024-12-03 04:19:18 +08:00
c9d0c157ec
Signed-off-by: jaime <yun.zhang@zilliz.com>
77 lines
1.6 KiB
Go
77 lines
1.6 KiB
Go
package etcd
|
|
|
|
import (
|
|
"sync"
|
|
|
|
clientv3 "go.etcd.io/etcd/client/v3"
|
|
"go.etcd.io/etcd/server/v3/embed"
|
|
"go.etcd.io/etcd/server/v3/etcdserver/api/v3client"
|
|
"go.uber.org/zap"
|
|
|
|
"github.com/milvus-io/milvus/pkg/log"
|
|
)
|
|
|
|
// EtcdServer is the singleton of embedded etcd server
|
|
var (
|
|
initOnce sync.Once
|
|
closeOnce sync.Once
|
|
etcdServer *embed.Etcd
|
|
)
|
|
|
|
// GetEmbedEtcdClient returns client of embed etcd server
|
|
func GetEmbedEtcdClient() (*clientv3.Client, error) {
|
|
client := v3client.New(etcdServer.Server)
|
|
return client, nil
|
|
}
|
|
|
|
// InitEtcdServer initializes embedded etcd server singleton.
|
|
func InitEtcdServer(
|
|
useEmbedEtcd bool,
|
|
configPath string,
|
|
dataDir string,
|
|
logPath string,
|
|
logLevel string,
|
|
) error {
|
|
if useEmbedEtcd {
|
|
var initError error
|
|
initOnce.Do(func() {
|
|
path := configPath
|
|
var cfg *embed.Config
|
|
if len(path) > 0 {
|
|
cfgFromFile, err := embed.ConfigFromFile(path)
|
|
if err != nil {
|
|
initError = err
|
|
}
|
|
cfg = cfgFromFile
|
|
} else {
|
|
cfg = embed.NewConfig()
|
|
}
|
|
cfg.Dir = dataDir
|
|
cfg.LogOutputs = []string{logPath}
|
|
cfg.LogLevel = logLevel
|
|
e, err := embed.StartEtcd(cfg)
|
|
if err != nil {
|
|
log.Error("failed to init embedded Etcd server", zap.Error(err))
|
|
initError = err
|
|
}
|
|
etcdServer = e
|
|
log.Info("finish init Etcd config", zap.String("path", path), zap.String("data", dataDir))
|
|
})
|
|
return initError
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func HasServer() bool {
|
|
return etcdServer != nil
|
|
}
|
|
|
|
// StopEtcdServer stops embedded etcd server singleton.
|
|
func StopEtcdServer() {
|
|
if etcdServer != nil {
|
|
closeOnce.Do(func() {
|
|
etcdServer.Close()
|
|
})
|
|
}
|
|
}
|