Milvus Client V2

pom.xml

@@ -77,7 +77,7 @@
         <slf4j.api.version>1.7.30</slf4j.api.version>
         <log4j.slf4j.version>2.17.1</log4j.slf4j.version>
         <junit.version>4.13.2</junit.version>
-        <junit.jupiter.version>5.7.0</junit.jupiter.version>
+        <junit.jupiter.version>5.10.1</junit.jupiter.version>
         <lombok.version>1.18.22</lombok.version>
         <okhttp.version>4.10.0</okhttp.version>
         <plexus.version>3.0.24</plexus.version>
@@ -89,11 +89,12 @@
         <maven.enforcer.plugin.version>3.0.0-M2</maven.enforcer.plugin.version>
         <maven.surefire.plugin.version>2.19.1</maven.surefire.plugin.version>
         <junit.platform.version>1.1.0</junit.platform.version>
-        <junit.jupiter.engine.version>5.1.0</junit.jupiter.engine.version>
+        <junit.jupiter.engine.version>5.10.1</junit.jupiter.engine.version>
         <jackson.version>2.12.7.1</jackson.version>
         <gson.version>2.10.1</gson.version>
         <kotlin.version>1.6.20</kotlin.version>
         <version.fastjson>1.2.83</version.fastjson>
+        <mockito.version>5.8.0</mockito.version>
     </properties>
 
     <dependencyManagement>
@@ -109,6 +110,18 @@
     </dependencyManagement>
 
     <dependencies>
+        <dependency>
+            <groupId>org.mockito</groupId>
+            <artifactId>mockito-core</artifactId>
+            <version>${mockito.version}</version>
+            <scope>test</scope>
+        </dependency>
+        <dependency>
+            <groupId>org.mockito</groupId>
+            <artifactId>mockito-junit-jupiter</artifactId>
+            <version>${mockito.version}</version>
+            <scope>test</scope>
+        </dependency>
         <dependency>
             <groupId>io.grpc</groupId>
             <artifactId>grpc-netty-shaded</artifactId>

src/main/java/io/milvus/v2/client/ConnectConfig.java

@@ -0,0 +1,60 @@
+package io.milvus.v2.client;
+
+import lombok.Builder;
+import lombok.Data;
+import lombok.experimental.SuperBuilder;
+
+import java.net.URI;
+import java.nio.charset.StandardCharsets;
+import java.util.Base64;
+import java.util.concurrent.TimeUnit;
+
+@Data
+@SuperBuilder
+public class ConnectConfig {
+    private String uri;
+    private String token;
+    private String username;
+    private String password;
+    private String databaseName;
+    @Builder.Default
+    private long connectTimeoutMs = 10000;
+    @Builder.Default
+    private long keepAliveTimeMs = 55000;
+    @Builder.Default
+    private long keepAliveTimeoutMs = 20000;
+    @Builder.Default
+    private boolean keepAliveWithoutCalls = false;
+    @Builder.Default
+    private long rpcDeadlineMs = 0; // Disabling deadline
+
+    private String clientKeyPath;
+    private String clientPemPath;
+    private String caPemPath;
+    private String serverPemPath;
+    private String serverName;
+
+    @Builder.Default
+    private boolean secure = true;
+    @Builder.Default
+    private long idleTimeoutMs = TimeUnit.MILLISECONDS.convert(24, TimeUnit.HOURS);
+
+    public String getHost() {
+        URI uri = URI.create(this.uri);
+        return uri.getHost();
+    }
+
+    public int getPort() {
+        URI uri = URI.create(this.uri);
+        return uri.getPort();
+    }
+
+    public String getAuthorization() {
+        if (token != null) {
+            return token;
+        }else if (username != null && password != null) {
+            return username + ":" + password;
+        }
+        return null;
+    }
+}