SerialPortDebugger/DatabaseWrapper.cpp at main · AndroidCoderPeng/SerialPortDebugger · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
#include "DatabaseWrapper.hpp"
#include "Logger.hpp"
#include "qvariant.h"

#include <QSqlError>
#include <QSqlQuery>

DatabaseWrapper *DatabaseWrapper::_instancePtr = nullptr;

DatabaseWrapper::DatabaseWrapper() : _queryPtr{nullptr} {}

DatabaseWrapper::~DatabaseWrapper() {
  if (_queryPtr) {
    delete _queryPtr;
  }
}

DatabaseWrapper *DatabaseWrapper::get() {
  if (!_instancePtr) {
    _instancePtr = new DatabaseWrapper();
  }
  return _instancePtr;
}

bool DatabaseWrapper::init() {
  _db = QSqlDatabase::addDatabase("QSQLITE");
  _db.setDatabaseName("commands.db");

  if (!_db.open()) {
    Logger::Tag("DatabaseWrapper")
        .eFmt("数据库打开失败：%s",
              _db.lastError().text().toStdString().c_str());
    return false;
  }

  _queryPtr = new QSqlQuery(_db);

  if (!_queryPtr->exec("CREATE TABLE IF NOT EXISTS commands ("
                       "id INTEGER PRIMARY KEY, "
                       "value TEXT UNIQUE, "
                       "remark TEXT)")) {
    Logger::Tag("DatabaseWrapper")
        .eFmt("表创建失败：%s",
              _queryPtr->lastError().text().toStdString().c_str());
    return false;
  }

  initDefaultCommands();
  return true;
}

void DatabaseWrapper::initDefaultCommands() {
  _queryPtr->exec("SELECT COUNT(*) FROM commands");
  _queryPtr->next();
  int count = _queryPtr->value(0).toInt();

  if (count == 0) {
    QList<DefaultCommand> defaults = {
        {"FE FE 02 10 FA", "上电"},
        {"FE FE 02 13 FA", "下电"},
        {"FE FE 02 3D FA", "电位查询"},
        {"FE FE 0F 3C 08 00 08 00 08 00 08 00 08 00 08 00 14 FA", "复位"},
        {"FE FE 03 12 01 FA", "状态检测"}};

    _queryPtr->prepare("INSERT INTO commands (value, remark) VALUES (?, ?)");
    for (const auto &cmd : defaults) {
      _queryPtr->addBindValue(cmd.value);
      _queryPtr->addBindValue(cmd.remark);
      if (!_queryPtr->exec()) {
        Logger::Tag("DatabaseWrapper")
            .eFmt("插入默认指令失败：%s",
                  _queryPtr->lastError().text().toStdString().c_str());
      }
    }
  }
}

void DatabaseWrapper::addCommand(const QString &command,
                                 const QString &remark) {
  _queryPtr->prepare("INSERT INTO commands (value, remark) VALUES (?, ?)");
  _queryPtr->addBindValue(command);
  _queryPtr->addBindValue(remark);

  if (!_queryPtr->exec()) {
    Logger::Tag("DatabaseWrapper")
        .eFmt("插入失败：%s",
              _queryPtr->lastError().text().toStdString().c_str());
  }
}

void DatabaseWrapper::updateCommand(int id, const QString &command,
                                    const QString &remark) {
  _queryPtr->prepare("UPDATE commands SET value = ?, remark = ? WHERE id = ?");
  _queryPtr->addBindValue(command);
  _queryPtr->addBindValue(remark);
  _queryPtr->addBindValue(id);

  if (!_queryPtr->exec()) {
    Logger::Tag("DatabaseWrapper")
        .eFmt("更新失败：%s",
              _queryPtr->lastError().text().toStdString().c_str());
  }
}

void DatabaseWrapper::deleteCommand(int id) {
  _queryPtr->prepare("DELETE FROM commands WHERE id = ?");
  _queryPtr->addBindValue(id);

  if (!_queryPtr->exec()) {
    Logger::Tag("DatabaseWrapper")
        .eFmt("删除失败：%s",
              _queryPtr->lastError().text().toStdString().c_str());
  }
}

QList<DatabaseCommand> DatabaseWrapper::getAllCommands() {
  QList<DatabaseCommand> commands;

  if (!_queryPtr->exec("SELECT id, value, remark FROM commands")) {
    Logger::Tag("DatabaseWrapper")
        .e(_queryPtr->lastError().text().toStdString().c_str());
    return commands;
  }

  while (_queryPtr->next()) {
    DatabaseCommand cmd;
    cmd.id = _queryPtr->value(0).toInt();
    cmd.value = _queryPtr->value(1).toString();
    cmd.remark = _queryPtr->value(2).toString();
    commands.append(cmd);
  }

  return commands;
}

bool DatabaseWrapper::commandExists(const QString &command) {
  _queryPtr->prepare("SELECT COUNT(*) FROM commands WHERE value = ?");
  _queryPtr->addBindValue(command);

  if (!_queryPtr->exec()) {
    return false;
  }

  if (_queryPtr->next()) {
    return _queryPtr->value(0).toInt() > 0;
  }

  return false;
}