From aef37808500a861c594f9ca5bf0b9e4cd253f16f Mon Sep 17 00:00:00 2001
From: ZMai <zmai@slb.com>
Date: Thu, 2 Feb 2023 13:08:54 -0600
Subject: [PATCH] Update IndexerQueueTaskBuilder for Azure and GCP to forward
 the new attribute 'ancestry_kinds

---
 .../indexer/azure/util/IndexerQueueTaskBuilderAzure.java | 1 +
 .../gcp/common/publish/ReprocessingTaskPublisher.java    | 9 ++++++++-
 .../indexer-gc/src/main/resources/application.properties | 1 +
 .../src/main/resources/application.properties            | 1 +
 4 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/provider/indexer-azure/src/main/java/org/opengroup/osdu/indexer/azure/util/IndexerQueueTaskBuilderAzure.java b/provider/indexer-azure/src/main/java/org/opengroup/osdu/indexer/azure/util/IndexerQueueTaskBuilderAzure.java
index cbd1c7d12..1bfcbe1cb 100644
--- a/provider/indexer-azure/src/main/java/org/opengroup/osdu/indexer/azure/util/IndexerQueueTaskBuilderAzure.java
+++ b/provider/indexer-azure/src/main/java/org/opengroup/osdu/indexer/azure/util/IndexerQueueTaskBuilderAzure.java
@@ -162,6 +162,7 @@ public class IndexerQueueTaskBuilderAzure extends IndexerQueueTaskBuilder {
         jo.addProperty(DpsHeaders.ACCOUNT_ID, headers.getPartitionIdWithFallbackToAccountId());
         jo.addProperty(DpsHeaders.DATA_PARTITION_ID, headers.getPartitionIdWithFallbackToAccountId());
         jo.addProperty(DpsHeaders.CORRELATION_ID, headers.getCorrelationId());
+        // Append the ancestry kinds used to prevent circular chasing
         if(attributes.containsKey(Constants.ANCESTRY_KINDS)) {
             jo.addProperty(Constants.ANCESTRY_KINDS, attributes.get(Constants.ANCESTRY_KINDS));
         }
diff --git a/provider/indexer-gc/src/main/java/org/opengroup/osdu/indexer/provider/gcp/common/publish/ReprocessingTaskPublisher.java b/provider/indexer-gc/src/main/java/org/opengroup/osdu/indexer/provider/gcp/common/publish/ReprocessingTaskPublisher.java
index 4caa625e5..d1936731c 100644
--- a/provider/indexer-gc/src/main/java/org/opengroup/osdu/indexer/provider/gcp/common/publish/ReprocessingTaskPublisher.java
+++ b/provider/indexer-gc/src/main/java/org/opengroup/osdu/indexer/provider/gcp/common/publish/ReprocessingTaskPublisher.java
@@ -34,6 +34,7 @@ import org.opengroup.osdu.core.gcp.oqm.driver.OqmDriver;
 import org.opengroup.osdu.core.gcp.oqm.model.OqmDestination;
 import org.opengroup.osdu.core.gcp.oqm.model.OqmMessage;
 import org.opengroup.osdu.core.gcp.oqm.model.OqmTopic;
+import org.opengroup.osdu.indexer.model.Constants;
 import org.opengroup.osdu.indexer.provider.gcp.indexing.processing.IndexerMessagingConfigProperties;
 import org.opengroup.osdu.indexer.util.IndexerQueueTaskBuilder;
 import org.springframework.context.annotation.Primary;
@@ -98,10 +99,16 @@ public class ReprocessingTaskPublisher extends IndexerQueueTaskBuilder {
     RecordChangedMessages recordChangedMessages = gson.fromJson(payload,
         RecordChangedMessages.class);
 
+    Map<String, String> attributes = getAttributesFromHeaders(headers);
+    // Append the ancestry kinds used to prevent circular chasing
+    if(recordChangedMessages.getAttributes().containsKey(Constants.ANCESTRY_KINDS)) {
+      attributes.put(Constants.ANCESTRY_KINDS, recordChangedMessages.getAttributes().get(Constants.ANCESTRY_KINDS));
+    }
+
     OqmMessage oqmMessage = OqmMessage.builder()
         .id(headers.getCorrelationId())
         .data(recordChangedMessages.getData())
-        .attributes(getAttributesFromHeaders(headers))
+        .attributes(attributes)
         .build();
 
     log.info("Reprocessing task: {} ,has been published.", oqmMessage);
diff --git a/provider/indexer-gc/src/main/resources/application.properties b/provider/indexer-gc/src/main/resources/application.properties
index 721d5d180..fb9c88bd6 100644
--- a/provider/indexer-gc/src/main/resources/application.properties
+++ b/provider/indexer-gc/src/main/resources/application.properties
@@ -49,6 +49,7 @@ STORAGE_SCHEMA_HOST=${STORAGE_API}/schemas
 SCHEMA_BASE_HOST=http://schema
 SCHEMA_PATH=/api/schema-service/v1/schema
 SCHEMA_HOST=${SCHEMA_BASE_HOST}${SCHEMA_PATH}
+SEARCH_HOST=${SEARCH_BASE_HOST}/api/search/v2
 
 records-changed-topic-name=records-changed
 schema-changed-topic-name=schema-changed
diff --git a/provider/indexer-ibm/src/main/resources/application.properties b/provider/indexer-ibm/src/main/resources/application.properties
index aaf4ceb6c..e22d69963 100644
--- a/provider/indexer-ibm/src/main/resources/application.properties
+++ b/provider/indexer-ibm/src/main/resources/application.properties
@@ -27,6 +27,7 @@ CRON_INDEX_CLEANUP_THRESHOLD_DAYS=3
 CRON_EMPTY_INDEX_CLEANUP_THRESHOLD_DAYS=7
 
 SCHEMA_HOST=${HOST}/api/schema-service/v1/schema
+SEARCH_HOST=${search_service_url}/api/search/v2
 
 storage_service_url=http://localhost:8082
 #storage_service_url=https://os-storage-ibm-osdu-r2.osduadev-a1c3eaf78a86806e299f5f3f207556f0-0000.us-south.containers.appdomain.cloud
-- 
GitLab