丁云鹏 5 달 전
부모
커밋
72c84192a1
1개의 변경된 파일3개의 추가작업 그리고 0개의 파일을 삭제
  1. 3 0
      recommend-model-produce/src/main/java/com/tzld/piaoquan/recommend/model/produce/i2i/I2IMilvusDataImport.java

+ 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()) {