etcd add connect time

apache · Apr 10, 2024 · 60209c4 · 60209c4
1 parent 466ada3
commit 60209c4
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 2 deletions.
diff --git a/...c/main/java/org/apache/eventmesh/dashboard/core/function/SDK/config/CreateEtcdConfig.java b/...c/main/java/org/apache/eventmesh/dashboard/core/function/SDK/config/CreateEtcdConfig.java
@@ -17,13 +17,21 @@
 
 package org.apache.eventmesh.dashboard.core.function.SDK.config;
 
+import lombok.AllArgsConstructor;
+import lombok.Builder;
 import lombok.Data;
+import lombok.NoArgsConstructor;
 
 @Data
+@Builder
+@AllArgsConstructor
+@NoArgsConstructor
 public class CreateEtcdConfig implements CreateSDKConfig {
 
     private String etcdServerAddress;
 
+    private Long connectTime;
+
     @Override
     public String getUniqueKey() {
         return etcdServerAddress;

diff --git a/...ain/java/org/apache/eventmesh/dashboard/core/function/SDK/operation/EtcdSDKOperation.java b/...ain/java/org/apache/eventmesh/dashboard/core/function/SDK/operation/EtcdSDKOperation.java
@@ -21,6 +21,7 @@
 import org.apache.eventmesh.dashboard.core.function.SDK.config.CreateEtcdConfig;
 import org.apache.eventmesh.dashboard.core.function.SDK.config.CreateSDKConfig;
 
+import java.time.Duration;
 import java.util.AbstractMap.SimpleEntry;
 
 import io.etcd.jetcd.Client;
@@ -39,6 +40,7 @@ public SimpleEntry<String, KV> createClient(CreateSDKConfig clientConfig) {
         try {
             final Client client = Client.builder()
                 .endpoints(getSplitEndpoints(etcdConfig))
+                .connectTimeout(Duration.ofSeconds(etcdConfig.getConnectTime()))
                 .build();
             kvClient = client.getKVClient();
         } catch (EtcdException e) {

diff --git a/...rg/apache/eventmesh/dashboard/core/function/SDK/operation/EtcdSDKCreateOperationTest.java b/...rg/apache/eventmesh/dashboard/core/function/SDK/operation/EtcdSDKCreateOperationTest.java
@@ -46,8 +46,11 @@ public class EtcdSDKCreateOperationTest {
 
     @Test
     void testCreateClient() {
-        final CreateEtcdConfig etcdConfig = new CreateEtcdConfig();
-        etcdConfig.setEtcdServerAddress(url);
+        final CreateEtcdConfig etcdConfig = CreateEtcdConfig.builder()
+            .etcdServerAddress(url)
+            .connectTime(10L)
+            .build();
+        // etcdConfig.setEtcdServerAddress(url);
         SimpleEntry<String, KV> simpleEntry = null;
         try {
             simpleEntry = etcdSDKOperation.createClient(etcdConfig);