summaryrefslogtreecommitdiffstats
path: root/ml/ml.cc
diff options
context:
space:
mode:
authorDaniel Baumann <daniel.baumann@progress-linux.org>2023-02-06 16:11:30 +0000
committerDaniel Baumann <daniel.baumann@progress-linux.org>2023-02-06 16:11:30 +0000
commitaa2fe8ccbfcb117efa207d10229eeeac5d0f97c7 (patch)
tree941cbdd387b41c1a81587c20a6df9f0e5e0ff7ab /ml/ml.cc
parentAdding upstream version 1.37.1. (diff)
downloadnetdata-aa2fe8ccbfcb117efa207d10229eeeac5d0f97c7.tar.xz
netdata-aa2fe8ccbfcb117efa207d10229eeeac5d0f97c7.zip
Adding upstream version 1.38.0.upstream/1.38.0
Signed-off-by: Daniel Baumann <daniel.baumann@progress-linux.org>
Diffstat (limited to 'ml/ml.cc')
-rw-r--r--ml/ml.cc119
1 files changed, 78 insertions, 41 deletions
diff --git a/ml/ml.cc b/ml/ml.cc
index 1a7d6ae25..461c83baa 100644
--- a/ml/ml.cc
+++ b/ml/ml.cc
@@ -2,6 +2,7 @@
#include "Config.h"
#include "Dimension.h"
+#include "Chart.h"
#include "Host.h"
#include <random>
@@ -45,56 +46,65 @@ void ml_init(void) {
Cfg.RandomNums.push_back(Gen());
}
-void ml_new_host(RRDHOST *RH) {
+void ml_host_new(RRDHOST *RH) {
if (!ml_enabled(RH))
return;
Host *H = new Host(RH);
- RH->ml_host = static_cast<ml_host_t>(H);
-
- H->startAnomalyDetectionThreads();
+ RH->ml_host = reinterpret_cast<ml_host_t *>(H);
}
-void ml_delete_host(RRDHOST *RH) {
- Host *H = static_cast<Host *>(RH->ml_host);
+void ml_host_delete(RRDHOST *RH) {
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
if (!H)
return;
- H->stopAnomalyDetectionThreads();
-
delete H;
RH->ml_host = nullptr;
}
-void ml_new_dimension(RRDDIM *RD) {
- RRDSET *RS = RD->rrdset;
-
- Host *H = static_cast<Host *>(RD->rrdset->rrdhost->ml_host);
+void ml_chart_new(RRDSET *RS) {
+ Host *H = reinterpret_cast<Host *>(RS->rrdhost->ml_host);
if (!H)
return;
- if (static_cast<unsigned>(RD->update_every) != H->updateEvery())
+ Chart *C = new Chart(RS);
+ RS->ml_chart = reinterpret_cast<ml_chart_t *>(C);
+
+ H->addChart(C);
+}
+
+void ml_chart_delete(RRDSET *RS) {
+ Host *H = reinterpret_cast<Host *>(RS->rrdhost->ml_host);
+ if (!H)
return;
- if (simple_pattern_matches(Cfg.SP_ChartsToSkip, rrdset_name(RS)))
+ Chart *C = reinterpret_cast<Chart *>(RS->ml_chart);
+ H->removeChart(C);
+
+ delete C;
+ RS->ml_chart = nullptr;
+}
+
+void ml_dimension_new(RRDDIM *RD) {
+ Chart *C = reinterpret_cast<Chart *>(RD->rrdset->ml_chart);
+ if (!C)
return;
Dimension *D = new Dimension(RD);
- RD->ml_dimension = static_cast<ml_dimension_t>(D);
- H->addDimension(D);
+ RD->ml_dimension = reinterpret_cast<ml_dimension_t *>(D);
+ C->addDimension(D);
}
-void ml_delete_dimension(RRDDIM *RD) {
- Dimension *D = static_cast<Dimension *>(RD->ml_dimension);
+void ml_dimension_delete(RRDDIM *RD) {
+ Dimension *D = reinterpret_cast<Dimension *>(RD->ml_dimension);
if (!D)
return;
- Host *H = static_cast<Host *>(RD->rrdset->rrdhost->ml_host);
- if (!H)
- delete D;
- else
- H->removeDimension(D);
+ Chart *C = reinterpret_cast<Chart *>(RD->rrdset->ml_chart);
+ C->removeDimension(D);
+ delete D;
RD->ml_dimension = nullptr;
}
@@ -102,7 +112,7 @@ char *ml_get_host_info(RRDHOST *RH) {
nlohmann::json ConfigJson;
if (RH && RH->ml_host) {
- Host *H = static_cast<Host *>(RH->ml_host);
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
H->getConfigAsJson(ConfigJson);
} else {
ConfigJson["enabled"] = false;
@@ -115,7 +125,7 @@ char *ml_get_host_runtime_info(RRDHOST *RH) {
nlohmann::json ConfigJson;
if (RH && RH->ml_host) {
- Host *H = static_cast<Host *>(RH->ml_host);
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
H->getDetectionInfoAsJson(ConfigJson);
} else {
return nullptr;
@@ -128,7 +138,7 @@ char *ml_get_host_models(RRDHOST *RH) {
nlohmann::json ModelsJson;
if (RH && RH->ml_host) {
- Host *H = static_cast<Host *>(RH->ml_host);
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
H->getModelsAsJson(ModelsJson);
return strdup(ModelsJson.dump(2, '\t').c_str());
}
@@ -136,30 +146,57 @@ char *ml_get_host_models(RRDHOST *RH) {
return nullptr;
}
-bool ml_is_anomalous(RRDDIM *RD, double Value, bool Exists) {
- Dimension *D = static_cast<Dimension *>(RD->ml_dimension);
- if (!D)
- return false;
+void ml_start_anomaly_detection_threads(RRDHOST *RH) {
+ if (RH && RH->ml_host) {
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
+ H->startAnomalyDetectionThreads();
+ }
+}
+
+void ml_stop_anomaly_detection_threads(RRDHOST *RH) {
+ if (RH && RH->ml_host) {
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
+ H->stopAnomalyDetectionThreads(true);
+ }
+}
- return D->predict(Value, Exists);
+void ml_cancel_anomaly_detection_threads(RRDHOST *RH) {
+ if (RH && RH->ml_host) {
+ Host *H = reinterpret_cast<Host *>(RH->ml_host);
+ H->stopAnomalyDetectionThreads(false);
+ }
}
-bool ml_streaming_enabled() {
- return Cfg.StreamADCharts;
+void ml_chart_update_begin(RRDSET *RS) {
+ Chart *C = reinterpret_cast<Chart *>(RS->ml_chart);
+ if (!C)
+ return;
+
+ C->updateBegin();
}
-#if defined(ENABLE_ML_TESTS)
+void ml_chart_update_end(RRDSET *RS) {
+ Chart *C = reinterpret_cast<Chart *>(RS->ml_chart);
+ if (!C)
+ return;
+
+ C->updateEnd();
+}
-#include "gtest/gtest.h"
+bool ml_is_anomalous(RRDDIM *RD, time_t CurrT, double Value, bool Exists) {
+ Dimension *D = reinterpret_cast<Dimension *>(RD->ml_dimension);
+ if (!D)
+ return false;
-int test_ml(int argc, char *argv[]) {
- (void) argc;
- (void) argv;
+ Chart *C = reinterpret_cast<Chart *>(RD->rrdset->ml_chart);
- ::testing::InitGoogleTest(&argc, argv);
- return RUN_ALL_TESTS();
+ bool IsAnomalous = D->predict(CurrT, Value, Exists);
+ C->updateDimension(D, IsAnomalous);
+ return IsAnomalous;
}
-#endif // ENABLE_ML_TESTS
+bool ml_streaming_enabled() {
+ return Cfg.StreamADCharts;
+}
#include "ml-private.h"