This is an automated email from the ASF dual-hosted git repository. acosentino pushed a commit to branch milvus-headers in repository https://gitbox.apache.org/repos/asf/camel.git
commit 6aad4d54b736976320612abf82a9988a1915a0cf Author: Andrea Cosentino <[email protected]> AuthorDate: Tue Mar 19 12:37:04 2024 +0100 Camel-Milvus: Improve headers Signed-off-by: Andrea Cosentino <[email protected]> --- .../java/org/apache/camel/component/milvus/Milvus.java | 14 -------------- .../org/apache/camel/component/milvus/MilvusProducer.java | 1 + .../camel/component/milvus/it/MilvusComponentIT.java | 2 ++ 3 files changed, 3 insertions(+), 14 deletions(-) diff --git a/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/Milvus.java b/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/Milvus.java index dbf45547b05..ebc92861673 100644 --- a/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/Milvus.java +++ b/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/Milvus.java @@ -29,24 +29,10 @@ public class Milvus { enums = "CREATE_COLLECTION,CREATE_INDEX,UPSERT,INSERT,SEARCH,DELETE") public static final String ACTION = "CamelMilvusAction"; - @Metadata(description = "Operation ID.", javaType = "long") - public static final String OPERATION_ID = "CamelMilvusOperationID"; - @Metadata(description = "Operation Status.", javaType = "String") public static final String OPERATION_STATUS = "CamelMilvusOperationStatus"; @Metadata(description = "Operation Status Value.", javaType = "int") public static final String OPERATION_STATUS_VALUE = "CamelMilvusOperationStatusValue"; - - @Metadata(description = "Include Payload.", javaType = "boolean", defaultValue = "true") - public static final String INCLUDE_PAYLOAD = "CamelMilvusWithPayload"; - public static final boolean DEFAULT_INCLUDE_PAYLOAD = true; - - @Metadata(description = "Include Vectors.", javaType = "boolean", defaultValue = "false") - public static final String INCLUDE_VECTORS = "CamelMilvusWithVectors"; - public static final boolean DEFAULT_INCLUDE_VECTORS = false; - - @Metadata(description = "The number of elements.", javaType = "int") - public static final String SIZE = "CamelMilvusSize"; } } diff --git a/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/MilvusProducer.java b/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/MilvusProducer.java index c6f23fcc687..cf58965ca76 100644 --- a/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/MilvusProducer.java +++ b/components/camel-milvus/src/main/java/org/apache/camel/component/milvus/MilvusProducer.java @@ -186,6 +186,7 @@ public class MilvusProducer extends DefaultProducer { private void populateResponse(R<?> r, Exchange exchange) { Message out = exchange.getMessage(); out.setHeader(Milvus.Headers.OPERATION_STATUS, r.getStatus()); + out.setHeader(Milvus.Headers.OPERATION_STATUS_VALUE, r.getStatus().intValue()); out.setBody(r.getData()); } } diff --git a/components/camel-milvus/src/test/java/org/apache/camel/component/milvus/it/MilvusComponentIT.java b/components/camel-milvus/src/test/java/org/apache/camel/component/milvus/it/MilvusComponentIT.java index 6220150f19c..5d6a5270f88 100644 --- a/components/camel-milvus/src/test/java/org/apache/camel/component/milvus/it/MilvusComponentIT.java +++ b/components/camel-milvus/src/test/java/org/apache/camel/component/milvus/it/MilvusComponentIT.java @@ -202,6 +202,7 @@ public class MilvusComponentIT extends MilvusTestSupport { assertThat(result).isNotNull(); assertThat(result.getException()).isNull(); assertThat(result.getMessage().getBody(SearchResponse.class).rowRecords.size() == 2); + System.err.println(result.getMessage().getHeaders()); } @Test @@ -240,6 +241,7 @@ public class MilvusComponentIT extends MilvusTestSupport { assertThat(result).isNotNull(); assertThat(result.getException()).isNull(); assertThat(result.getMessage().getBody(SearchResponse.class).rowRecords.size() == 0); + System.err.println(result.getMessage().getHeaders()); } private List<List<Float>> generateFloatVectors(int count) {
