[Logs UI] Replace legacy es client usage in category examples (#100716) (#101055)

kibanamachine · weltenwort · web-flow · commit f7b99c227845 · 2021-06-01T11:47:09.000-04:00
Co-authored-by: Kibana Machine &lt;42973632+kibanamachine@users.noreply.github.com&gt;

Co-authored-by: Felix Stürmer &lt;weltenwort@users.noreply.github.com&gt;
diff --git a/x-pack/plugins/infra/server/lib/log_analysis/log_entry_categories_analysis.ts b/x-pack/plugins/infra/server/lib/log_analysis/log_entry_categories_analysis.ts
@@ -6,7 +6,7 @@
  */
 
 import type { estypes } from '@elastic/elasticsearch';
-import type { ILegacyScopedClusterClient } from 'src/core/server';
+import type { ElasticsearchClient } from 'src/core/server';
 import {
   compareDatasetsByMaximumAnomalyScore,
   getJobId,
@@ -136,7 +136,7 @@ export async function getLogEntryCategoryDatasets(
 
 export async function getLogEntryCategoryExamples(
   context: {
-    core: { elasticsearch: { legacy: { client: ILegacyScopedClusterClient } } };
+    core: { elasticsearch: { client: { asCurrentUser: ElasticsearchClient } } };
     infra: {
       mlAnomalyDetectors: MlAnomalyDetectors;
       mlSystem: MlSystem;
@@ -402,7 +402,7 @@ async function fetchTopLogEntryCategoryHistograms(
 }
 
 async function fetchLogEntryCategoryExamples(
-  requestContext: { core: { elasticsearch: { legacy: { client: ILegacyScopedClusterClient } } } },
+  requestContext: { core: { elasticsearch: { client: { asCurrentUser: ElasticsearchClient } } } },
   indices: string,
   runtimeMappings: estypes.RuntimeFields,
   timestampField: string,
@@ -417,19 +417,20 @@ async function fetchLogEntryCategoryExamples(
   const {
     hits: { hits },
   } = decodeOrThrow(logEntryCategoryExamplesResponseRT)(
-    await requestContext.core.elasticsearch.legacy.client.callAsCurrentUser(
-      'search',
-      createLogEntryCategoryExamplesQuery(
-        indices,
-        runtimeMappings,
-        timestampField,
-        tiebreakerField,
-        startTime,
-        endTime,
-        categoryQuery,
-        exampleCount
+    (
+      await requestContext.core.elasticsearch.client.asCurrentUser.search(
+        createLogEntryCategoryExamplesQuery(
+          indices,
+          runtimeMappings,
+          timestampField,
+          tiebreakerField,
+          startTime,
+          endTime,
+          categoryQuery,
+          exampleCount
+        )
       )
-    )
+    ).body
   );
 
   const esSearchSpan = finalizeEsSearchSpan();
