Enable send to background in Timelion (elastic#82232)

* Enable send to background in Timelion * add jest
alexwizp · Nov 2, 2020 · be2132f · be2132f
1 parent 9d3583e
commit be2132f
Show file tree

Hide file tree

Showing 6 changed files with 126 additions and 101 deletions.
diff --git a/src/plugins/vis_type_timelion/public/helpers/plugin_services.ts b/src/plugins/vis_type_timelion/public/helpers/plugin_services.ts
@@ -17,14 +17,16 @@
  * under the License.
  */
 
-import { IndexPatternsContract } from 'src/plugins/data/public';
-import { SavedObjectsClientContract } from 'kibana/public';
+import type { IndexPatternsContract, ISearchStart } from 'src/plugins/data/public';
+import type { SavedObjectsClientContract } from 'kibana/public';
 import { createGetterSetter } from '../../../kibana_utils/public';
 
 export const [getIndexPatterns, setIndexPatterns] = createGetterSetter<IndexPatternsContract>(
   'IndexPatterns'
 );
 
+export const [getDataSearch, setDataSearch] = createGetterSetter<ISearchStart>('Search');
+
 export const [getSavedObjectsClient, setSavedObjectsClient] = createGetterSetter<
   SavedObjectsClientContract
 >('SavedObjectsClient');
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
@@ -23,6 +23,7 @@ import { TimeRange, Filter, esQuery, Query } from '../../../data/public';
 import { TimelionVisDependencies } from '../plugin';
 import { getTimezone } from './get_timezone';
 import { TimelionVisParams } from '../timelion_vis_fn';
+import { getDataSearch } from '../helpers/plugin_services';
 
 interface Stats {
   cacheCount: number;
@@ -93,6 +94,7 @@ export function getTimelionRequestHandler({
 
     // parse the time range client side to make sure it behaves like other charts
     const timeRangeBounds = timefilter.calculateBounds(timeRange);
+    const sessionId = getDataSearch().session.getSessionId();
 
     try {
       return await http.post('/api/timelion/run', {
@@ -109,6 +111,7 @@ export function getTimelionRequestHandler({
             interval: visParams.interval,
             timezone,
           },
+          sessionId,
         }),
       });
     } catch (e) {

diff --git a/src/plugins/vis_type_timelion/public/plugin.ts b/src/plugins/vis_type_timelion/public/plugin.ts
@@ -36,7 +36,7 @@ import { VisualizationsSetup } from '../../visualizations/public';
 
 import { getTimelionVisualizationConfig } from './timelion_vis_fn';
 import { getTimelionVisDefinition } from './timelion_vis_type';
-import { setIndexPatterns, setSavedObjectsClient } from './helpers/plugin_services';
+import { setIndexPatterns, setSavedObjectsClient, setDataSearch } from './helpers/plugin_services';
 import { ConfigSchema } from '../config';
 
 import { getArgValueSuggestions } from './helpers/arg_value_suggestions';
@@ -104,6 +104,7 @@ export class TimelionVisPlugin
   public start(core: CoreStart, plugins: TimelionVisStartDependencies) {
     setIndexPatterns(plugins.data.indexPatterns);
     setSavedObjectsClient(core.savedObjects.client);
+    setDataSearch(plugins.data.search);
 
     return {
       getArgValueSuggestions,

diff --git a/src/plugins/vis_type_timelion/server/routes/run.ts b/src/plugins/vis_type_timelion/server/routes/run.ts
@@ -75,6 +75,7 @@ export function runRoute(
               to: schema.maybe(schema.string()),
             })
           ),
+          sessionId: schema.maybe(schema.string()),
         }),
       },
     },