|
|
#include <sqlite3.h>
//
#include <stdio.h>
#include "db_service.hpp"
//
#include "iflytop/components/sqlite_orm/sqlite_orm.hpp"
using namespace std;using namespace iflytop;using namespace iflytop::db;using namespace sqlite_orm;using namespace nlohmann;
// 数据库工具使用手册
// auto notJohn = storage.get_all<User>(where(c(&User::firstName) != "John"));
// 主键必须是int,或者不要主键
#define USER_DB_STRUCT \
USER_DB, /**/ \ make_table("users", /**/ \ make_column("id", &User::id, primary_key().autoincrement()), /**/ \ make_column("uid", &User::uid), /**/ \ make_column("passwd", &User::passwd), /**/ \ make_column("permission_level", &User::permission_level), /**/ \ make_column("visible", &User::visible))
#define SETTING_DB_STRUCT \
make_table("settings", /**/ \ make_column("id", &Setting::id, primary_key()), /**/ \ make_column("name", &Setting::name), /**/ \ make_column("name_ch", &Setting::name_ch), /**/ \ make_column("val_lower_limit", &Setting::val_lower_limit), /**/ \ make_column("val_upper_limit", &Setting::val_upper_limit), /**/ \ make_column("permission_level", &Setting::permission_level), /**/ \ make_column("val", &Setting::val))
#define DISINFECTION_RECORD_DB_STRUCT \
DISINFECTION_RECORD_DB, \ make_table("disinfection_records", /**/ \ make_column("id", &DisinfectionRecord::id, primary_key().autoincrement()), /**/ \ make_column("uuid", &DisinfectionRecord::uuid), /**/ \ make_column("uid", &DisinfectionRecord::uid), /**/ \ make_column("date", &DisinfectionRecord::date), /**/ \ make_column("loglevel", &DisinfectionRecord::loglevel), /**/ \ make_column("duration", &DisinfectionRecord::duration)), /**/ \ make_table("sensor_records", /**/ \ make_column("id", &SensorRecord::id, primary_key().autoincrement()), /**/ \ make_column("disinfection_id", &SensorRecord::disinfection_id), /**/ \ make_column("date", &SensorRecord::date), /**/ \ make_column("heating_strip", &SensorRecord::heating_strip), /**/ \ make_column("air_compressor", &SensorRecord::air_compressor), /**/ \ make_column("sprinkler_pump", &SensorRecord::sprinkler_pump), /**/ \ make_column("disinfectant_volume", &SensorRecord::disinfectant_volume), /**/ \ make_column("h2o2_1", &SensorRecord::h2o2_1), /**/ \ make_column("temp_1", &SensorRecord::temp_1), /**/ \ make_column("humid_1", &SensorRecord::humid_1), /**/ \ make_column("saturation_1", &SensorRecord::saturation_1), /**/ \ make_column("h2o2_2", &SensorRecord::h2o2_2), /**/ \ make_column("temp_2", &SensorRecord::temp_2), /**/ \ make_column("humid_2", &SensorRecord::humid_2), /**/ \ make_column("saturation_2", &SensorRecord::saturation_2), /**/ \ make_column("h2o2_3", &SensorRecord::h2o2_3), /**/ \ make_column("temp_3", &SensorRecord::temp_3), /**/ \ make_column("humid_3", &SensorRecord::humid_3), /**/ \ make_column("saturation_3", &SensorRecord::saturation_3))
DBService::DBService(/* args */) {}
void DBService::initialize() { /**
* @brief 初始化用户表 */
//
init_usr_db(); init_setting_db(); init_disinfection_record_db();
json settings = getAllSettingJson(); logger->info("settings: {}", settings.dump());}
void DBService::init_usr_db() { bool suc = false; do { try { logger->info("init user db"); auto storage = make_storage(USER_DB_STRUCT); storage.sync_schema(); auto admin = storage.get_all<User>(where(c(&User::uid) == "admin")); if (admin.size() == 0) { storage.insert<User>({-1, "admin", "9973", 0, true}); // 管理员
} auto user = storage.get_all<User>(where(c(&User::uid) == "user")); if (user.size() == 0) { storage.insert<User>({-1, "user", "0000", 3, true}); // 普通用户
} suc = true; } catch (const std::exception& e) { logger->error("init user db failed: {}", e.what()); system("rm -rf user.db"); sleep(1); } } while (!suc);}void DBService::init_setting_db() { bool suc = false;#if 0
id setting_name setting_name_ch val_upper_limit val_lower_limit permission_level val0 stoped_gs 消毒停止过氧化氢溶度 0 2000 1 10001 continued_gs 消毒继续过氧化氢溶度 0 2000 1 8002 stoped_satur 消毒停止过氧化氢相对饱和度 0 100 1 803 continued_satur 消毒继续过氧化氢相对饱和度 0 100 1 604 max_humidity 允许消毒最大湿度 0 100 1 905 drainage_pump_speed 排液蠕动泵转速 0 2000 2 5006 injection_pump_speed 喷射蠕动泵转速 0 2000 2 500#endif
do { try { logger->info("init setting db"); auto storage = make_storage(SETTING_DB, SETTING_DB_STRUCT); storage.sync_schema();
if (storage.get_all<Setting>(where(c(&Setting::id) == 1)).size() == 0) //
storage.insert(Setting{1, "stoped_gs", "消毒停止过氧化氢溶度", 0, 2000, 0, 1800});
if (storage.get_all<Setting>(where(c(&Setting::id) == 2)).size() == 0) storage.insert(Setting{2, "continued_gs", "消毒继续过氧化氢溶度", 0, 2000, 0, 1500});
if (storage.get_all<Setting>(where(c(&Setting::id) == 3)).size() == 0) storage.insert(Setting{3, "stoped_satur", "消毒停止过氧化氢相对饱和度", 0, 100, 0, 85});
if (storage.get_all<Setting>(where(c(&Setting::id) == 4)).size() == 0) storage.insert(Setting{4, "continued_satur", "消毒继续过氧化氢相对饱和度", 0, 100, 0, 70});
if (storage.get_all<Setting>(where(c(&Setting::id) == 5)).size() == 0) //
storage.insert(Setting{5, "max_humidity", "允许消毒最大湿度", 0, 100, 0, 90});
if (storage.get_all<Setting>(where(c(&Setting::id) == 6)).size() == 0) storage.insert(Setting{6, "drainage_pump_speed", "排液蠕动泵转速", 0, 40, 0, 40}); // g/min
if (storage.get_all<Setting>(where(c(&Setting::id) == 7)).size() == 0) storage.insert(Setting{7, "injection_pump_speed", "喷射蠕动泵转速", 0, 40, 0, 40}); // g/min
if (storage.get_all<Setting>(where(c(&Setting::id) == 8)).size() == 0) //
storage.insert(Setting{8, "pre_heat_time_s", "预热时间", 0, 600, 0, 120});
if (storage.get_all<Setting>(where(c(&Setting::id) == 9)).size() == 0) //
storage.insert(Setting{9, "stoped_humi", "消毒停止相对湿度", 0, 100, 0, 85});
if (storage.get_all<Setting>(where(c(&Setting::id) == 10)).size() == 0) //
storage.insert(Setting{10, "continued_humi", "消毒继续相对湿度", 0, 100, 0, 70});
suc = true; } catch (const std::exception& e) { logger->error("init setting db failed: {}", e.what()); system("rm -rf setting.db"); sleep(1); } } while (!suc);}
void DBService::init_disinfection_record_db() { bool suc = false; do { try { logger->info("init disinfection record db"); auto storage = make_storage(DISINFECTION_RECORD_DB_STRUCT); storage.sync_schema(); suc = true; } catch (const std::exception& e) { logger->error("init disinfection record db failed: {}", e.what()); system("rm -rf disinfection_record.db"); sleep(1); } } while (!suc);}
list<shared_ptr<User>> DBService::getAllUser() { list<shared_ptr<User>> users; auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema(); auto all = usertable.get_all<User>(); for (auto& u : all) { users.push_back(make_shared<User>(u)); } return users;}void DBService::addUser(string uid, string passwd, int permission_level) { auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema(); logger->info("add user: {} {} {}", uid, passwd, permission_level); usertable.insert(User{-1, uid, passwd, permission_level, true});}void DBService::delUser(int id) { auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema(); /**
* @brief find admin user */ auto admin = usertable.get_all<User>(where(c(&User::uid) == "admin")); ZCHECK(admin.size() == 1, "admin user not found"); if (admin[0].id == id) { logger->error("can not delete admin user"); return; } auto remove_user = usertable.get_all<User>(where(c(&User::id) == id)); if (remove_user.size() == 0) { logger->error("remove user fail, user not found"); return; } logger->info("delete user: {}:{}", id, remove_user[0].uid); usertable.remove<User>(where(c(&User::id) == id));}
void DBService::updateUserPermissionLevel(int id, int permission_level) { auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema();
auto user = usertable.get_all<User>(where(c(&User::id) == id)); if (user.size() == 0) { logger->error("update user permission level fail, user not found"); return; } logger->info("update user permission level: {} {} -> {}", id, user[0].permission_level, permission_level); user[0].permission_level = permission_level; usertable.update(user[0]);}
json DBService::getAllUserJson() { json j_users; auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema(); auto all = usertable.get_all<User>(); for (auto& u : all) { json j_user; j_user["uid"] = u.uid; j_user["passwd"] = u.passwd; j_user["permission_level"] = u.permission_level; j_user["visible"] = u.visible; j_users.push_back(j_user); } return j_users;}
shared_ptr<User> DBService::getUser(string uid) { auto usertable = make_storage(USER_DB_STRUCT); usertable.sync_schema(); auto user = usertable.get_all<User>(where(c(&User::uid) == uid)); if (user.size() == 0) { return nullptr; } return make_shared<User>(user[0]);}
list<shared_ptr<db::Setting>> DBService::getAllSetting() { list<shared_ptr<db::Setting>> settings; auto settingtable = make_storage(SETTING_DB, SETTING_DB_STRUCT); settingtable.sync_schema(); auto all = settingtable.get_all<Setting>(); for (auto& s : all) { settings.push_back(make_shared<Setting>(s)); } return settings;}json DBService::getAllSettingJson() { json j_settings; auto settingtable = make_storage(SETTING_DB, SETTING_DB_STRUCT); settingtable.sync_schema(); auto all = settingtable.get_all<Setting>(); for (auto& s : all) { json j_setting; j_setting["id"] = s.id; j_setting["name"] = s.name; j_setting["name_ch"] = s.name_ch; j_setting["val"] = s.val; j_setting["val_lower_limit"] = s.val_lower_limit; j_setting["val_upper_limit"] = s.val_upper_limit; j_setting["permission_level"] = s.permission_level; j_setting["val"] = s.val; j_settings.push_back(j_setting); } return j_settings;}
bool DBService::setSettingVal(int id, int val) { auto settingtable = make_storage(SETTING_DB, SETTING_DB_STRUCT); settingtable.sync_schema(); auto setting = settingtable.get_all<Setting>(where(c(&Setting::id) == id)); if (setting.size() == 0) { return false; } if (setting[0].val < setting[0].val_lower_limit || setting[0].val > setting[0].val_upper_limit) { return false; } setting[0].val = val; settingtable.update(setting[0]); return true;}bool DBService::setSettingVal(string setting_name, int val) { logger->info("set setting val: {} {}", setting_name, val); auto settingtable = make_storage(SETTING_DB, SETTING_DB_STRUCT); settingtable.sync_schema(); auto setting = settingtable.get_all<Setting>(where(c(&Setting::name) == setting_name)); if (setting.size() == 0) { logger->error("set setting val failed: {} not found", setting_name); return false; } if (setting[0].val < setting[0].val_lower_limit || setting[0].val > setting[0].val_upper_limit) { logger->error("set setting val failed: {} out of range", setting_name); return false; } setting[0].val = val; Setting s = setting[0]; settingtable.update(s); settingtable.sync_schema(); return true;}
int DBService::getSettingVal(string name) { auto settingtable = make_storage(SETTING_DB, SETTING_DB_STRUCT); settingtable.sync_schema(); auto setting = settingtable.get_all<Setting>(where(c(&Setting::name) == name)); if (setting.size() == 0) { return -1; } return setting[0].val;}
json DBService::getAllRecords(string disinfection_id) { json j_records; auto recordtable = make_storage(DISINFECTION_RECORD_DB_STRUCT);
// int id;
// string disinfection_id;
// string date;
// int heating_strip;
// int air_compressor;
// int sprinkler_pump;
// int disinfectant_volume;
// int h2o2_1;
// int temp_1;
// int humid_1;
// int saturation_1;
// int h2o2_2;
// int temp_2;
// int humid_2;
// int saturation_2;
// int h2o2_3;
// int temp_3;
// int humid_3;
// int saturation_3;
recordtable.sync_schema(); auto all = recordtable.get_all<SensorRecord>(where(c(&SensorRecord::disinfection_id) == disinfection_id)); for (auto& r : all) { json j_record; j_record["id"] = r.id; j_record["disinfection_id"] = r.disinfection_id; j_record["date"] = r.date; j_record["heating_strip"] = r.heating_strip; j_record["air_compressor"] = r.air_compressor; j_record["sprinkler_pump"] = r.sprinkler_pump; j_record["disinfectant_volume"] = r.disinfectant_volume; j_record["h2o2_1"] = r.h2o2_1; j_record["temp_1"] = r.temp_1; j_record["humid_1"] = r.humid_1; j_record["saturation_1"] = r.saturation_1; j_record["h2o2_2"] = r.h2o2_2; j_record["temp_2"] = r.temp_2; j_record["humid_2"] = r.humid_2; j_record["saturation_2"] = r.saturation_2; j_record["h2o2_3"] = r.h2o2_3; j_record["temp_3"] = r.temp_3; j_record["humid_3"] = r.humid_3; j_record["saturation_3"] = r.saturation_3; j_records.push_back(j_record); } return j_records;}void DBService::insertSensorRecord(shared_ptr<db::SensorRecord> record) { if (!record) { logger->error("insert sensor record failed: record is null"); return; } auto recordtable = make_storage(DISINFECTION_RECORD_DB_STRUCT); recordtable.sync_schema(); recordtable.insert(*record);}void DBService::insertDisinfectionRecord(shared_ptr<db::DisinfectionRecord> record) { if (!record) { logger->error("insert disinfection record failed: record is null"); return; } auto recordtable = make_storage(DISINFECTION_RECORD_DB_STRUCT); recordtable.sync_schema(); recordtable.insert(*record);}void DBService::setDisinfectionRecordDuration(string disinfection_id, int duration) { auto recordtable = make_storage(DISINFECTION_RECORD_DB_STRUCT); recordtable.sync_schema(); auto record = recordtable.get_all<DisinfectionRecord>(where(c(&DisinfectionRecord::uuid) == disinfection_id)); if (record.size() == 0) { logger->error("set disinfection record duration failed: {} not found", disinfection_id); return; } record[0].duration = duration; recordtable.update(record[0]);}
|
