Initial commit

tests_schema/tests_sources/sqlite_statement.cpp

@@ -0,0 +1,90 @@
+#include "sqlite_statement.h"
+#include <iostream>
+
+// Constructor: Prepares the SQL statement
+SQLiteStatement::SQLiteStatement(sqlite3 *db, const std::string &query)
+    : db_(db), stmt_(nullptr) {
+  if (db == nullptr) {
+    throw std::invalid_argument("Database connection is null");
+  }
+
+  if (sqlite3_prepare_v2(db, query.c_str(), -1, &stmt_, nullptr) != SQLITE_OK) {
+    std::cerr << "Error preparing SQL statement:" << "\"" + query + "\""
+              << std::endl;
+    std::cerr << "Error message:" << sqlite3_errmsg(db) << std::endl;
+
+    if (stmt_) {
+      sqlite3_finalize(stmt_);
+    }
+
+    throw std::runtime_error("Failed to prepare SQLite statement: " + query);
+  }
+}
+
+// Destructor: Finalizes the statement
+SQLiteStatement::~SQLiteStatement() { finalize(); }
+
+// Bind integer value
+bool SQLiteStatement::bind(int index, int value) {
+  return sqlite3_bind_int(stmt_, index, value) == SQLITE_OK;
+}
+
+// Bind double value
+bool SQLiteStatement::bind(int index, double value) {
+  return sqlite3_bind_double(stmt_, index, value) == SQLITE_OK;
+}
+
+// Bind text value
+bool SQLiteStatement::bind(int index, const std::string &value) {
+  return sqlite3_bind_text(stmt_, index, value.c_str(), -1, SQLITE_TRANSIENT) ==
+         SQLITE_OK;
+}
+
+// Bind NULL value
+bool SQLiteStatement::bindNull(int index) {
+  return sqlite3_bind_null(stmt_, index) == SQLITE_OK;
+}
+
+// Run the statement
+// step() returns true if there are more rows
+bool SQLiteStatement::execute() { return step() == false; }
+
+// Execute the statement
+bool SQLiteStatement::step() {
+  int rc = sqlite3_step(stmt_);
+
+  if (rc == SQLITE_ROW) {
+    return true;
+  }
+  if (rc == SQLITE_DONE) {
+    return false;
+  }
+
+  std::cerr << "Failed to step:" << sqlite3_errmsg(db_) << std::endl;
+  return false;
+}
+
+// Get the number of rows affected
+int SQLiteStatement::changes() { return sqlite3_changes(db_); }
+
+// Reset the statement for re-use
+void SQLiteStatement::reset() { sqlite3_reset(stmt_); }
+
+// Finalize and clean up the statement
+void SQLiteStatement::finalize() {
+  if (stmt_) {
+    sqlite3_finalize(stmt_);
+    stmt_ = nullptr;
+  }
+}
+
+// Adapter for sqlite3_column_count
+int SQLiteStatement::getColumnCount() const {
+  return sqlite3_column_count(stmt_);
+}
+
+// Adapter for sqlite3_column_text
+std::string SQLiteStatement::getColumnText(int column_index) const {
+  const unsigned char *text = sqlite3_column_text(stmt_, column_index);
+  return text ? reinterpret_cast<const char *>(text) : "";
+}