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

…2040)

* Add _primary preference only for segment replication enabled indices

Signed-off-by: Suraj Singh <surajrider@gmail.com>

* Add test for segment replication tests

Signed-off-by: Suraj Singh <surajrider@gmail.com>

* Remove isSegRepEnabled check and use _primary_first preference

Signed-off-by: Suraj Singh <surajrider@gmail.com>

* Remove unused ClusterState reference

Signed-off-by: Suraj Singh <surajrider@gmail.com>

* Self review

Signed-off-by: Suraj Singh <surajrider@gmail.com>

---------

Signed-off-by: Suraj Singh <surajrider@gmail.com>
(cherry picked from commit 9c12628)
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

datasources/src/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()) {