init

libsqlite3/inc/us/Sqlite3.hpp

@@ -0,0 +1,98 @@
+#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;
+	};
+}