[Async search Discover] Poll for results when current request complet…

…es (#8555) (#8655)

* poll for results when current request completes



* Changeset file for PR #8555 created/updated

* added localization to error string



* simplified error message for user



---------



(cherry picked from commit 6379548)

Signed-off-by: Amardeepsingh Siglani <[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>
Co-authored-by: opensearch-changeset-bot[bot] <154024398+opensearch-changeset-bot[bot]@users.noreply.github.com>

changelogs/fragments/8555.yml

@@ -0,0 +1,2 @@
+fix:
+- Refactored polling logic to poll for results once current request completes ([#8555](https://github.com/opensearch-project/OpenSearch-Dashboards/pull/8555))

src/plugins/data/common/utils/helpers.ts

@@ -28,43 +28,37 @@
  * under the License.
  */
 
-import { timer } from 'rxjs';
-import { filter, mergeMap, take, takeWhile } from 'rxjs/operators';
-import {
-  PollQueryResultsHandler,
-  FetchStatusResponse,
-  QueryFailedStatusResponse,
-} from '../data_frames';
+import { i18n } from '@osd/i18n';
+import { PollQueryResultsHandler, FetchStatusResponse } from '../data_frames';
 
 export interface QueryStatusOptions {
   pollQueryResults: PollQueryResultsHandler;
   queryId?: string;
   interval?: number;
 }
 
-export const handleQueryResults = <T>(
+export const delay = (ms: number) => new Promise((res) => setTimeout(res, ms));
+
+export const handleQueryResults = async <T>(
   options: QueryStatusOptions
 ): Promise<FetchStatusResponse> => {
-  const { pollQueryResults, interval = 5000, queryId } = options;
+  const { pollQueryResults, interval = 5000 } = options;
+  let queryResultsRes: FetchStatusResponse;
+  let queryStatus;
+  do {
+    // Wait for the given interval in ms before polling for the query status/results
+    await delay(interval);
+    queryResultsRes = await pollQueryResults();
+    queryStatus = queryResultsRes?.status?.toUpperCase();
+  } while (queryStatus !== 'SUCCESS' && queryStatus !== 'FAILED');
+
+  if (queryStatus === 'FAILED') {
+    throw new Error(
+      i18n.translate('data.search.request.failed', {
+        defaultMessage: 'An error occurred while executing the search query',
+      })
+    );
+  }
 
-  return timer(0, interval)
-    .pipe(
-      mergeMap(() => pollQueryResults()),
-      takeWhile((response: FetchStatusResponse) => {
-        const status = response?.status?.toUpperCase();
-        return status !== 'SUCCESS' && status !== 'FAILED';
-      }, true),
-      filter((response: FetchStatusResponse) => {
-        const status = response?.status?.toUpperCase();
-        if (status === 'FAILED') {
-          throw (
-            (response as QueryFailedStatusResponse).body.error ??
-            new Error(`Failed to fetch results ${queryId ?? ''}`)
-          );
-        }
-        return status === 'SUCCESS';
-      }),
-      take(1)
-    )
-    .toPromise();
+  return queryResultsRes;
 };