feat: add test for compress

openGemini · Jan 18, 2025 · 8bf081c · 8bf081c
1 parent f6ca445
commit 8bf081c
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 24 deletions.
diff --git a/opengemini-client-common/src/main/java/io/opengemini/client/common/BaseClient.java b/opengemini-client-common/src/main/java/io/opengemini/client/common/BaseClient.java
@@ -81,31 +81,33 @@ public BaseClient(Configuration conf) {
     }
 
     private void applyCodec(Configuration config, Map<String, List<String>> headers) {
-
-        List<String> acceptHeader = new ArrayList<>();
-        switch (config.getContentType()) {
-            case MSGPACK:
-                acceptHeader.add("application/msgpack");
-                break;
-            case JSON:
-                acceptHeader.add("application/json");
-                break;
-        }
+        if (config.getContentType() != null) {
+            List<String> acceptHeader = new ArrayList<>();
+            switch (config.getContentType()) {
+                case MSGPACK:
+                    acceptHeader.add("application/msgpack");
+                    break;
+                case JSON:
+                    acceptHeader.add("application/json");
+                    break;
+            }
         headers.put("Accept", acceptHeader);
-
-        List<String> acceptEncodingHeader = new ArrayList<>();
-        switch (config.getCompressMethod()) {
-            case GZIP:
-                acceptEncodingHeader.add("gzip");
-                break;
-            case ZSTD:
-                acceptEncodingHeader.add("zstd");
-                break;
-            case SNAPPY:
-                acceptEncodingHeader.add("snappy");
-                break;
         }
-        headers.put("Accept-Encoding", acceptEncodingHeader);
+        if(config.getCompressMethod() != null) {
+            List<String> acceptEncodingHeader = new ArrayList<>();
+            switch (config.getCompressMethod()) {
+                case GZIP:
+                    acceptEncodingHeader.add("gzip");
+                    break;
+                case ZSTD:
+                    acceptEncodingHeader.add("zstd");
+                    break;
+                case SNAPPY:
+                    acceptEncodingHeader.add("snappy");
+                    break;
+            }
+            headers.put("Accept-Encoding", acceptEncodingHeader);
+        }
     }
 
     /**

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
@@ -16,7 +16,6 @@
 
 package io.opengemini.client.impl;
 
-import com.sun.org.apache.bcel.internal.generic.RETURN;
 import io.github.openfacade.http.BasicAuthRequestFilter;
 import io.github.openfacade.http.HttpClient;
 import io.github.openfacade.http.HttpClientConfig;