propagate execution context for timelion

elastic · Aug 10, 2021 · bf2511c · bf2511c
1 parent fa0d98a
commit bf2511c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 1 deletion.
diff --git a/src/plugins/vis_type_timelion/public/helpers/timelion_request_handler.ts b/src/plugins/vis_type_timelion/public/helpers/timelion_request_handler.ts
@@ -7,6 +7,7 @@
  */
 
 import { i18n } from '@kbn/i18n';
+import type { KibanaExecutionContext } from 'kibana/public';
 import { KibanaContext, TimeRange, Filter, esQuery, Query } from '../../../data/public';
 import { TimelionVisDependencies } from '../plugin';
 import { getTimezone } from './get_timezone';
@@ -60,12 +61,14 @@ export function getTimelionRequestHandler({
     query,
     visParams,
     searchSessionId,
+    executionContext,
   }: {
     timeRange: TimeRange;
     filters: Filter[];
     query: Query;
     visParams: TimelionVisParams;
     searchSessionId?: string;
+    executionContext?: KibanaExecutionContext;
   }): Promise<TimelionSuccessResponse> {
     const dataSearch = getDataSearch();
     const expression = visParams.expression;
@@ -110,6 +113,7 @@ export function getTimelionRequestHandler({
             searchSession: searchSessionOptions,
           }),
         }),
+        context: executionContext,
       });
     } catch (e) {
       if (e && e.body) {

diff --git a/src/plugins/vis_type_timelion/public/timelion_vis_fn.ts b/src/plugins/vis_type_timelion/public/timelion_vis_fn.ts
@@ -59,7 +59,7 @@ export const getTimelionVisualizationConfig = (
       help: '',
     },
   },
-  async fn(input, args, { getSearchSessionId }) {
+  async fn(input, args, { getSearchSessionId, getExecutionContext }) {
     const timelionRequestHandler = getTimelionRequestHandler(dependencies);
 
     const visParams = { expression: args.expression, interval: args.interval };
@@ -70,6 +70,7 @@ export const getTimelionVisualizationConfig = (
       filters: get(input, 'filters') as Filter[],
       visParams,
       searchSessionId: getSearchSessionId(),
+      executionContext: getExecutionContext(),
     });
 
     response.visType = TIMELION_VIS_NAME;