[ML] Adding jobs stats to functions shared in setup contract (#88673)

* [ML] Adding jobs stats to functions shared in setup contract * updating types * adding datafeeds Co-authored-by: Kibana Machine <[email protected]>
elastic · Jan 25, 2021 · f15a1e6 · f15a1e6
1 parent 601d9fd
commit f15a1e6
Showing 1 changed file with 45 additions and 1 deletion.
diff --git a/x-pack/plugins/ml/server/shared_services/providers/anomaly_detectors.ts b/x-pack/plugins/ml/server/shared_services/providers/anomaly_detectors.ts
@@ -5,7 +5,12 @@
  */
 
 import { KibanaRequest, SavedObjectsClientContract } from 'kibana/server';
-import { Job } from '../../../common/types/anomaly_detection_jobs';
+import {
+  Job,
+  JobStats,
+  Datafeed,
+  DatafeedStats,
+} from '../../../common/types/anomaly_detection_jobs';
 import { GetGuards } from '../shared_services';
 
 export interface AnomalyDetectorsProvider {
@@ -14,6 +19,9 @@ export interface AnomalyDetectorsProvider {
     savedObjectsClient: SavedObjectsClientContract
   ): {
     jobs(jobId?: string): Promise<{ count: number; jobs: Job[] }>;
+    jobStats(jobId?: string): Promise<{ count: number; jobs: JobStats[] }>;
+    datafeeds(datafeedId?: string): Promise<{ count: number; datafeeds: Datafeed[] }>;
+    datafeedStats(datafeedId?: string): Promise<{ count: number; datafeeds: DatafeedStats[] }>;
   };
 }
 
@@ -36,6 +44,42 @@ export function getAnomalyDetectorsProvider(getGuards: GetGuards): AnomalyDetect
               return body;
             });
         },
+        async jobStats(jobId?: string) {
+          return await getGuards(request, savedObjectsClient)
+            .isFullLicense()
+            .hasMlCapabilities(['canGetJobs'])
+            .ok(async ({ mlClient }) => {
+              const { body } = await mlClient.getJobStats<{
+                count: number;
+                jobs: JobStats[];
+              }>(jobId !== undefined ? { job_id: jobId } : undefined);
+              return body;
+            });
+        },
+        async datafeeds(datafeedId?: string) {
+          return await getGuards(request, savedObjectsClient)
+            .isFullLicense()
+            .hasMlCapabilities(['canGetDatafeeds'])
+            .ok(async ({ mlClient }) => {
+              const { body } = await mlClient.getDatafeeds<{
+                count: number;
+                datafeeds: Datafeed[];
+              }>(datafeedId !== undefined ? { datafeed_id: datafeedId } : undefined);
+              return body;
+            });
+        },
+        async datafeedStats(datafeedId?: string) {
+          return await getGuards(request, savedObjectsClient)
+            .isFullLicense()
+            .hasMlCapabilities(['canGetDatafeeds'])
+            .ok(async ({ mlClient }) => {
+              const { body } = await mlClient.getDatafeedStats<{
+                count: number;
+                datafeeds: DatafeedStats[];
+              }>(datafeedId !== undefined ? { datafeed_id: datafeedId } : undefined);
+              return body;
+            });
+        },
       };
     },
   };