| | |
| | | } |
| | | } |
| | | |
| | | // 设置数据库连接 |
| | | void AlarmManager::setDatabase(BL::Database* db) { |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | m_pDB = db; |
| | | } |
| | | |
| | | // 初始化报警表 |
| | | bool AlarmManager::initAlarmTable() { |
| | | char path[MAX_PATH]; |
| | |
| | | unit_id TEXT NOT NULL, |
| | | description TEXT NOT NULL, |
| | | start_time DATETIME NOT NULL, |
| | | end_time DATETIME NOT NULL, |
| | | end_time DATETIME, |
| | | FOREIGN KEY (device_id) REFERENCES devices(device_id), |
| | | FOREIGN KEY (unit_id) REFERENCES units(unit_id) |
| | | ) |
| | |
| | | |
| | | // 插入单元数据 |
| | | for (int i = 1; i <= 3; ++i) { |
| | | for (int j = 1; j <= 3; ++j) { |
| | | for (int j = 0; j <= 3; ++j) { |
| | | int unitId = j; |
| | | std::string deviceId = std::to_string(i); |
| | | std::string unitName = "Unit" + std::to_string(j); |
| | |
| | | } |
| | | } |
| | | |
| | | /* |
| | | // 初始化随机数生成器 |
| | | std::random_device rd; |
| | | std::mt19937 gen(rd()); |
| | |
| | | std::cerr << "Failed to insert alarm data." << std::endl; |
| | | } |
| | | } |
| | | */ |
| | | } |
| | | |
| | | // 添加报警信息 |
| | |
| | | |
| | | return result; |
| | | #else |
| | | for (AlarmDataMap::const_iterator it = m_mapCache.begin(); it != m_mapCache.end(); ++it) { |
| | | const AlarmData& alarm = it->second; |
| | | if (alarm.nId == alarmData.nId && |
| | | alarm.nDeviceId == alarmData.nDeviceId && |
| | | alarm.nUnitId == alarmData.nUnitId) { |
| | | |
| | | alarmEventId = it->first; |
| | | return false; |
| | | } |
| | | } |
| | | |
| | | // 构建插入查询并使用 RETURNING 获取插入后的 alarm_event_id |
| | | std::ostringstream query; |
| | | query << "INSERT INTO alarms (id, severity_level, device_id, unit_id, description, start_time, end_time) " |
| | |
| | | return lastInsertId; |
| | | } |
| | | |
| | | // 通过 alarm_event_id 解除报警(更新结束时间) |
| | | bool AlarmManager::clearAlarmByEventId(int alarmEventId, const std::string& endTime) { |
| | | if (!m_pDB) { |
| | | return false; |
| | | } |
| | | |
| | | bool result = true; |
| | | auto it = m_mapCache.find(alarmEventId); |
| | | if (it != m_mapCache.end()) { |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | |
| | | std::ostringstream query; |
| | | query << "UPDATE alarms SET end_time = '" << endTime << "' WHERE alarm_event_id = " << alarmEventId << ";"; |
| | | bool result = m_pDB->executeQuery(query.str()); |
| | | if (result) { |
| | | m_mapCache.erase(it); |
| | | } |
| | | } |
| | | |
| | | return result; |
| | | } |
| | | |
| | | // 通过多个属性查找并解除报警(更新结束时间) |
| | | bool AlarmManager::clearAlarmByAttributes(int nId, int nDeviceId, int nUnitId, const std::string& endTime) { |
| | | if (!m_pDB) { |
| | | return false; |
| | | } |
| | | |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | |
| | | // 先在缓存中查找匹配的 alarm_event_id |
| | | int alarmEventId = -1; |
| | | for (AlarmDataMap::const_iterator it = m_mapCache.begin(); it != m_mapCache.end(); ++it) { |
| | | const AlarmData& alarm = it->second; |
| | | if (alarm.nId == nId && |
| | | alarm.nDeviceId == nDeviceId && |
| | | alarm.nUnitId == nUnitId) { |
| | | |
| | | alarmEventId = it->first; |
| | | break; |
| | | } |
| | | } |
| | | |
| | | // 如果没找到匹配的记录,则直接返回 false |
| | | if (alarmEventId == -1) { |
| | | return false; |
| | | } |
| | | |
| | | // 构建 SQL 语句,使用找到的 alarm_event_id 来更新结束时间 |
| | | std::ostringstream query; |
| | | query << "UPDATE alarms SET end_time = '" << endTime << "' WHERE alarm_event_id = " << alarmEventId << ";"; |
| | | bool result = m_pDB->executeQuery(query.str()); |
| | | if (result) { |
| | | m_mapCache.erase(alarmEventId); |
| | | } |
| | | |
| | | return result; |
| | | } |
| | | |
| | | // 读取报警文件 |
| | | bool AlarmManager::readAlarmFile(const std::string& filename) { |
| | | std::ifstream file(filename); |