feat: discov support standalone etcd

zzycn003 · Apr 20, 2022 · 9bf4ed8 · 9bf4ed8
1 parent fd33765
commit 9bf4ed8
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 0 deletions.
diff --git a/core/discov/clients.go b/core/discov/clients.go
@@ -39,3 +39,7 @@ func extractId(etcdKey string) (string, bool) {
 func makeEtcdKey(key string, id int64) string {
 	return fmt.Sprintf("%s%c%d", key, internal.Delimiter, id)
 }
+
+func MarkEtcdClientStandalone() {
+	internal.NewClient = internal.DialClientStandalone
+}
diff --git a/core/discov/internal/registry.go b/core/discov/internal/registry.go
@@ -312,6 +312,18 @@ func DialClient(endpoints []string) (EtcdClient, error) {
 	})
 }
 
+// DialClientStandalone dials an standalone etcd with given endpoionts.
+func DialClientStandalone(endpoints []string) (EtcdClient, error) {
+	return clientv3.New(clientv3.Config{
+		Endpoints:            endpoints,
+		//AutoSyncInterval:     autoSyncInterval,
+		DialTimeout:          DialTimeout,
+		DialKeepAliveTime:    dialKeepAliveTime,
+		DialKeepAliveTimeout: DialTimeout,
+		//RejectOldCluster:     true,
+	})
+}
+
 func getClusterKey(endpoints []string) string {
 	sort.Strings(endpoints)
 	return strings.Join(endpoints, endpointsSeparator)