Support for multiple datasource sessions (#251) (#252)

* support for multiple sessions



* lint fixes



* async helperlint fix



* update default import



* restore snapshot



* update snapshots



* lint fix for prefer named exports



* fix lint for main.test.tsx



* remove eslintcache



---------


(cherry picked from commit 96d5644)

Signed-off-by: Shenoy Pratik <[email protected]>
Signed-off-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>

common/utils/async_query_helpers.ts

@@ -12,24 +12,32 @@ import {
   POLL_INTERVAL_MS,
 } from '../constants';
 
-export const setAsyncSessionId = (value: string | null) => {
+export const setAsyncSessionId = (dataSource: string, value: string | null) => {
   if (value !== null) {
-    sessionStorage.setItem(ASYNC_QUERY_SESSION_ID, value);
+    sessionStorage.setItem(`${ASYNC_QUERY_SESSION_ID}_${dataSource}`, value);
   }
 };
 
-export const getAsyncSessionId = () => {
-  return sessionStorage.getItem(ASYNC_QUERY_SESSION_ID);
+export const getAsyncSessionId = (dataSource: string) => {
+  return sessionStorage.getItem(`${ASYNC_QUERY_SESSION_ID}_${dataSource}`);
 };
 
-export const getJobId = (query: {}, http: CoreStart['http'], callback) => {
+export const getJobId = (
+  currentDataSource: string,
+  query: {},
+  http: CoreStart['http'],
+  callback
+) => {
   http
     .post(ASYNC_QUERY_ENDPOINT, {
-      body: JSON.stringify({ ...query, sessionId: getAsyncSessionId() ?? undefined }),
+      body: JSON.stringify({
+        ...query,
+        sessionId: getAsyncSessionId(currentDataSource) ?? undefined,
+      }),
     })
     .then((res) => {
       const id = res.data.resp.queryId;
-      setAsyncSessionId(_.get(res.data.resp, 'sessionId', null));
+      setAsyncSessionId(currentDataSource, _.get(res.data.resp, 'sessionId', null));
       if (id === undefined) {
         console.error(JSON.parse(res.data.body));
       }
@@ -51,14 +59,14 @@ export const pollQueryStatus = (id: string, http: CoreStart['http'], callback) =
         status === 'scheduled' ||
         status === 'waiting'
       ) {
-        callback({ status: status });
+        callback({ status });
         setTimeout(() => pollQueryStatus(id, http, callback), POLL_INTERVAL_MS);
       } else if (status === 'failed') {
         const results = res.data.resp;
         callback({ status: 'FAILED', error: results.error });
       } else if (status === 'success') {
         const results = _.get(res.data.resp, 'datarows');
-        callback({ status: 'SUCCESS', results: results });
+        callback({ status: 'SUCCESS', results });
       }
     })
     .catch((err) => {

public/components/Main/main.test.tsx

@@ -17,7 +17,7 @@ import {
   mockQueryTranslationResponse,
   mockResultWithNull,
 } from '../../../test/mocks/mockData';
-import Main from './main';
+import { Main } from './main';
 
 const setBreadcrumbsMock = jest.fn();
 
@@ -129,7 +129,6 @@ describe('<Main /> spec', () => {
   });
 
   it('click clear button', async () => {
-    let postRequestFlag = 0;
     const client = httpClientMock;
     client.get = jest.fn().mockResolvedValue(mockDatasourcesQuery);
     client.post = jest.fn(() => {