123
7 days ago
parent
76cd4c9aa6
commit
73c25cb34b
Binary file not shown.
@ -0,0 +1,203 @@
|
|||||||
|
/**
|
||||||
|
* @file EnemyStatsDialog.h
|
||||||
|
* @brief 敌情统计对话框定义
|
||||||
|
* @author Qt UI Optimizer
|
||||||
|
* @date 2024-07-08
|
||||||
|
* @version 1.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
#ifndef ENEMYSTATS_DIALOG_H
|
||||||
|
#define ENEMYSTATS_DIALOG_H
|
||||||
|
|
||||||
|
#include <QDialog>
|
||||||
|
#include <QVBoxLayout>
|
||||||
|
#include <QHBoxLayout>
|
||||||
|
#include <QTableWidget>
|
||||||
|
#include <QTableWidgetItem>
|
||||||
|
#include <QHeaderView>
|
||||||
|
#include <QLabel>
|
||||||
|
#include <QPushButton>
|
||||||
|
#include <QGroupBox>
|
||||||
|
#include <QTimer>
|
||||||
|
#include <QDateTime>
|
||||||
|
#include <QMessageBox>
|
||||||
|
#include <QFileDialog>
|
||||||
|
#include <QTextStream>
|
||||||
|
#include <QApplication>
|
||||||
|
#include <QDesktopWidget>
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @struct EnemyInfo
|
||||||
|
* @brief 敌情信息结构体
|
||||||
|
*/
|
||||||
|
struct EnemyInfo {
|
||||||
|
QString id; ///< 敌人ID/编号
|
||||||
|
double longitude; ///< 经度坐标
|
||||||
|
double latitude; ///< 纬度坐标
|
||||||
|
QString threatLevel; ///< 威胁等级
|
||||||
|
QDateTime discoveryTime; ///< 发现时间
|
||||||
|
QString enemyType; ///< 敌人类型
|
||||||
|
QString status; ///< 状态
|
||||||
|
};
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @class EnemyStatsDialog
|
||||||
|
* @brief 敌情统计对话框
|
||||||
|
*
|
||||||
|
* 提供敌情数据的表格显示、统计分析和数据导出功能
|
||||||
|
*/
|
||||||
|
class EnemyStatsDialog : public QDialog
|
||||||
|
{
|
||||||
|
Q_OBJECT
|
||||||
|
|
||||||
|
public:
|
||||||
|
/**
|
||||||
|
* @brief 构造函数
|
||||||
|
* @param parent 父组件指针
|
||||||
|
*/
|
||||||
|
explicit EnemyStatsDialog(QWidget *parent = nullptr);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 析构函数
|
||||||
|
*/
|
||||||
|
~EnemyStatsDialog();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 添加敌情信息
|
||||||
|
* @param enemy 敌情信息结构体
|
||||||
|
*/
|
||||||
|
void addEnemyInfo(const EnemyInfo &enemy);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 更新敌情信息
|
||||||
|
* @param id 敌人ID
|
||||||
|
* @param enemy 更新后的敌情信息
|
||||||
|
*/
|
||||||
|
void updateEnemyInfo(const QString &id, const EnemyInfo &enemy);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 删除敌情信息
|
||||||
|
* @param id 敌人ID
|
||||||
|
*/
|
||||||
|
void removeEnemyInfo(const QString &id);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 清空所有敌情信息
|
||||||
|
*/
|
||||||
|
void clearAllEnemies();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 获取敌情总数
|
||||||
|
* @return 敌情总数
|
||||||
|
*/
|
||||||
|
int getEnemyCount() const;
|
||||||
|
|
||||||
|
signals:
|
||||||
|
/**
|
||||||
|
* @brief 敌情数据更新信号
|
||||||
|
* @param totalCount 敌情总数
|
||||||
|
* @param highThreatCount 高威胁敌情数量
|
||||||
|
*/
|
||||||
|
void enemyDataUpdated(int totalCount, int highThreatCount);
|
||||||
|
|
||||||
|
private slots:
|
||||||
|
/**
|
||||||
|
* @brief 刷新数据槽函数
|
||||||
|
*/
|
||||||
|
void onRefreshData();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 导出数据槽函数
|
||||||
|
*/
|
||||||
|
void onExportData();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 表格行选择变化槽函数
|
||||||
|
*/
|
||||||
|
void onTableSelectionChanged();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 自动刷新定时器槽函数
|
||||||
|
*/
|
||||||
|
void onAutoRefresh();
|
||||||
|
|
||||||
|
private:
|
||||||
|
/**
|
||||||
|
* @brief 设置UI界面
|
||||||
|
*/
|
||||||
|
void setupUI();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 设置表格
|
||||||
|
*/
|
||||||
|
void setupTable();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 设置统计面板
|
||||||
|
*/
|
||||||
|
void setupStatsPanel();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 应用样式表
|
||||||
|
*/
|
||||||
|
void applyStyles();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 连接信号槽
|
||||||
|
*/
|
||||||
|
void connectSignals();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 更新统计信息
|
||||||
|
*/
|
||||||
|
void updateStatistics();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 加载测试数据
|
||||||
|
*/
|
||||||
|
void loadTestData();
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 获取威胁等级颜色
|
||||||
|
* @param threatLevel 威胁等级
|
||||||
|
* @return 对应的颜色
|
||||||
|
*/
|
||||||
|
QColor getThreatLevelColor(const QString &threatLevel);
|
||||||
|
|
||||||
|
/**
|
||||||
|
* @brief 格式化坐标显示
|
||||||
|
* @param longitude 经度
|
||||||
|
* @param latitude 纬度
|
||||||
|
* @return 格式化后的坐标字符串
|
||||||
|
*/
|
||||||
|
QString formatCoordinates(double longitude, double latitude);
|
||||||
|
|
||||||
|
// UI组件
|
||||||
|
QVBoxLayout *m_mainLayout; ///< 主布局
|
||||||
|
QHBoxLayout *m_contentLayout; ///< 内容布局
|
||||||
|
|
||||||
|
// 表格组件
|
||||||
|
QGroupBox *m_tableGroup; ///< 表格分组框
|
||||||
|
QTableWidget *m_enemyTable; ///< 敌情表格
|
||||||
|
|
||||||
|
// 统计面板组件
|
||||||
|
QGroupBox *m_statsGroup; ///< 统计分组框
|
||||||
|
QLabel *m_totalCountLabel; ///< 总数标签
|
||||||
|
QLabel *m_highThreatLabel; ///< 高威胁数量标签
|
||||||
|
QLabel *m_mediumThreatLabel; ///< 中威胁数量标签
|
||||||
|
QLabel *m_lowThreatLabel; ///< 低威胁数量标签
|
||||||
|
QLabel *m_lastUpdateLabel; ///< 最后更新时间标签
|
||||||
|
|
||||||
|
// 操作按钮
|
||||||
|
QPushButton *m_refreshBtn; ///< 刷新按钮
|
||||||
|
QPushButton *m_exportBtn; ///< 导出按钮
|
||||||
|
QPushButton *m_closeBtn; ///< 关闭按钮
|
||||||
|
|
||||||
|
// 数据存储
|
||||||
|
QList<EnemyInfo> m_enemyList; ///< 敌情信息列表
|
||||||
|
|
||||||
|
// 定时器
|
||||||
|
QTimer *m_autoRefreshTimer; ///< 自动刷新定时器
|
||||||
|
};
|
||||||
|
|
||||||
|
#endif // ENEMYSTATS_DIALOG_H
|
||||||
@ -0,0 +1,430 @@
|
|||||||
|
/**
|
||||||
|
* @file EnemyDatabase.cpp
|
||||||
|
* @brief 敌情数据库管理类实现
|
||||||
|
* @author Qt UI Optimizer
|
||||||
|
* @date 2024-07-08
|
||||||
|
* @version 1.0
|
||||||
|
*/
|
||||||
|
|
||||||
|
#include "core/database/EnemyDatabase.h"
|
||||||
|
#include "core/database/DatabaseHelper.h"
|
||||||
|
#include <QSqlDatabase>
|
||||||
|
#include <QSqlQuery>
|
||||||
|
#include <QSqlError>
|
||||||
|
#include <QUuid>
|
||||||
|
|
||||||
|
EnemyDatabase* EnemyDatabase::m_instance = nullptr;
|
||||||
|
|
||||||
|
EnemyDatabase::EnemyDatabase(QObject *parent)
|
||||||
|
: QObject(parent)
|
||||||
|
, m_connectionName("EnemyDatabase_Connection")
|
||||||
|
, m_isInitialized(false)
|
||||||
|
{
|
||||||
|
}
|
||||||
|
|
||||||
|
EnemyDatabase::~EnemyDatabase()
|
||||||
|
{
|
||||||
|
if (QSqlDatabase::contains(m_connectionName)) {
|
||||||
|
QSqlDatabase::removeDatabase(m_connectionName);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
EnemyDatabase* EnemyDatabase::getInstance()
|
||||||
|
{
|
||||||
|
if (m_instance == nullptr) {
|
||||||
|
m_instance = new EnemyDatabase();
|
||||||
|
}
|
||||||
|
return m_instance;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::initializeDatabase()
|
||||||
|
{
|
||||||
|
if (m_isInitialized) {
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = DatabaseHelper::createTempConnection(m_connectionName);
|
||||||
|
if (!db.isOpen()) {
|
||||||
|
emit databaseError("无法连接到数据库");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
if (!createTables()) {
|
||||||
|
emit databaseError("创建敌情数据表失败");
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
m_isInitialized = true;
|
||||||
|
qDebug() << "敌情数据库初始化成功";
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::createTables()
|
||||||
|
{
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
QString createTableSQL = R"(
|
||||||
|
CREATE TABLE IF NOT EXISTS enemy_records (
|
||||||
|
id VARCHAR(50) PRIMARY KEY,
|
||||||
|
longitude DOUBLE NOT NULL,
|
||||||
|
latitude DOUBLE NOT NULL,
|
||||||
|
threat_level VARCHAR(20) NOT NULL,
|
||||||
|
discovery_time DATETIME NOT NULL,
|
||||||
|
enemy_type VARCHAR(50),
|
||||||
|
status VARCHAR(20) DEFAULT '活跃',
|
||||||
|
description TEXT,
|
||||||
|
update_time DATETIME DEFAULT CURRENT_TIMESTAMP,
|
||||||
|
INDEX idx_threat_level (threat_level),
|
||||||
|
INDEX idx_status (status),
|
||||||
|
INDEX idx_discovery_time (discovery_time)
|
||||||
|
)
|
||||||
|
)";
|
||||||
|
|
||||||
|
query.prepare(createTableSQL);
|
||||||
|
return executeQuery(query, "创建敌情记录表");
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase EnemyDatabase::getDatabase()
|
||||||
|
{
|
||||||
|
if (QSqlDatabase::contains(m_connectionName)) {
|
||||||
|
return QSqlDatabase::database(m_connectionName);
|
||||||
|
}
|
||||||
|
return DatabaseHelper::createTempConnection(m_connectionName);
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::executeQuery(QSqlQuery &query, const QString &operation)
|
||||||
|
{
|
||||||
|
if (!query.exec()) {
|
||||||
|
QString errorMsg = QString("%1失败: %2").arg(operation).arg(query.lastError().text());
|
||||||
|
qDebug() << errorMsg;
|
||||||
|
emit databaseError(errorMsg);
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
return true;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::addEnemyRecord(const EnemyRecord &record)
|
||||||
|
{
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
QString insertSQL = R"(
|
||||||
|
INSERT INTO enemy_records
|
||||||
|
(id, longitude, latitude, threat_level, discovery_time, enemy_type, status, description, update_time)
|
||||||
|
VALUES (?, ?, ?, ?, ?, ?, ?, ?, ?)
|
||||||
|
)";
|
||||||
|
|
||||||
|
query.prepare(insertSQL);
|
||||||
|
query.addBindValue(record.id);
|
||||||
|
query.addBindValue(record.longitude);
|
||||||
|
query.addBindValue(record.latitude);
|
||||||
|
query.addBindValue(record.threatLevel);
|
||||||
|
query.addBindValue(record.discoveryTime);
|
||||||
|
query.addBindValue(record.enemyType);
|
||||||
|
query.addBindValue(record.status);
|
||||||
|
query.addBindValue(record.description);
|
||||||
|
query.addBindValue(QDateTime::currentDateTime());
|
||||||
|
|
||||||
|
bool success = executeQuery(query, "添加敌情记录");
|
||||||
|
if (success) {
|
||||||
|
emit enemyDataUpdated();
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::updateEnemyRecord(const EnemyRecord &record)
|
||||||
|
{
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
QString updateSQL = R"(
|
||||||
|
UPDATE enemy_records SET
|
||||||
|
longitude = ?, latitude = ?, threat_level = ?, discovery_time = ?,
|
||||||
|
enemy_type = ?, status = ?, description = ?, update_time = ?
|
||||||
|
WHERE id = ?
|
||||||
|
)";
|
||||||
|
|
||||||
|
query.prepare(updateSQL);
|
||||||
|
query.addBindValue(record.longitude);
|
||||||
|
query.addBindValue(record.latitude);
|
||||||
|
query.addBindValue(record.threatLevel);
|
||||||
|
query.addBindValue(record.discoveryTime);
|
||||||
|
query.addBindValue(record.enemyType);
|
||||||
|
query.addBindValue(record.status);
|
||||||
|
query.addBindValue(record.description);
|
||||||
|
query.addBindValue(QDateTime::currentDateTime());
|
||||||
|
query.addBindValue(record.id);
|
||||||
|
|
||||||
|
bool success = executeQuery(query, "更新敌情记录");
|
||||||
|
if (success) {
|
||||||
|
emit enemyDataUpdated();
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::deleteEnemyRecord(const QString &enemyId)
|
||||||
|
{
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("DELETE FROM enemy_records WHERE id = ?");
|
||||||
|
query.addBindValue(enemyId);
|
||||||
|
|
||||||
|
bool success = executeQuery(query, "删除敌情记录");
|
||||||
|
if (success) {
|
||||||
|
emit enemyDataUpdated();
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
|
QList<EnemyRecord> EnemyDatabase::getAllEnemyRecords()
|
||||||
|
{
|
||||||
|
QList<EnemyRecord> records;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT * FROM enemy_records ORDER BY discovery_time DESC");
|
||||||
|
|
||||||
|
if (executeQuery(query, "获取所有敌情记录")) {
|
||||||
|
while (query.next()) {
|
||||||
|
records.append(queryToRecord(query));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
EnemyRecord EnemyDatabase::getEnemyRecord(const QString &enemyId)
|
||||||
|
{
|
||||||
|
EnemyRecord record;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return record;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return record;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT * FROM enemy_records WHERE id = ?");
|
||||||
|
query.addBindValue(enemyId);
|
||||||
|
|
||||||
|
if (executeQuery(query, "获取敌情记录") && query.next()) {
|
||||||
|
record = queryToRecord(query);
|
||||||
|
}
|
||||||
|
|
||||||
|
return record;
|
||||||
|
}
|
||||||
|
|
||||||
|
QList<EnemyRecord> EnemyDatabase::getEnemyRecordsByThreatLevel(const QString &threatLevel)
|
||||||
|
{
|
||||||
|
QList<EnemyRecord> records;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT * FROM enemy_records WHERE threat_level = ? ORDER BY discovery_time DESC");
|
||||||
|
query.addBindValue(threatLevel);
|
||||||
|
|
||||||
|
if (executeQuery(query, "按威胁等级获取敌情记录")) {
|
||||||
|
while (query.next()) {
|
||||||
|
records.append(queryToRecord(query));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QList<EnemyRecord> EnemyDatabase::getEnemyRecordsByStatus(const QString &status)
|
||||||
|
{
|
||||||
|
QList<EnemyRecord> records;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT * FROM enemy_records WHERE status = ? ORDER BY discovery_time DESC");
|
||||||
|
query.addBindValue(status);
|
||||||
|
|
||||||
|
if (executeQuery(query, "按状态获取敌情记录")) {
|
||||||
|
while (query.next()) {
|
||||||
|
records.append(queryToRecord(query));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QList<EnemyRecord> EnemyDatabase::getEnemyRecordsByTimeRange(const QDateTime &startTime, const QDateTime &endTime)
|
||||||
|
{
|
||||||
|
QList<EnemyRecord> records;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT * FROM enemy_records WHERE discovery_time BETWEEN ? AND ? ORDER BY discovery_time DESC");
|
||||||
|
query.addBindValue(startTime);
|
||||||
|
query.addBindValue(endTime);
|
||||||
|
|
||||||
|
if (executeQuery(query, "按时间范围获取敌情记录")) {
|
||||||
|
while (query.next()) {
|
||||||
|
records.append(queryToRecord(query));
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return records;
|
||||||
|
}
|
||||||
|
|
||||||
|
QMap<QString, int> EnemyDatabase::getEnemyStatistics()
|
||||||
|
{
|
||||||
|
QMap<QString, int> statistics;
|
||||||
|
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return statistics;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return statistics;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT threat_level, COUNT(*) as count FROM enemy_records GROUP BY threat_level");
|
||||||
|
|
||||||
|
if (executeQuery(query, "获取敌情统计")) {
|
||||||
|
while (query.next()) {
|
||||||
|
QString threatLevel = query.value("threat_level").toString();
|
||||||
|
int count = query.value("count").toInt();
|
||||||
|
statistics[threatLevel] = count;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return statistics;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::clearAllEnemyRecords()
|
||||||
|
{
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("DELETE FROM enemy_records");
|
||||||
|
|
||||||
|
bool success = executeQuery(query, "清空所有敌情记录");
|
||||||
|
if (success) {
|
||||||
|
emit enemyDataUpdated();
|
||||||
|
}
|
||||||
|
return success;
|
||||||
|
}
|
||||||
|
|
||||||
|
bool EnemyDatabase::enemyExists(const QString &enemyId)
|
||||||
|
{
|
||||||
|
if (!m_isInitialized && !initializeDatabase()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlDatabase db = getDatabase();
|
||||||
|
if (!db.isValid()) {
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QSqlQuery query(db);
|
||||||
|
query.prepare("SELECT COUNT(*) FROM enemy_records WHERE id = ?");
|
||||||
|
query.addBindValue(enemyId);
|
||||||
|
|
||||||
|
if (executeQuery(query, "检查敌人ID是否存在") && query.next()) {
|
||||||
|
return query.value(0).toInt() > 0;
|
||||||
|
}
|
||||||
|
|
||||||
|
return false;
|
||||||
|
}
|
||||||
|
|
||||||
|
QString EnemyDatabase::generateNewEnemyId()
|
||||||
|
{
|
||||||
|
QString prefix = "ENEMY";
|
||||||
|
int counter = 1;
|
||||||
|
QString newId;
|
||||||
|
|
||||||
|
do {
|
||||||
|
newId = QString("%1%2").arg(prefix).arg(counter, 3, 10, QChar('0'));
|
||||||
|
counter++;
|
||||||
|
} while (enemyExists(newId) && counter < 1000);
|
||||||
|
|
||||||
|
return newId;
|
||||||
|
}
|
||||||
|
|
||||||
|
EnemyRecord EnemyDatabase::queryToRecord(const QSqlQuery &query)
|
||||||
|
{
|
||||||
|
EnemyRecord record;
|
||||||
|
record.id = query.value("id").toString();
|
||||||
|
record.longitude = query.value("longitude").toDouble();
|
||||||
|
record.latitude = query.value("latitude").toDouble();
|
||||||
|
record.threatLevel = query.value("threat_level").toString();
|
||||||
|
record.discoveryTime = query.value("discovery_time").toDateTime();
|
||||||
|
record.enemyType = query.value("enemy_type").toString();
|
||||||
|
record.status = query.value("status").toString();
|
||||||
|
record.description = query.value("description").toString();
|
||||||
|
record.updateTime = query.value("update_time").toDateTime();
|
||||||
|
return record;
|
||||||
|
}
|
||||||
Loading…
Reference in new issue