From 5cb426e7f3f515aca63a5f150e8724232678f78f Mon Sep 17 00:00:00 2001
From: CJ Cenizal <cj@cenizal.com>
Date: Mon, 23 Jul 2018 16:01:34 -0700
Subject: [PATCH] Add rollupSearchStrategy and SearchError (#20505)

* Add /api/rollup/search endpoint and rollupSearchStrategy.
* Add SearchError for surfacing courier search errors.
- Use toastNotifications to surface errors in visualization editor.
- Add call-out react directive and use it to surface rollup errors in the visualization editor sidebar.
- Temporarily assign timezone and interval values from rollup job to the search to avoid errors.
---
 src/ui/public/courier/fetch/fetch_now.js      |   6 +-
 .../public/vis/editors/default/sidebar.html   |   9 ++
 src/ui/public/vis/editors/default/sidebar.js  |  19 +++
 .../public/vis/editors/default/sidebar.less   |   3 +
 x-pack/plugins/rollup/index.js                |   8 +-
 x-pack/plugins/rollup/public/search/index.js  |   7 ++
 .../plugins/rollup/public/search/register.js  |  10 ++
 .../public/search/rollup_search_strategy.js   | 114 ++++++++++++++++++
 .../server/client/elasticsearch_rollup.js     |  15 +++
 .../plugins/rollup/server/routes/api/index.js |   1 +
 .../rollup/server/routes/api/search.js        |  36 ++++++
 11 files changed, 225 insertions(+), 3 deletions(-)
 create mode 100644 src/ui/public/vis/editors/default/sidebar.less
 create mode 100644 x-pack/plugins/rollup/public/search/index.js
 create mode 100644 x-pack/plugins/rollup/public/search/register.js
 create mode 100644 x-pack/plugins/rollup/public/search/rollup_search_strategy.js
 create mode 100644 x-pack/plugins/rollup/server/routes/api/search.js

diff --git a/src/ui/public/courier/fetch/fetch_now.js b/src/ui/public/courier/fetch/fetch_now.js
index adc3635d9e9c0..7bb3890fbf95e 100644
--- a/src/ui/public/courier/fetch/fetch_now.js
+++ b/src/ui/public/courier/fetch/fetch_now.js
@@ -73,7 +73,11 @@ export function FetchNowProvider(Private, Promise) {
     return startRequests(searchRequests)
       .then(function () {
         replaceAbortedRequests();
-        return callClient(searchRequests);
+        return callClient(searchRequests)
+          .catch(() => {
+            // Silently swallow errors that result from search requests so the consumer can surface
+            // them as notifications instead of courier forcing fatal errors.
+          });
       })
       .then(function (responses) {
         replaceAbortedRequests();
diff --git a/src/ui/public/vis/editors/default/sidebar.html b/src/ui/public/vis/editors/default/sidebar.html
index 9460225ff079d..6ca0ad94989b3 100644
--- a/src/ui/public/vis/editors/default/sidebar.html
+++ b/src/ui/public/vis/editors/default/sidebar.html
@@ -19,6 +19,15 @@
     </div>
 
     <nav class="navbar navbar-default subnav visEditorSidebar__nav">
+      <div class="visEditorSidebarError" ng-if="sidebar.persistentError">
+        <call-out
+          color="'danger'"
+          iconType="'alert'"
+          title="sidebar.persistentError"
+          size="'s'"
+        />
+      </div>
+
       <div class="container-fluid">
 
         <!-- tabs -->
diff --git a/src/ui/public/vis/editors/default/sidebar.js b/src/ui/public/vis/editors/default/sidebar.js
index 886de3fbd6e0d..96bd4f755ea2d 100644
--- a/src/ui/public/vis/editors/default/sidebar.js
+++ b/src/ui/public/vis/editors/default/sidebar.js
@@ -23,6 +23,8 @@ import './vis_options';
 import { uiModules } from '../../../modules';
 import sidebarTemplate from './sidebar.html';
 
+import './sidebar.less';
+
 uiModules
   .get('app/visualize')
   .directive('visEditorSidebar', function () {
@@ -32,6 +34,8 @@ uiModules
       scope: true,
       controllerAs: 'sidebar',
       controller: function ($scope) {
+        this.persistentError = undefined;
+
         $scope.$watch('vis.type', (visType) => {
           if (visType) {
             this.showData = visType.schemas.buckets || visType.schemas.metrics;
@@ -46,6 +50,21 @@ uiModules
             this.section = this.section || (this.showData ? 'data' : _.get(visType, 'editorConfig.optionTabs[0].name'));
           }
         });
+
+        // TODO: This doesn't update when requestError is set.
+        $scope.$watch('vis.requestError', (requestError) => {
+          if (requestError && requestError.messsage) {
+            const { message } = requestError;
+            const isRollupError = message.includes('rollup');
+
+            if (isRollupError) {
+              this.persistentError = message;
+              return;
+            }
+          }
+
+          this.persistentError = undefined;
+        });
       }
     };
   });
diff --git a/src/ui/public/vis/editors/default/sidebar.less b/src/ui/public/vis/editors/default/sidebar.less
new file mode 100644
index 0000000000000..9e346211e073f
--- /dev/null
+++ b/src/ui/public/vis/editors/default/sidebar.less
@@ -0,0 +1,3 @@
+.visEditorSidebarError {
+  padding: 8px;
+}
diff --git a/x-pack/plugins/rollup/index.js b/x-pack/plugins/rollup/index.js
index 42879f503b445..2f91455c91288 100644
--- a/x-pack/plugins/rollup/index.js
+++ b/x-pack/plugins/rollup/index.js
@@ -7,9 +7,9 @@
 import { resolve } from 'path';
 import { PLUGIN } from './common/constants';
 import { registerLicenseChecker } from './server/lib/register_license_checker';
-import { registerIndicesRoute, registerFieldsForWildcardRoute } from './server/routes/api';
+import { registerIndicesRoute, registerFieldsForWildcardRoute, registerSearchRoute } from './server/routes/api';
 
-export function rollup(kibana)  {
+export function rollup(kibana) {
   return new kibana.Plugin({
     id: PLUGIN.ID,
     publicDir: resolve(__dirname, 'public'),
@@ -22,11 +22,15 @@ export function rollup(kibana)  {
       visualize: [
         'plugins/rollup/visualize',
       ],
+      search: [
+        'plugins/rollup/search',
+      ],
     },
     init: function (server) {
       registerLicenseChecker(server);
       registerIndicesRoute(server);
       registerFieldsForWildcardRoute(server);
+      registerSearchRoute(server);
     }
   });
 }
diff --git a/x-pack/plugins/rollup/public/search/index.js b/x-pack/plugins/rollup/public/search/index.js
new file mode 100644
index 0000000000000..4d2947eb36f46
--- /dev/null
+++ b/x-pack/plugins/rollup/public/search/index.js
@@ -0,0 +1,7 @@
+/*
+* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+* or more contributor license agreements. Licensed under the Elastic License;
+* you may not use this file except in compliance with the Elastic License.
+*/
+
+import './register';
diff --git a/x-pack/plugins/rollup/public/search/register.js b/x-pack/plugins/rollup/public/search/register.js
new file mode 100644
index 0000000000000..9f0da80f14245
--- /dev/null
+++ b/x-pack/plugins/rollup/public/search/register.js
@@ -0,0 +1,10 @@
+/*
+* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+* or more contributor license agreements. Licensed under the Elastic License;
+* you may not use this file except in compliance with the Elastic License.
+*/
+
+import { addSearchStrategy } from 'ui/courier';
+import { rollupSearchStrategy } from './rollup_search_strategy';
+
+addSearchStrategy(rollupSearchStrategy);
diff --git a/x-pack/plugins/rollup/public/search/rollup_search_strategy.js b/x-pack/plugins/rollup/public/search/rollup_search_strategy.js
new file mode 100644
index 0000000000000..37e61ad00fa15
--- /dev/null
+++ b/x-pack/plugins/rollup/public/search/rollup_search_strategy.js
@@ -0,0 +1,114 @@
+/*
+* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+* or more contributor license agreements. Licensed under the Elastic License;
+* you may not use this file except in compliance with the Elastic License.
+*/
+
+import { kfetchAbortable } from 'ui/kfetch';
+import { SearchError } from 'ui/courier';
+
+export const rollupSearchStrategy = {
+  id: 'rollup',
+
+  search: async ({ searchRequests, Promise }) => {
+    // TODO: Batch together requests to hit a bulk rollup search endpoint.
+    const searchRequest = searchRequests[0];
+    const {
+      index: { title: indexPattern },
+      body: {
+        size,
+        aggs,
+      },
+    } = await searchRequest.getFetchParams();
+
+    function findDateHistogram(aggs) {
+      if (Array.isArray(aggs)) {
+        for (let i = 0; i < aggs.length; i++) {
+          const dateHistogram = findDateHistogram(aggs[i]);
+
+          if (dateHistogram) {
+            return dateHistogram;
+          }
+        }
+      } else if (typeof aggs === 'object') {
+        const aggNames = Object.keys(aggs);
+        const aggsList = aggNames.map(aggName => aggs[aggName]);
+
+        if (aggsList.includes('date_histogram')) {
+          return aggs;
+        }
+
+        return findDateHistogram(aggsList);
+      }
+    }
+
+    // TODO: Temporarily automatically assign same timezone and interval as what's defined by
+    // the rollup job. This should be done by the visualization itself.
+    const searchableAggs = JSON.parse(searchRequest.source.getField('index').originalBody.typeMeta);
+    const { time_zone: timeZone, interval } = findDateHistogram(searchableAggs);
+
+    Object.keys(aggs).forEach(aggName => {
+      const subAggs = aggs[aggName];
+
+      Object.keys(subAggs).forEach(subAggName => {
+        if (subAggName === 'date_histogram') {
+          const subAgg = subAggs[subAggName];
+          subAgg.time_zone = timeZone;
+          subAgg.interval = interval;
+        }
+      });
+    });
+
+    const index = indexPattern;
+    const query = {
+      'size': size,
+      'aggregations': aggs,
+    };
+
+    const {
+      fetching,
+      abort,
+    } = kfetchAbortable({
+      pathname: '../api/rollup/search',
+      method: 'POST',
+      body: JSON.stringify({ index, query }),
+    });
+
+    // TODO: Implement this. Search requests which can't be sent.
+    const failedSearchRequests = [];
+
+    return {
+      // Munge data into shape expected by consumer.
+      searching: new Promise((resolve, reject) => {
+        fetching.then(result => {
+          resolve([ result ]);
+        }).catch(error => {
+          const {
+            body: { statusText, error: title, message },
+            res: { url },
+          } = error;
+
+          // Format kfetch error as a SearchError.
+          const searchError = new SearchError({
+            status: statusText,
+            title,
+            message,
+            path: url,
+          });
+
+          reject(searchError);
+        });
+      }),
+      abort,
+      failedSearchRequests,
+    };
+  },
+
+  isViable: (indexPattern) => {
+    if (!indexPattern) {
+      return false;
+    }
+
+    return indexPattern.type === 'rollup';
+  },
+};
diff --git a/x-pack/plugins/rollup/server/client/elasticsearch_rollup.js b/x-pack/plugins/rollup/server/client/elasticsearch_rollup.js
index 22148a0b112b1..10399861cad64 100644
--- a/x-pack/plugins/rollup/server/client/elasticsearch_rollup.js
+++ b/x-pack/plugins/rollup/server/client/elasticsearch_rollup.js
@@ -26,5 +26,20 @@ export const elasticsearchJsPlugin = (Client, config, components) => {
     ],
     method: 'GET'
   });
+
+  rollup.search = ca({
+    urls: [
+      {
+        fmt: '/<%=index%>/_rollup_search',
+        req: {
+          index: {
+            type: 'string'
+          }
+        }
+      }
+    ],
+    needBody: true,
+    method: 'POST'
+  });
 };
 
diff --git a/x-pack/plugins/rollup/server/routes/api/index.js b/x-pack/plugins/rollup/server/routes/api/index.js
index 5e9e9644d422e..0bf361d6c6399 100644
--- a/x-pack/plugins/rollup/server/routes/api/index.js
+++ b/x-pack/plugins/rollup/server/routes/api/index.js
@@ -6,3 +6,4 @@
 
 export { registerIndicesRoute } from './indices';
 export { registerFieldsForWildcardRoute } from './index_patterns';
+export { registerSearchRoute } from './search';
diff --git a/x-pack/plugins/rollup/server/routes/api/search.js b/x-pack/plugins/rollup/server/routes/api/search.js
new file mode 100644
index 0000000000000..a77040daad78a
--- /dev/null
+++ b/x-pack/plugins/rollup/server/routes/api/search.js
@@ -0,0 +1,36 @@
+/*
+* Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
+* or more contributor license agreements. Licensed under the Elastic License;
+* you may not use this file except in compliance with the Elastic License.
+*/
+import { callWithRequestFactory } from '../../lib/call_with_request_factory';
+import { isEsErrorFactory } from '../../lib/is_es_error_factory';
+import { wrapEsError, wrapUnknownError } from '../../lib/error_wrappers';
+
+export function registerSearchRoute(server) {
+  const isEsError = isEsErrorFactory(server);
+
+  server.route({
+    path: '/api/rollup/search',
+    method: 'POST',
+    handler: async (request, reply) => {
+      const { index, query } = request.payload;
+      const callWithRequest = callWithRequestFactory(server, request);
+
+      try {
+        const results = await callWithRequest('rollup.search', {
+          index,
+          body: query,
+        });
+
+        reply(results);
+      } catch(err) {
+        if (isEsError(err)) {
+          return reply(wrapEsError(err));
+        }
+
+        reply(wrapUnknownError(err));
+      }
+    },
+  });
+}