From 2bbe27d94f1a57099130578495243be14b7421df Mon Sep 17 00:00:00 2001
From: Junqiu Lei <junqiu@amazon.com>
Date: Thu, 6 Jun 2024 11:33:15 -0700
Subject: [PATCH] Add data source reference id in data layer search request

Signed-off-by: Junqiu Lei <junqiu@amazon.com>
---
 CHANGELOG.md                          | 1 +
 public/model/layerRenderController.ts | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/CHANGELOG.md b/CHANGELOG.md
index 7dfe96ec..39419d06 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -16,6 +16,7 @@ The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 ### Features
 ### Enhancements
 ### Bug Fixes
+* Add data source reference id in data layer search request ([#623](https://github.com/opensearch-project/dashboards-maps/pull/623))
 ### Infrastructure
 ### Documentation
 ### Maintenance
diff --git a/public/model/layerRenderController.ts b/public/model/layerRenderController.ts
index 77786413..6576a4d4 100644
--- a/public/model/layerRenderController.ts
+++ b/public/model/layerRenderController.ts
@@ -76,6 +76,7 @@ export const prepareDataLayerSource = (
           ...(timeFilters ? [timeFilters] : []),
         ]);
       }
+      const dataSourceId = indexPattern?.dataSourceRef?.id;
       const request = {
         params: {
           index: indexPatternRefName,
@@ -85,6 +86,7 @@ export const prepareDataLayerSource = (
             query: mergedQuery,
           },
         },
+        ...(!!dataSourceId && { dataSourceId }),
       };
 
       const search$ = data.search.search(request).subscribe({