Przeglądaj źródła

set default options if call Cloud (#1345)

Signed-off-by: lentitude2tk <xushuang.hu@zilliz.com>
xushuang.hu 1 miesiąc temu
rodzic
commit
d1c94b30d6

+ 11 - 0
sdk-bulkwriter/src/main/java/io/milvus/bulkwriter/BaseBulkImport.java

@@ -30,6 +30,7 @@ import java.util.Map;
 public class BaseBulkImport {
     protected static String postRequest(String url, String apiKey, Map<String, Object> params, int timeout) {
         try {
+            setDefaultOptionsIfCallCloud(params, apiKey);
             kong.unirest.HttpResponse<String> response = Unirest.post(url)
                     .connectTimeout(timeout)
                     .headers(httpHeaders(apiKey))
@@ -85,4 +86,14 @@ public class BaseBulkImport {
             ExceptionUtils.throwUnExpectedException(String.format("Failed to request url: %s, code: %s, message: %s", url, innerCode, innerMessage));
         }
     }
+
+    private static void setDefaultOptionsIfCallCloud(Map<String, Object> params, String apiKey) {
+        if (StringUtils.isNotEmpty(apiKey)) {
+            Map<String, Object> options = new HashMap<>();
+            options.put("sdk", "java");
+            options.put("scene", "bulkWriter");
+
+            params.put("options", options);
+        }
+    }
 }