丁云鹏 5 hónapja
szülő
commit
72c84192a1

+ 3 - 0
recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IMilvusDataImport.java

@@ -22,6 +22,7 @@ import java.util.ArrayList;
 import java.util.HashMap;
 import java.util.List;
 import java.util.Map;
+import java.util.concurrent.TimeUnit;
 
 /**
  * @author dyp
@@ -53,10 +54,12 @@ public class I2IMilvusDataImport {
             ConnectParam connectParam = ConnectParam.newBuilder()
                     .withUri(milvusUrl)
                     .withToken(milvusToken)
+                    .withConnectTimeout(60L, TimeUnit.SECONDS)
                     .build();
             RetryParam retryParam = RetryParam.newBuilder()
                     .withMaxRetryTimes(3)
                     .build();
+            System.out.println(connectParam);
             MilvusClient milvusClient = new MilvusServiceClient(connectParam).withRetry(retryParam);
             List<String> batch = new ArrayList<>();
             while (lines.hasNext()) {