Multi datasource support

common/utils/async_query_helpers.ts

@@ -26,8 +26,9 @@
 export const executeAsyncQuery = (
   currentDataSource: string,
   query: {},
+  dataSourceId: string,
   pollingCallback: PollingCallback,
-  onErrorCallback?: (errorMessage: string) => void
+  onErrorCallback?: (errorMessage: string) => void,
 ) => {
   let jobId: string | undefined;
   let isQueryFulfilled = false;
@@ -74,7 +75,7 @@
   };
 
   const pollQueryStatus = (id: string, callback: PollingCallback) => {
     !isQueryCancelled &&
       http
         .get(ASYNC_QUERY_JOB_ENDPOINT + id)
         .then((res: AsyncApiResponse) => {

common/utils/fetch_datasources.ts

@@ -0,0 +1,54 @@
+/*
+ * Copyright OpenSearch Contributors
+ * SPDX-License-Identifier: Apache-2.0
+ */
+
+
+import { CoreStart } from "../../../../src/core/public";
+
+export const fetchDataSources = (http: CoreStart['http'], dataSourceId: string, urlDataSource: string, onSuccess, onError) => {
+    let dataOptions: { label: string; options?: any; }[] = [];
+    let urlSourceFound = false;
+    if(!dataSourceId){
+        dataSourceId = ''
+    }
+
+    http.get(`/api/get_datasources/${dataSourceId}`)
+      .then((res) => {
+        const data = res.data.resp;
+        const connectorGroups = {};
+
+        data.forEach((item) => {
+          const connector = item.connector;
+          const name = item.name;
+
+          if (connector === 'S3GLUE') {
+            if (!connectorGroups[connector]) {
+              connectorGroups[connector] = [];
+            }
+
+            connectorGroups[connector].push(name);
+            if (name === urlDataSource) {
+              urlSourceFound = true;
+            }
+          }
+        });
+
+        for (const connector in connectorGroups) {
+          if (connectorGroups.hasOwnProperty(connector)) {
+            const connectorNames = connectorGroups[connector];
+
+            dataOptions.push({
+              label: connector,
+              options: connectorNames.map((name) => ({ label: name })),
+            });
+          }
+        }
+
+        onSuccess(dataOptions, urlSourceFound);
+      })
+      .catch((err) => {
+        console.error(err);
+        onError(err);
+      });
+  };

public/application.tsx

@@ -7,13 +7,15 @@
 import React from 'react';
 import ReactDOM from 'react-dom';
 import { AppMountParameters, CoreStart } from '../../../src/core/public';
-import { AppPluginStartDependencies } from './types';
+import { DataSourceManagementPluginSetup } from '../../../src/plugins/data_source_management/public';
 import { WorkbenchApp } from './components/app';
+import { AppPluginStartDependencies } from './types';
 
 export const renderApp = (
-  { notifications, http, chrome }: CoreStart,
-  { navigation }: AppPluginStartDependencies,
-  { appBasePath, element }: AppMountParameters
+  { notifications, http, chrome, savedObjects }: CoreStart,
+  { navigation, dataSource }: AppPluginStartDependencies,
+  { appBasePath, element, setHeaderActionMenu }: AppMountParameters,
+  dataSourceManagement: DataSourceManagementPluginSetup
 ) => {
   ReactDOM.render(
     <WorkbenchApp
@@ -22,6 +24,10 @@ export const renderApp = (
       http={http}
       navigation={navigation}
       chrome={chrome}
+      savedObjects={savedObjects}
+      dataSourceEnabled={!!dataSource}
+      dataSourceManagement={dataSourceManagement}
+      setActionMenu={setHeaderActionMenu}
     />,
     element
   );