SmartLockerTools/Database/Database.cpp

72 lines
2.5 KiB
C++
Raw Normal View History

#include "Database.h"
#include "BoostLog.h"
#include "sqlite3.h"
bool Database::addPalmFeature(const PalmFeature &palm) {
sqlite3_stmt *statement = nullptr;
constexpr const char *sql = "INSERT INTO palm_feature (username,feature) VALUES (?,?);";
int status = sqlite3_prepare_v2(m_sqlite, sql, -1, &statement, 0);
if (status != SQLITE_OK) {
LOG(error) << "Failed to prepare statement: " << sqlite3_errmsg(m_sqlite);
return false;
}
sqlite3_bind_text(statement, 1, palm.username.c_str(), -1, SQLITE_TRANSIENT);
sqlite3_bind_blob(statement, 2, palm.feature.data(), palm.feature.size(), SQLITE_TRANSIENT);
status = sqlite3_step(statement);
if (status != SQLITE_DONE) {
LOG(error) << "Failed to execute statement: " << sqlite3_errmsg(m_sqlite);
return false;
}
sqlite3_finalize(statement);
return true;
}
2024-05-24 10:23:05 +08:00
PalmFeatures Database::palmFeatures() const {
PalmFeatures ret;
sqlite3_stmt *statement = nullptr;
constexpr const char *sql = "SELECT * FROM palm_feature";
if (sqlite3_prepare_v2(m_sqlite, sql, -1, &statement, NULL) != SQLITE_OK) {
LOG(error) << "Can't prepare statement: " << sqlite3_errmsg(m_sqlite);
return ret;
}
while (sqlite3_step(statement) == SQLITE_ROW) {
PalmFeature feature;
feature.id = sqlite3_column_int64(statement, 0);
feature.username = reinterpret_cast<const char *>(sqlite3_column_text(statement, 1));
auto data = reinterpret_cast<const uint8_t *>(sqlite3_column_blob(statement, 2));
int size = sqlite3_column_bytes(statement, 2);
feature.feature = std::vector<uint8_t>(data, data + size);
ret.push_back(feature);
}
return ret;
}
bool Database::open(const std::string &path) {
if (sqlite3_open(path.c_str(), &m_sqlite) != SQLITE_OK) {
LOG(error) << "sqlite3_open() failed: " << sqlite3_errmsg(m_sqlite);
return false;
}
initializeTables();
return true;
}
void Database::initializeTables() {
constexpr const char *sql = R"(
CREATE TABLE IF NOT EXISTS palm_feature(
id INTEGER PRIMARY KEY,
username TEXT NOT NULL,
feature BLOB
);
)";
char *errorMessage = nullptr;
int status = sqlite3_exec(m_sqlite, sql, 0, 0, &errorMessage);
if (status != SQLITE_OK) {
LOG(error) << "SQL error: " << errorMessage;
sqlite3_free(errorMessage);
}
}