From 8074e0a2e17df7b6bd3fdae54707399d8352bad4 Mon Sep 17 00:00:00 2001
From: rozetko <rozetko@hastic.io>
Date: Mon, 20 May 2019 16:26:21 +0300
Subject: [PATCH] Not-ending learning for anomaly detector #665 (#674)

---
 server/src/controllers/analytics_controller.ts | 5 +++--
 server/src/models/analytic_units/types.ts      | 1 +
 2 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/server/src/controllers/analytics_controller.ts b/server/src/controllers/analytics_controller.ts
index d80fff5..0a146b5 100644
--- a/server/src/controllers/analytics_controller.ts
+++ b/server/src/controllers/analytics_controller.ts
@@ -290,6 +290,8 @@ export async function runLearning(id: AnalyticUnit.AnalyticUnitId, from?: number
     if(result.status !== AnalyticUnit.AnalyticUnitStatus.SUCCESS) {
       throw new Error(result.error);
     }
+    // TODO: rename SUCCESS to something better
+    await AnalyticUnit.setStatus(id, AnalyticUnit.AnalyticUnitStatus.SUCCESS);
     await AnalyticUnitCache.setData(id, result.payload.cache);
   } catch (err) {
     const message = err.message || JSON.stringify(err);
@@ -331,8 +333,7 @@ export async function runDetect(id: AnalyticUnit.AnalyticUnitId, from?: number,
       { detector, analyticUnitType, lastDetectionTime: unit.lastDetectionTime, data, cache: oldCache }
     );
     console.log(`run task, id:${id}`);
-    // TODO: status: detection
-    await AnalyticUnit.setStatus(id, AnalyticUnit.AnalyticUnitStatus.LEARNING);
+    await AnalyticUnit.setStatus(id, AnalyticUnit.AnalyticUnitStatus.DETECTION);
     const result = await runTask(task);
 
     if(result.status === AnalyticUnit.AnalyticUnitStatus.FAILED) {
diff --git a/server/src/models/analytic_units/types.ts b/server/src/models/analytic_units/types.ts
index 27be73f..58ab20a 100644
--- a/server/src/models/analytic_units/types.ts
+++ b/server/src/models/analytic_units/types.ts
@@ -6,6 +6,7 @@ export enum AnalyticUnitStatus {
   READY = 'READY',
   PENDING = 'PENDING',
   LEARNING = 'LEARNING',
+  DETECTION = 'DETECTION',
   SUCCESS = 'SUCCESS',
   FAILED = 'FAILED'
 };