[Backport 2.x] Sanitize create acceleration queries and direct queries

common/constants/data_sources.ts

@@ -38,6 +38,7 @@ export const ACCELERATION_ADD_FIELDS_TEXT = '(add fields here)';
 export const ACCELERATION_INDEX_NAME_REGEX = /^[a-z0-9_]+$/;
 export const ACCELERATION_S3_URL_REGEX = /^(s3|s3a):\/\/[a-zA-Z0-9.\-]+/;
 export const SPARK_HIVE_TABLE_REGEX = /Provider:\s*hive/;
+export const SANITIZE_QUERY_REGEX = /\s+/g;
 export const TIMESTAMP_DATATYPE = 'timestamp';
 
 export const ACCELERATION_INDEX_TYPES = [

public/components/common/search/direct_search.tsx

@@ -18,11 +18,15 @@
   EuiPopoverFooter,
   EuiToolTip,
 } from '@elastic/eui';
+import { i18n } from '@osd/i18n';
 import { isEmpty, isEqual } from 'lodash';
 import React, { useEffect, useState } from 'react';
 import { batch, useDispatch, useSelector } from 'react-redux';
-import { i18n } from '@osd/i18n';
-import { ASYNC_POLLING_INTERVAL, QUERY_LANGUAGE } from '../../../../common/constants/data_sources';
+import {
+  ASYNC_POLLING_INTERVAL,
+  QUERY_LANGUAGE,
+  SANITIZE_QUERY_REGEX,
+} from '../../../../common/constants/data_sources';
 import {
   APP_ANALYTICS_TAB_ID_REGEX,
   RAW_QUERY,
@@ -60,7 +64,7 @@
   tempQuery: string;
   handleQueryChange: (query: string) => void;
   handleQuerySearch: () => void;
   dslService: any;
 }
 
 export interface IDatePickerProps {
@@ -70,11 +74,11 @@
   setEndTime: () => void;
   setTimeRange: () => void;
   setIsOutputStale: () => void;
   handleTimePickerChange: (timeRange: string[]) => any;
   handleTimeRangePickerRefresh: () => any;
 }
 
 export const DirectSearch = (props: any) => {
   const {
     query,
     tempQuery,
@@ -115,7 +119,7 @@
     error: pollingError,
     startPolling,
     stopPolling,
   } = usePolling<any, any>((params) => {
     return sqlService.fetchWithJobId(params);
   }, ASYNC_POLLING_INTERVAL);
 
@@ -223,9 +227,10 @@
       );
     });
     const sessionId = getAsyncSessionId(explorerSearchMetadata.datasources[0].label);
+    const requestQuery = tempQuery || query;
     const requestPayload = {
       lang: lang.toLowerCase(),
-      query: tempQuery || query,
+      query: requestQuery.replaceAll(SANITIZE_QUERY_REGEX, ' '),
       datasource: explorerSearchMetadata.datasources[0].label,
     } as DirectQueryRequest;
 
@@ -329,20 +334,20 @@
         })
       );
     }
   }, [pollingResult, pollingError]);
 
   useEffect(() => {
     return () => {
       stopPolling();
     };
   }, []);
 
   useEffect(() => {
     if (!explorerSearchMetadata.isPolling) {
       stopPolling();
       setIsQueryRunning(false);
     }
   }, [explorerSearchMetadata.isPolling]);
 
   return (
     <div className="globalQueryBar">

public/components/datasources/components/manage/accelerations/create_accelerations_flyout/create/create_acceleration_button.tsx

@@ -5,6 +5,7 @@
 
 import { EuiButton } from '@elastic/eui';
 import React, { useEffect, useState } from 'react';
+import { SANITIZE_QUERY_REGEX } from '../../../../../../../../common/constants/data_sources';
 import { CreateAccelerationForm } from '../../../../../../../../common/types/data_connections';
 import {
   DirectQueryLoadingStatus,
@@ -39,7 +40,7 @@
 
     const requestPayload: DirectQueryRequest = {
       lang: 'sql',
-      query: accelerationQueryBuilder(accelerationFormData),
+      query: accelerationQueryBuilder(accelerationFormData).replaceAll(SANITIZE_QUERY_REGEX, ' '),
       datasource: accelerationFormData.dataSource,
     };
 
@@ -60,7 +61,7 @@
       setIsLoading(false);
       setToast('Create acceleration query failed', 'success');
     }
   }, [directqueryLoadStatus]);
 
   return (
     <EuiButton onClick={createAcceleration} fill isLoading={isLoading}>