Add _primary preference only for segment replication enabled indices (#…

…2040) * Add _primary preference only for segment replication enabled indices Signed-off-by: Suraj Singh <[email protected]> * Add test for segment replication tests Signed-off-by: Suraj Singh <[email protected]> * Remove isSegRepEnabled check and use _primary_first preference Signed-off-by: Suraj Singh <[email protected]> * Remove unused ClusterState reference Signed-off-by: Suraj Singh <[email protected]> * Self review Signed-off-by: Suraj Singh <[email protected]> --------- Signed-off-by: Suraj Singh <[email protected]>
opensearch-project · Sep 1, 2023 · 9c12628 · 9c12628
1 parent 5903e6c
commit 9c12628
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/...main/java/org/opensearch/sql/datasources/storage/OpenSearchDataSourceMetadataStorage.java b/...main/java/org/opensearch/sql/datasources/storage/OpenSearchDataSourceMetadataStorage.java
@@ -224,6 +224,8 @@ private List<DataSourceMetadata> searchInDataSourcesIndex(QueryBuilder query) {
     searchSourceBuilder.query(query);
     searchSourceBuilder.size(DATASOURCE_QUERY_RESULT_SIZE);
     searchRequest.source(searchSourceBuilder);
+    // https://github.com/opensearch-project/sql/issues/1801.
+    searchRequest.preference("_primary_first");
     ActionFuture<SearchResponse> searchResponseActionFuture;
     try (ThreadContext.StoredContext ignored =
         client.threadPool().getThreadContext().stashContext()) {