blob: 52a0cd095db65677f5e859028b67195d1dea80e6 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
|
// SPDX-License-Identifier: GPL-3.0-or-later
#ifndef ML_HOST_H
#define ML_HOST_H
#include "Config.h"
#include "Dimension.h"
#include "ml-private.h"
#include "json/single_include/nlohmann/json.hpp"
namespace ml {
class RrdHost {
public:
RrdHost(RRDHOST *RH) : RH(RH) {};
RRDHOST *getRH() { return RH; }
unsigned updateEvery() { return RH->rrd_update_every; }
std::string getUUID() {
char S[UUID_STR_LEN];
uuid_unparse_lower(RH->host_uuid, S);
return S;
}
void addDimension(Dimension *D);
void removeDimension(Dimension *D);
void getConfigAsJson(nlohmann::json &Json) const;
virtual ~RrdHost() {};
protected:
RRDHOST *RH;
// Protect dimension and lock maps
std::mutex Mutex;
std::unordered_map<RRDDIM *, Dimension *> DimensionsMap;
std::unordered_map<Dimension *, std::mutex> LocksMap;
};
class TrainableHost : public RrdHost {
public:
TrainableHost(RRDHOST *RH) : RrdHost(RH) {}
void train();
void updateResourceUsage() {
std::lock_guard<std::mutex> Lock(ResourceUsageMutex);
getrusage(RUSAGE_THREAD, &ResourceUsage);
}
void getResourceUsage(struct rusage *RU) {
std::lock_guard<std::mutex> Lock(ResourceUsageMutex);
memcpy(RU, &ResourceUsage, sizeof(struct rusage));
}
void getModelsAsJson(nlohmann::json &Json);
private:
std::pair<Dimension *, Duration<double>> findDimensionToTrain(const TimePoint &NowTP);
void trainDimension(Dimension *D, const TimePoint &NowTP);
struct rusage ResourceUsage{};
std::mutex ResourceUsageMutex;
};
class DetectableHost : public TrainableHost {
public:
DetectableHost(RRDHOST *RH) : TrainableHost(RH) {}
void startAnomalyDetectionThreads();
void stopAnomalyDetectionThreads();
void getDetectionInfoAsJson(nlohmann::json &Json) const;
private:
void detect();
void detectOnce();
private:
std::thread TrainingThread;
std::thread DetectionThread;
CalculatedNumber HostAnomalyRate{0.0};
size_t NumAnomalousDimensions{0};
size_t NumNormalDimensions{0};
size_t NumTrainedDimensions{0};
size_t NumActiveDimensions{0};
};
using Host = DetectableHost;
} // namespace ml
#endif /* ML_HOST_H */
|
