40 lines
961 B
Go
40 lines
961 B
Go
package gfetcd
|
||
|
||
import (
|
||
"path"
|
||
"time"
|
||
|
||
"github.com/coreos/etcd/clientv3"
|
||
)
|
||
|
||
const (
|
||
DEFAULT_DIAL_TIMEOUT = time.Second * 5
|
||
DEFAULT_LEASE_TTL = 5
|
||
)
|
||
|
||
// 基础配置，公共字段
|
||
type BasicConfig struct {
|
||
DialTimeout int `toml:"dial_timeout"` // 连接超时时间
|
||
EndPoints []string `toml:"endpoints"` // etcd的地址
|
||
Prefix string `toml:"prefix"` // etcd服务的路径前缀，区分环境用
|
||
Debug bool `toml:"debug"` // 使用debug模式
|
||
}
|
||
|
||
func etcdDial(cfg *BasicConfig) (*clientv3.Client, error) {
|
||
dialTimeout := DEFAULT_DIAL_TIMEOUT
|
||
if cfg.DialTimeout > 0 {
|
||
dialTimeout = time.Duration(cfg.DialTimeout) * time.Second
|
||
}
|
||
|
||
return clientv3.New(clientv3.Config{
|
||
Endpoints: cfg.EndPoints,
|
||
DialTimeout: dialTimeout,
|
||
DialKeepAliveTime: 2 * time.Second,
|
||
DialKeepAliveTimeout: 2 * time.Second,
|
||
})
|
||
}
|
||
|
||
func ServiceKey(opts ...string) string {
|
||
return path.Join(opts...)
|
||
}
|