Fix ppl commands (#8199)

Signed-off-by: sumukhswamy <sumukhhs@amazon.com>
opensearch-project · Sep 19, 2024 · 6ebc1e3 · 6ebc1e3
1 parent 6f9672b
commit 6ebc1e3
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/plugins/query_enhancements/public/search/ppl_search_interceptor.ts b/src/plugins/query_enhancements/public/search/ppl_search_interceptor.ts
@@ -15,12 +15,12 @@ import {
   SearchInterceptorDeps,
 } from '../../../data/public';
 import {
-  formatDate,
-  SEARCH_STRATEGY,
   API,
   EnhancedFetchContext,
   fetch,
+  formatDate,
   QueryAggConfig,
+  SEARCH_STRATEGY,
 } from '../../common';
 import { QueryEnhancementsPluginStartDependencies } from '../types';
 
@@ -73,8 +73,10 @@ export class PPLSearchInterceptor extends SearchInterceptor {
     const query: Query = this.queryService.queryString.getQuery();
     const dataset = query.dataset;
     if (!dataset || !dataset.timeFieldName) return query;
+    const [baseQuery, ...afterPipeParts] = query.query.split('|');
+    const afterPipe = afterPipeParts.length > 0 ? ` | ${afterPipeParts.join('|').trim()}` : '';
     const timeFilter = this.getTimeFilter(dataset.timeFieldName);
-    return { ...query, query: query.query + timeFilter };
+    return { ...query, query: baseQuery + timeFilter + afterPipe };
   }
 
   private getAggConfig(request: IOpenSearchDashboardsSearchRequest, query: Query) {