feat: add test compress

Signed-off-by: kaixuan xu <[email protected]>
openGemini · Jan 18, 2025 · e0ea955 · e0ea955
1 parent b2d6ed2
commit e0ea955
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 7 deletions.
diff --git a/opengemini-client/src/main/java/io/opengemini/client/impl/OpenGeminiClient.java b/opengemini-client/src/main/java/io/opengemini/client/impl/OpenGeminiClient.java
@@ -21,7 +21,15 @@
 import io.github.openfacade.http.HttpClientConfig;
 import io.github.openfacade.http.HttpClientFactory;
 import io.github.openfacade.http.HttpResponse;
-import io.opengemini.client.api.*;
+import io.opengemini.client.api.AuthConfig;
+import io.opengemini.client.api.AuthType;
+import io.opengemini.client.api.CompressMethod;
+import io.opengemini.client.api.Configuration;
+import io.opengemini.client.api.ContentType;
+import io.opengemini.client.api.OpenGeminiException;
+import io.opengemini.client.api.Pong;
+import io.opengemini.client.api.Query;
+import io.opengemini.client.api.QueryResult;
 import io.opengemini.client.common.BaseAsyncClient;
 import io.opengemini.client.common.HeaderConst;
 import io.opengemini.client.common.JacksonService;
@@ -119,14 +127,16 @@ protected CompletableFuture<Pong> executePing() {
     }
 
     private <T> T processResponseBody(HttpResponse response, Class<T> type) throws IOException {
-        String contentType = response.headers().get("Content-Type") != null ? response.headers().get("Content-Type").get(0) : null;
-        String contentEncoding = response.headers().get("Content-Encoding") != null ? response.headers().get("Content-Encoding").get(0) : null;
+        String contentType = response.headers().get("Content-Type") != null
+                ? response.headers().get("Content-Type").get(0) : null;
+        String contentEncoding = response.headers().get("Content-Encoding") != null
+                ? response.headers().get("Content-Encoding").get(0) : null;
         byte[] body = processCompression(contentEncoding, response.body(), type);
 
        return processContentType(contentType, body, type);
     }
 
-    private <T>  byte[] processCompression(String compressMethod, byte[] body,  Class<T> type ) throws IOException {
+    private <T>  byte[] processCompression(String compressMethod, byte[] body,  Class<T> type) throws IOException {
         byte[] decompressedBody = null;
         if (CompressMethod.GZIP.getValue().equals(compressMethod)) {
              GzipCompressor compressor = new GzipCompressor();

diff --git a/opengemini-client/src/test/java/io/opengemini/client/impl/OpenGeminiClientWriteTest.java b/opengemini-client/src/test/java/io/opengemini/client/impl/OpenGeminiClientWriteTest.java
@@ -18,7 +18,15 @@
 
 import io.github.openfacade.http.HttpClientConfig;
 import io.github.openfacade.http.HttpClientEngine;
-import io.opengemini.client.api.*;
+import io.opengemini.client.api.Address;
+import io.opengemini.client.api.CompressMethod;
+import io.opengemini.client.api.Configuration;
+import io.opengemini.client.api.OpenGeminiException;
+import io.opengemini.client.api.Point;
+import io.opengemini.client.api.Query;
+import io.opengemini.client.api.QueryResult;
+import io.opengemini.client.api.RpConfig;
+import io.opengemini.client.api.Series;
 import org.junit.jupiter.api.AfterAll;
 import org.junit.jupiter.api.Assertions;
 import org.junit.jupiter.api.TestInstance;
@@ -58,7 +66,8 @@ protected List<OpenGeminiClient> clientList() throws OpenGeminiException {
             clients.add(OpenGeminiClientFactory.create(configuration));
         }
 
-        List<CompressMethod> compressMethods = Arrays.asList(CompressMethod.GZIP, CompressMethod.ZSTD, CompressMethod.SNAPPY);
+        List<CompressMethod> compressMethods = Arrays.asList(CompressMethod.GZIP,
+                CompressMethod.ZSTD, CompressMethod.SNAPPY);
         for (CompressMethod compressMethod : compressMethods) {
             HttpClientConfig httpConfig = new HttpClientConfig.Builder()
                     .engine(HttpClientEngine.AsyncHttpClient)
@@ -117,7 +126,8 @@ void write_point_with_more_fields(OpenGeminiClient client) throws Exception {
         writeRsp.get();
         Thread.sleep(3000);
 
-        Query selectQuery = new Query("select * from " + measurementName, databaseName, "");
+        Query selectQuery = new Query("select * from " + measurementName,
+                databaseName, "");
         CompletableFuture<QueryResult> rst = client.query(selectQuery);
         QueryResult queryResult = rst.get();