update exception throw

zilliztech · Oct 31, 2024 · 78120e5 · 78120e5
1 parent bdb669d
commit 78120e5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 7 deletions.
diff --git a/...n/java/org/apache/seatunnel/connectors/seatunnel/milvus/sink/MilvusBufferBatchWriter.java b/...n/java/org/apache/seatunnel/connectors/seatunnel/milvus/sink/MilvusBufferBatchWriter.java
@@ -216,7 +216,6 @@ public boolean needFlush() {
     }
 
     public void flush() throws Exception {
-        log.info("Starting to put {} records to Milvus.", this.writeCache.get());
         // Flush the batch writer
         // Get the number of records completed
         if (this.milvusDataCache.isEmpty()) {

diff --git a/...main/java/org/apache/seatunnel/connectors/seatunnel/milvus/source/MilvusSourceReader.java b/...main/java/org/apache/seatunnel/connectors/seatunnel/milvus/source/MilvusSourceReader.java
@@ -255,8 +255,7 @@ private void queryIteratorData(
                                     "Iterate next data from milvus failed, batchSize = {}, throw exception",
                                     batchSize,
                                     e);
-                            throw new MilvusConnectorException(
-                                    MilvusConnectionErrorCode.READ_DATA_FAIL, e);
+                            throw e;
                         }
                         log.error(
                                 "Iterate next data from milvus failed, batchSize = {}, will retry after 30 s, maxRetry: {}",
@@ -267,8 +266,7 @@ private void queryIteratorData(
                     } else {
                         // if this error, we need to reduce batch size and try again, so throw
                         // exception here
-                        throw new MilvusConnectorException(
-                                MilvusConnectionErrorCode.READ_DATA_FAIL, e);
+                        throw e;
                     }
                 }
             }
@@ -281,7 +279,7 @@ private void queryIteratorData(
                 Thread.sleep(30000);
                 queryIteratorData(tablePath, partitionName, tableSchema, output, batchSize / 2);
             } else {
-                throw new MilvusConnectorException(MilvusConnectionErrorCode.READ_DATA_FAIL, e);
+                throw new MilvusConnectorException(MilvusConnectionErrorCode.READ_DATA_FAIL, e.getMessage(), e);
             }
         }
     }

diff --git a/seatunnel-connectors-v2/pom.xml b/seatunnel-connectors-v2/pom.xml
@@ -41,7 +41,7 @@
         <module>connector-pinecone</module>
         <module>connector-qdrant</module>
         <module>connector-tencent-vectordb</module>
-        <module>connector-astradb</module>
+<!--        <module>connector-astradb</module>-->
 <!--        <module>connector-cdc</module>-->
 <!--        <module>connector-clickhouse</module>-->
 <!--        <module>connector-http</module>-->