feat: add query method of jdk http client (#11)

Signed-off-by: zhiheng123 <[email protected]>
Co-authored-by: ZhangJian He <[email protected]>

Author: zhiheng123

opengemini-client-api/src/main/java/io/opengemini/client/api/Address.java

@@ -1,13 +1,17 @@
 package io.opengemini.client.api;
 
+import lombok.AllArgsConstructor;
 import lombok.Getter;
+import lombok.NoArgsConstructor;
 import lombok.Setter;
 
 /**
  * Address configuration for providing service.
  */
 @Getter
 @Setter
+@NoArgsConstructor
+@AllArgsConstructor
 public class Address {
     /**
      * Host service ip or domain.

opengemini-client-api/src/main/java/io/opengemini/client/api/BaseConfiguration.java

@@ -1,6 +1,5 @@
 package io.opengemini.client.api;
 
-
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;

opengemini-client-api/src/main/java/io/opengemini/client/api/Query.java

@@ -1,4 +1,29 @@
 package io.opengemini.client.api;
 
+import lombok.AllArgsConstructor;
+import lombok.Getter;
+import lombok.Setter;
+
+@AllArgsConstructor
+@Getter
+@Setter
 public class Query {
+    /*
+     * the query command
+     */
+    private String command;
+
+    /*
+     * the database name of the query command using
+     */
+    private String database;
+
+    /*
+     * the rp name of the query command using
+     */
+    private String retentionPolicy;
+
+    public Query(String command){
+        this.command = command;
+    };
 }

opengemini-client-api/src/main/java/io/opengemini/client/api/QueryResult.java

@@ -1,4 +1,16 @@
 package io.opengemini.client.api;
 
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+import java.util.List;
+
+@Getter
+@Setter
+@ToString
 public class QueryResult {
+    private List<SeriesResult> results;
+
+    private String error;
 }

opengemini-client-api/src/main/java/io/opengemini/client/api/Series.java

@@ -0,0 +1,21 @@
+package io.opengemini.client.api;
+
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+import java.util.List;
+import java.util.Map;
+
+@Getter
+@Setter
+@ToString
+public class Series {
+    private String name;
+
+    private Map<String, String> tags;
+
+    private List<String> columns;
+
+    private List<List<Object>> values;
+}

opengemini-client-api/src/main/java/io/opengemini/client/api/SeriesResult.java

@@ -0,0 +1,18 @@
+package io.opengemini.client.api;
+
+import com.fasterxml.jackson.annotation.JsonIgnoreProperties;
+import lombok.Getter;
+import lombok.Setter;
+import lombok.ToString;
+
+import java.util.List;
+
+@Getter
+@Setter
+@ToString
+@JsonIgnoreProperties({"statement_id"})
+public class SeriesResult {
+    private List<Series> series;
+
+    private String error;
+}

opengemini-client-api/src/main/java/io/opengemini/client/api/TlsConfig.java

@@ -1,6 +1,5 @@
 package io.opengemini.client.api;
 
-
 import lombok.AllArgsConstructor;
 import lombok.Getter;
 import lombok.NoArgsConstructor;
@@ -12,7 +11,6 @@
 @NoArgsConstructor
 @AllArgsConstructor
 public class TlsConfig {
-
     public String keyStorePath;
 
     @ToString.Exclude

opengemini-client-jdk/src/main/java/io/opengemini/client/jdk/OpenGeminiJdkClient.java

@@ -1,12 +1,25 @@
 package io.opengemini.client.jdk;
 
+import com.fasterxml.jackson.core.JsonProcessingException;
 import io.opengemini.client.api.Address;
+import io.opengemini.client.api.OpenGeminiException;
+import io.opengemini.client.api.Query;
+import io.opengemini.client.api.QueryResult;
 import io.opengemini.client.api.SslContextUtil;
 import io.opengemini.client.api.TlsConfig;
+import io.opengemini.client.jdk.common.OpenGeminiCommon;
 
+import java.net.URI;
+import java.net.URISyntaxException;
+import java.net.URLEncoder;
 import java.net.http.HttpClient;
+import java.net.http.HttpRequest;
+import java.net.http.HttpResponse;
+import java.nio.charset.StandardCharsets;
 import java.util.ArrayList;
 import java.util.List;
+import java.util.concurrent.CompletableFuture;
+import java.util.concurrent.atomic.AtomicInteger;
 
 public class OpenGeminiJdkClient {
 
@@ -16,6 +29,8 @@ public class OpenGeminiJdkClient {
 
     private final HttpClient client;
 
+    private final AtomicInteger prevIndex = new AtomicInteger(0);
+
     public OpenGeminiJdkClient(Configuration conf) {
         this.conf = conf;
         HttpClient.Builder builder = HttpClient.newBuilder()
@@ -38,4 +53,49 @@ public OpenGeminiJdkClient(Configuration conf) {
         }
         this.client = builder.build();
     }
+
+    public String encodeUtf8(String str) {
+        return URLEncoder.encode(str, StandardCharsets.UTF_8);
+    }
+
+    private URI buildUri(String url, String command) throws URISyntaxException {
+        // avoid multi-thread conflict, realize simple round-robin
+        int idx = prevIndex.addAndGet(1);
+        idx = idx % this.conf.getAddresses().size();
+
+        return buildUri(idx, url, command);
+    }
+
+    private URI buildUri(int index, String url, String command) throws URISyntaxException {
+        StringBuilder sb = new StringBuilder(this.serverUrls.get(index))
+                .append(url)
+                .append("?q=")
+                .append(encodeUtf8(command));
+
+        return new URI(sb.toString());
+    }
+
+    public CompletableFuture<QueryResult> query(Query query) throws URISyntaxException {
+        HttpRequest request = HttpRequest.newBuilder()
+                .uri(buildUri("/query", query.getCommand()))
+                .GET()
+                .build();
+
+        return client.sendAsync(request, HttpResponse.BodyHandlers.ofString()).thenCompose(response ->{
+            CompletableFuture<QueryResult> failedFuture = new CompletableFuture<>();
+            if (response.statusCode() >= 200 && response.statusCode() < 300) {
+                try {
+                    QueryResult body = OpenGeminiCommon.converJson2Bean(response.body(), QueryResult.class);
+                    failedFuture.complete(body);
+                } catch (JsonProcessingException e) {
+                    failedFuture.completeExceptionally(e);
+                }
+                return failedFuture;
+            }
+
+            failedFuture.completeExceptionally(new OpenGeminiException("http error: " + response.body(),
+                        response.statusCode()));
+            return failedFuture;
+        });
+    }
 }

opengemini-client-jdk/src/main/java/io/opengemini/client/jdk/common/OpenGeminiCommon.java

@@ -0,0 +1,11 @@
+package io.opengemini.client.jdk.common;
+
+import com.fasterxml.jackson.core.JsonProcessingException;
+import com.fasterxml.jackson.databind.ObjectMapper;
+
+public class OpenGeminiCommon {
+    private static final ObjectMapper objectMapper = new ObjectMapper();
+    public static <T> T converJson2Bean(String json, Class<T> cls) throws JsonProcessingException {
+        return objectMapper.readValue(json, cls);
+    }
+}

opengemini-client-jdk/src/main/java/io/opengemini/client/jdk/common/package-info.java

@@ -0,0 +1 @@
+package io.opengemini.client.jdk.common;