99 lines
2.3 KiB
C++
99 lines
2.3 KiB
C++
#include <iostream>
|
|
#include <memory>
|
|
#include <sqlite3.h>
|
|
namespace qing {
|
|
|
|
|
|
class Sqlite3_stmt {
|
|
public:
|
|
Sqlite3_stmt(sqlite3 *db, const char *sql) {
|
|
int rc = sqlite3_prepare_v2(db, sql, -1, &stmt, nullptr);
|
|
if (rc != SQLITE_OK) {
|
|
std::cerr << "Failed to prepare sql stmt: " << sqlite3_errmsg(db) << std::endl;
|
|
throw std::runtime_error("Failed to prepare sql stmt");
|
|
}
|
|
}
|
|
|
|
~Sqlite3_stmt() {
|
|
sqlite3_finalize(stmt);
|
|
}
|
|
|
|
void bind_int(int idx, int data) {
|
|
int rc = sqlite3_bind_int(stmt, idx, data);
|
|
if (rc != SQLITE_OK) {
|
|
std::cerr << "Failed to bind an integer to sql stmt." << std::endl;
|
|
throw std::runtime_error("Failed to bind an integer to sql stmt");
|
|
}
|
|
}
|
|
|
|
void bind_text(int idx, const std::string &data) {
|
|
int rc = sqlite3_bind_text(stmt, idx, data.c_str(), -1, SQLITE_TRANSIENT);
|
|
if (rc != SQLITE_OK) {
|
|
std::cerr << "Failed to bind a text to sql stmt." << std::endl;
|
|
throw std::runtime_error("Failed to bind a text to sql stmt");
|
|
}
|
|
}
|
|
|
|
bool step() {
|
|
int rc = sqlite3_step(stmt);
|
|
if (rc != SQLITE_ROW && rc != SQLITE_DONE) {
|
|
throw std::runtime_error("Failed to query");
|
|
}
|
|
return rc == SQLITE_ROW && rc != SQLITE_DONE;
|
|
}
|
|
|
|
/*
|
|
* int len = sqlite3_column_bytes(stmt, 1); // 获取字节长度（不含结束符）
|
|
* std::string name(reinterpret_cast<const char*>(rawName), len);
|
|
*/
|
|
int read_int(int idx) {
|
|
return sqlite3_column_int(stmt, idx);
|
|
}
|
|
|
|
|
|
const unsigned char *read_text(int idx) {
|
|
return sqlite3_column_text(stmt, idx);
|
|
}
|
|
|
|
private:
|
|
sqlite3_stmt* stmt;
|
|
};
|
|
|
|
class Sqlite3 {
|
|
|
|
public:
|
|
Sqlite3(const std::string& filename) {
|
|
int rc = sqlite3_open(filename.c_str(), &db);
|
|
if (rc != SQLITE_OK) {
|
|
std::cerr << "Failed to open database: " << sqlite3_errmsg(db) << std::endl;
|
|
sqlite3_close(db);
|
|
throw std::runtime_error("Failed to open sqlite3 database.");
|
|
}
|
|
}
|
|
|
|
~Sqlite3() {
|
|
sqlite3_close(db);
|
|
}
|
|
|
|
bool exec(const std::string& sql) {
|
|
char *errMsg;
|
|
int rc = sqlite3_exec(db, sql.c_str(), 0, 0, &errMsg);
|
|
if (rc != SQLITE_OK) {
|
|
std::cerr << "Failed to execute at sqlite3: " << errMsg << std::endl;
|
|
sqlite3_free(errMsg);
|
|
//sqlite3_close(db);
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
Sqlite3_stmt stmt(std::string& sql) {
|
|
return Sqlite3_stmt(db, sql.c_str());
|
|
}
|
|
|
|
private:
|
|
sqlite3* db;
|
|
};
|
|
}
|