summaryrefslogtreecommitdiffstats
path: root/ml/Database.cc
diff options
context:
space:
mode:
Diffstat (limited to '')
-rw-r--r--ml/Database.cc127
1 files changed, 127 insertions, 0 deletions
diff --git a/ml/Database.cc b/ml/Database.cc
new file mode 100644
index 000000000..06d0cdecb
--- /dev/null
+++ b/ml/Database.cc
@@ -0,0 +1,127 @@
+// SPDX-License-Identifier: GPL-3.0-or-later
+
+#include "Database.h"
+
+const char *ml::Database::SQL_CREATE_ANOMALIES_TABLE =
+ "CREATE TABLE IF NOT EXISTS anomaly_events( "
+ " anomaly_detector_name text NOT NULL, "
+ " anomaly_detector_version int NOT NULL, "
+ " host_id text NOT NULL, "
+ " after int NOT NULL, "
+ " before int NOT NULL, "
+ " anomaly_event_info text, "
+ " PRIMARY KEY( "
+ " anomaly_detector_name, anomaly_detector_version, "
+ " host_id, after, before "
+ " ) "
+ ");";
+
+const char *ml::Database::SQL_INSERT_ANOMALY =
+ "INSERT INTO anomaly_events( "
+ " anomaly_detector_name, anomaly_detector_version, "
+ " host_id, after, before, anomaly_event_info) "
+ "VALUES (?1, ?2, ?3, ?4, ?5, ?6);";
+
+const char *ml::Database::SQL_SELECT_ANOMALY =
+ "SELECT anomaly_event_info FROM anomaly_events WHERE"
+ " anomaly_detector_name == ?1 AND"
+ " anomaly_detector_version == ?2 AND"
+ " host_id == ?3 AND"
+ " after == ?4 AND"
+ " before == ?5;";
+
+const char *ml::Database::SQL_SELECT_ANOMALY_EVENTS =
+ "SELECT after, before FROM anomaly_events WHERE"
+ " anomaly_detector_name == ?1 AND"
+ " anomaly_detector_version == ?2 AND"
+ " host_id == ?3 AND"
+ " after >= ?4 AND"
+ " before <= ?5;";
+
+using namespace ml;
+
+bool Statement::prepare(sqlite3 *Conn) {
+ if (!Conn)
+ return false;
+
+ if (ParsedStmt)
+ return true;
+
+ int RC = sqlite3_prepare_v2(Conn, RawStmt, -1, &ParsedStmt, nullptr);
+ if (RC == SQLITE_OK)
+ return true;
+
+ std::string Msg = "Statement \"%s\" preparation failed due to \"%s\"";
+ error(Msg.c_str(), RawStmt, sqlite3_errstr(RC));
+
+ return false;
+}
+
+bool Statement::bindValue(size_t Pos, const std::string &Value) {
+ int RC = sqlite3_bind_text(ParsedStmt, Pos, Value.c_str(), -1, SQLITE_TRANSIENT);
+ if (RC == SQLITE_OK)
+ return true;
+
+ error("Failed to bind text '%s' (pos = %zu) in statement '%s'.", Value.c_str(), Pos, RawStmt);
+ return false;
+}
+
+bool Statement::bindValue(size_t Pos, const int Value) {
+ int RC = sqlite3_bind_int(ParsedStmt, Pos, Value);
+ if (RC == SQLITE_OK)
+ return true;
+
+ error("Failed to bind integer %d (pos = %zu) in statement '%s'.", Value, Pos, RawStmt);
+ return false;
+}
+
+bool Statement::resetAndClear(bool Ret) {
+ int RC = sqlite3_reset(ParsedStmt);
+ if (RC != SQLITE_OK) {
+ error("Could not reset statement: '%s'", RawStmt);
+ return false;
+ }
+
+ RC = sqlite3_clear_bindings(ParsedStmt);
+ if (RC != SQLITE_OK) {
+ error("Could not clear bindings in statement: '%s'", RawStmt);
+ return false;
+ }
+
+ return Ret;
+}
+
+Database::Database(const std::string &Path) {
+ // Get sqlite3 connection handle.
+ int RC = sqlite3_open(Path.c_str(), &Conn);
+ if (RC != SQLITE_OK) {
+ std::string Msg = "Failed to initialize ML DB at %s, due to \"%s\"";
+ error(Msg.c_str(), Path.c_str(), sqlite3_errstr(RC));
+
+ sqlite3_close(Conn);
+ Conn = nullptr;
+ return;
+ }
+
+ // Create anomaly events table if it does not exist.
+ char *ErrMsg;
+ RC = sqlite3_exec(Conn, SQL_CREATE_ANOMALIES_TABLE, nullptr, nullptr, &ErrMsg);
+ if (RC == SQLITE_OK)
+ return;
+
+ error("SQLite error during database initialization, rc = %d (%s)", RC, ErrMsg);
+ error("SQLite failed statement: %s", SQL_CREATE_ANOMALIES_TABLE);
+
+ sqlite3_free(ErrMsg);
+ sqlite3_close(Conn);
+ Conn = nullptr;
+}
+
+Database::~Database() {
+ if (!Conn)
+ return;
+
+ int RC = sqlite3_close(Conn);
+ if (RC != SQLITE_OK)
+ error("Could not close connection properly (rc=%d)", RC);
+}