diff --git a/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/sink/MilvusBufferBatchWriter.java b/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/sink/MilvusBufferBatchWriter.java index a47e9e4d021..8224154401e 100644 --- a/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/sink/MilvusBufferBatchWriter.java +++ b/seatunnel-connectors-v2/connector-milvus/src/main/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/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/source/MilvusSourceReader.java b/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/source/MilvusSourceReader.java index db4ffad4304..67a3266a904 100644 --- a/seatunnel-connectors-v2/connector-milvus/src/main/java/org/apache/seatunnel/connectors/seatunnel/milvus/source/MilvusSourceReader.java +++ b/seatunnel-connectors-v2/connector-milvus/src/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 index 794ee865d82..3d781f009af 100644 --- a/seatunnel-connectors-v2/pom.xml +++ b/seatunnel-connectors-v2/pom.xml @@ -41,7 +41,7 @@ connector-pinecone connector-qdrant connector-tencent-vectordb - connector-astradb +