shim inside strategies

Liza K · Liza K · commit 2ba358fee81d · 2020-11-24T13:21:16.000+02:00
diff --git a/src/plugins/data/server/search/es_search/response_utils.ts b/src/plugins/data/server/search/es_search/response_utils.ts
@@ -18,6 +18,7 @@
  */
 
 import { SearchResponse } from 'elasticsearch';
+import { shimHitsTotal } from '..';
 
 /**
  * Get the `total`/`loaded` for this response (see `IKibanaSearchResponse`). Note that `skipped` is
@@ -36,7 +37,7 @@ export function getTotalLoaded(response: SearchResponse<unknown>) {
  */
 export function toKibanaSearchResponse(rawResponse: SearchResponse<unknown>) {
   return {
-    rawResponse,
+    rawResponse: shimHitsTotal(rawResponse),
     isPartial: false,
     isRunning: false,
     ...getTotalLoaded(rawResponse),
diff --git a/src/plugins/data/server/search/routes/search.ts b/src/plugins/data/server/search/routes/search.ts
@@ -21,7 +21,6 @@ import { first } from 'rxjs/operators';
 import { schema } from '@kbn/config-schema';
 import type { IRouter } from 'src/core/server';
 import { getRequestAbortedSignal } from '../../lib';
-import { shimHitsTotal } from './shim_hits_total';
 
 export function registerSearchRoute(router: IRouter): void {
   router.post(
@@ -66,12 +65,7 @@ export function registerSearchRoute(router: IRouter): void {
           .toPromise();
 
         return res.ok({
-          body: {
-            ...response,
-            ...{
-              rawResponse: shimHitsTotal(response.rawResponse),
-            },
-          },
+          body: response,
         });
       } catch (err) {
         return res.customError({
diff --git a/src/plugins/data/server/search/search_service.ts b/src/plugins/data/server/search/search_service.ts
@@ -43,7 +43,7 @@ import { AggsService } from './aggs';
 
 import { FieldFormatsStart } from '../field_formats';
 import { IndexPatternsServiceStart } from '../index_patterns';
-import { getCallMsearch, registerMsearchRoute, registerSearchRoute, shimHitsTotal } from './routes';
+import { getCallMsearch, registerMsearchRoute, registerSearchRoute } from './routes';
 import { ES_SEARCH_STRATEGY, esSearchStrategyProvider } from './es_search';
 import { DataPluginStart } from '../plugin';
 import { UsageCollectionSetup } from '../../../usage_collection/server';
diff --git a/x-pack/plugins/data_enhanced/server/search/response_utils.ts b/x-pack/plugins/data_enhanced/server/search/response_utils.ts
@@ -5,7 +5,7 @@
  */
 
 import { ApiResponse } from '@elastic/elasticsearch';
-import { getTotalLoaded } from '../../../../../src/plugins/data/server';
+import { getTotalLoaded, shimHitsTotal } from '../../../../../src/plugins/data/server';
 import { AsyncSearchResponse, EqlSearchResponse } from './types';
 import { EqlSearchStrategyResponse } from '../../common/search';
 
@@ -15,7 +15,7 @@ import { EqlSearchStrategyResponse } from '../../common/search';
 export function toAsyncKibanaSearchResponse(response: AsyncSearchResponse) {
   return {
     id: response.id,
-    rawResponse: response.response,
+    rawResponse: shimHitsTotal(response.response),
     isPartial: response.is_partial,
     isRunning: response.is_running,
     ...getTotalLoaded(response.response),
@@ -31,7 +31,7 @@ export function toEqlKibanaSearchResponse(
 ): EqlSearchStrategyResponse {
   return {
     id: response.body.id,
-    rawResponse: response,
+    rawResponse: shimHitsTotal(response.response),
     isPartial: response.body.is_partial,
     isRunning: response.body.is_running,
   };
