| | |
| | | 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; |
| | | } |
| | | } |
| | | */ |
| | | } |
| | | |
| | | // 添加报警信息 |
| | | bool AlarmManager::addAlarm(const AlarmData& alarmData) { |
| | | bool AlarmManager::addAlarm(const AlarmData& alarmData, int& alarmEventId) { |
| | | if (!m_pDB) { |
| | | return false; |
| | | } |
| | | |
| | | // 构建插入查询 |
| | | std::ostringstream query; |
| | | query << "INSERT INTO alarms (id, severity_level, device_id, unit_id, description, start_time, end_time) VALUES (" |
| | | << alarmData.nId << ", " // 错误码 |
| | | << alarmData.nSeverityLevel << ", " // 报警等级 |
| | | << alarmData.nDeviceId << ", " // 设备ID |
| | | << alarmData.nUnitId << ", '" // 单元ID |
| | | << alarmData.strDescription << "', '" // 描述 |
| | | << alarmData.strStartTime << "', '" // 开始时间 |
| | | << alarmData.strEndTime << "')"; // 结束时间 |
| | | |
| | | // 使用锁保护多线程安全 |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | return m_pDB->executeQuery(query.str()); |
| | | #if 0 |
| | | // 开始事务 |
| | | m_pDB->executeQuery("BEGIN TRANSACTION;"); |
| | | |
| | | // 构建插入查询 |
| | | std::ostringstream query; |
| | | query << "INSERT INTO alarms (id, severity_level, device_id, unit_id, description, start_time, end_time) VALUES (" |
| | | << alarmData.nId << ", " // 错误码 |
| | | << alarmData.nSeverityLevel << ", " // 报警等级 |
| | | << alarmData.nDeviceId << ", " // 设备ID |
| | | << alarmData.nUnitId << ", '" // 单元ID |
| | | << alarmData.strDescription << "', '" // 描述 |
| | | << alarmData.strStartTime << "', '" // 开始时间 |
| | | << alarmData.strEndTime << "')"; // 结束时间 |
| | | |
| | | // 使用锁保护多线程安全 |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | |
| | | // 执行插入查询 |
| | | bool result = m_pDB->executeQuery(query.str()); |
| | | if (result) { |
| | | alarmEventId = getLastInsertId(); |
| | | m_alarmCache[alarmEventId] = alarmData; |
| | | } |
| | | |
| | | // 提交事务 |
| | | m_pDB->executeQuery("COMMIT;"); |
| | | |
| | | return result; |
| | | #else |
| | | // 构建插入查询并使用 RETURNING 获取插入后的 alarm_event_id |
| | | std::ostringstream query; |
| | | query << "INSERT INTO alarms (id, severity_level, device_id, unit_id, description, start_time, end_time) " |
| | | << "VALUES (" << alarmData.nId << ", " << alarmData.nSeverityLevel << ", " << alarmData.nDeviceId << ", " |
| | | << alarmData.nUnitId << ", '" << alarmData.strDescription << "', '" << alarmData.strStartTime << "', '" |
| | | << alarmData.strEndTime << "') RETURNING alarm_event_id;"; |
| | | |
| | | // 使用锁保护多线程安全 |
| | | std::lock_guard<std::mutex> lock(m_mutex); |
| | | |
| | | // 执行查询并获取结果 |
| | | auto results = m_pDB->fetchResults(query.str()); |
| | | if (!results.empty() && !results[0].empty()) { |
| | | try { |
| | | // 提取并转换 alarm_event_id |
| | | alarmEventId = std::stoi(results[0][0]); |
| | | // 将插入的报警数据添加到缓存 |
| | | m_mapCache[alarmEventId] = alarmData; |
| | | return true; |
| | | } |
| | | catch (const std::exception& e) { |
| | | std::cerr << "Error parsing alarm_event_id: " << e.what() << std::endl; |
| | | return false; |
| | | } |
| | | } |
| | | |
| | | return false; |
| | | #endif |
| | | } |
| | | |
| | | // 查询所有报警数据 |
| | |
| | | return result[0][0]; // 返回查询到的单元名称 |
| | | } |
| | | |
| | | // 获取最近插入的 alarm_event_id |
| | | int AlarmManager::getLastInsertId() { |
| | | std::string query = "SELECT last_insert_rowid();"; |
| | | auto results = m_pDB->fetchResults(query); |
| | | |
| | | if (results.empty() || results[0].empty()) { |
| | | return -1; |
| | | } |
| | | |
| | | // 从查询结果中获取最后插入的 ID |
| | | int lastInsertId = std::stoi(results[0][0]); |
| | | 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); |
| | |
| | | auto it = m_mapAlarm.find(alarmID); |
| | | if (it != m_mapAlarm.end()) { |
| | | alarms.push_back(it->second); |
| | | } |
| | | } |
| | | else { |
| | | std::cerr << "未找到 AlarmID: " << alarmID << std::endl; |
| | | } |