mirror/sqlgen
1
0
Fork 0
mirror of https://github.com/getml/sqlgen.git synced 2026-05-06 23:21:48 -05:00
Code Issues Packages Projects Releases Wiki Activity

Added write functions

Browse Source
This commit is contained in:
Dr. Patrick Urbanke
2025-04-05 10:42:14 +02:00
parent 412205d2a8
commit 7f9393ee46
7 changed files with 177 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/sqlgen/sqlite/Connection.cpp
+19 -9
View File
@@ -26,13 +26,15 @@ std::string Connection::create_table_to_sql(
std::stringstream stream;
stream << "CREATE TABLE ";
if (_stmt.if_not_exists) {
stream << "IF NOT EXISTS ";
}
if (_stmt.table.schema) {
stream << "\"" << *_stmt.table.schema << "\".";
}
stream << "\"" << _stmt.table.name << "\" ";
if (_stmt.if_not_exists) {
stream << "IF NOT EXISTS ";
}
stream << "(";
stream << internal::strings::join(
@@ -78,9 +80,9 @@ std::string Connection::insert_to_sql(const dynamic::Insert& _stmt) noexcept {
if (_stmt.table.schema) {
stream << "\"" << *_stmt.table.schema << "\".";
}
stream << "\"" << _stmt.table.name << "\" ";
stream << "\"" << _stmt.table.name << "\"";
stream << "(";
stream << " (";
stream << internal::strings::join(
", ", internal::collect::vector(_stmt.columns | transform(in_quotes)));
stream << ")";
@@ -159,16 +161,18 @@ Result<Nothing> Connection::write(
}
for (const auto& row : _data) {
for (size_t i = 0; i < row.size(); ++i) {
const auto num_cols = static_cast<int>(row.size());
for (int i = 0; i < num_cols; ++i) {
if (row[i]) {
const auto res =
sqlite3_bind_text(stmt_.get(), static_cast<int>(i), row[i]->c_str(),
sqlite3_bind_text(stmt_.get(), i + 1, row[i]->c_str(),
static_cast<int>(row[i]->size()), SQLITE_STATIC);
if (res != SQLITE_OK) {
return error(sqlite3_errmsg(conn_.get()));
}
} else {
const auto res = sqlite3_bind_null(stmt_.get(), static_cast<int>(i));
const auto res = sqlite3_bind_null(stmt_.get(), i + 1);
if (res != SQLITE_OK) {
return error(sqlite3_errmsg(conn_.get()));
}
@@ -176,7 +180,7 @@ Result<Nothing> Connection::write(
}
auto res = sqlite3_step(stmt_.get());
if (res != SQLITE_OK) {
if (res != SQLITE_OK && res != SQLITE_ROW && res != SQLITE_DONE) {
return error(sqlite3_errmsg(conn_.get()));
}
@@ -186,6 +190,12 @@ Result<Nothing> Connection::write(
}
}
// We need to reset the statement to avoid segfaults.
const auto res = sqlite3_clear_bindings(stmt_.get());
if (res != SQLITE_OK) {
return error(sqlite3_errmsg(conn_.get()));
}
return Nothing{};
}