From 94fb7943865ce0bc95feaae6a0a647ee3e4ac790 Mon Sep 17 00:00:00 2001
From: James Gowdy <jgowdy@elastic.co>
Date: Wed, 26 Feb 2020 12:41:36 +0000
Subject: [PATCH] [ML] Adding get records endpoint wrapper (#58500) (#58574)

---
 .../ml/server/client/elasticsearch_ml.js      | 12 +++++
 .../ml/server/routes/anomaly_detectors.ts     | 48 ++++++++++++++++++-
 .../plugins/ml/server/routes/apidoc.json      |  2 +
 3 files changed, 61 insertions(+), 1 deletion(-)

diff --git a/x-pack/legacy/plugins/ml/server/client/elasticsearch_ml.js b/x-pack/legacy/plugins/ml/server/client/elasticsearch_ml.js
index d7bbba2f15b99..c732ce00c7fe4 100644
--- a/x-pack/legacy/plugins/ml/server/client/elasticsearch_ml.js
+++ b/x-pack/legacy/plugins/ml/server/client/elasticsearch_ml.js
@@ -451,6 +451,18 @@ export const elasticsearchJsPlugin = (Client, config, components) => {
     method: 'POST',
   });
 
+  ml.records = ca({
+    url: {
+      fmt: '/_ml/anomaly_detectors/<%=jobId%>/results/records',
+      req: {
+        jobId: {
+          type: 'string',
+        },
+      },
+    },
+    method: 'POST',
+  });
+
   ml.buckets = ca({
     urls: [
       {
diff --git a/x-pack/legacy/plugins/ml/server/routes/anomaly_detectors.ts b/x-pack/legacy/plugins/ml/server/routes/anomaly_detectors.ts
index 927646e4f0acc..99dbdec9e945b 100644
--- a/x-pack/legacy/plugins/ml/server/routes/anomaly_detectors.ts
+++ b/x-pack/legacy/plugins/ml/server/routes/anomaly_detectors.ts
@@ -376,11 +376,57 @@ export function jobRoutes({ xpackMainPlugin, router }: RouteInitialization) {
     })
   );
 
+  /**
+   * @apiGroup AnomalyDetectors
+   *
+   * @api {post} /api/ml/anomaly_detectors/:jobId/results/records  Retrieves anomaly records for a job.
+   * @apiName GetRecords
+   * @apiDescription Retrieves anomaly records for a job.
+   *
+   * @apiParam {String} jobId Job ID.
+   *
+   * @apiSuccess {Number} count
+   * @apiSuccess {Object[]} records
+   */
+  router.post(
+    {
+      path: '/api/ml/anomaly_detectors/{jobId}/results/records',
+      validate: {
+        params: schema.object({
+          jobId: schema.string(),
+        }),
+        body: schema.object({
+          desc: schema.maybe(schema.boolean()),
+          end: schema.maybe(schema.string()),
+          exclude_interim: schema.maybe(schema.boolean()),
+          'page.from': schema.maybe(schema.number()),
+          'page.size': schema.maybe(schema.number()),
+          record_score: schema.maybe(schema.number()),
+          sort: schema.maybe(schema.string()),
+          start: schema.maybe(schema.string()),
+        }),
+      },
+    },
+    licensePreRoutingFactory(xpackMainPlugin, async (context, request, response) => {
+      try {
+        const results = await context.ml!.mlClient.callAsCurrentUser('ml.records', {
+          jobId: request.params.jobId,
+          ...request.body,
+        });
+        return response.ok({
+          body: results,
+        });
+      } catch (e) {
+        return response.customError(wrapError(e));
+      }
+    })
+  );
+
   /**
    * @apiGroup AnomalyDetectors
    *
    * @api {post} /api/ml/anomaly_detectors/:jobId/results/buckets  Obtain bucket scores for the specified job ID
-   * @apiName GetOverallBuckets
+   * @apiName GetBuckets
    * @apiDescription The get buckets API presents a chronological view of the records, grouped by bucket.
    *
    * @apiParam {String} jobId Job ID.
diff --git a/x-pack/legacy/plugins/ml/server/routes/apidoc.json b/x-pack/legacy/plugins/ml/server/routes/apidoc.json
index 946e3bd71d6c3..c5aa3e4d792fd 100644
--- a/x-pack/legacy/plugins/ml/server/routes/apidoc.json
+++ b/x-pack/legacy/plugins/ml/server/routes/apidoc.json
@@ -31,6 +31,8 @@
     "DeleteAnomalyDetectorsJob",
     "ValidateAnomalyDetector",
     "ForecastAnomalyDetector",
+    "GetRecords",
+    "GetBuckets",
     "GetOverallBuckets",
     "GetCategories",
     "FileDataVisualizer",