From 2a638374b3e0a4cc44aee1ba9136a55f4d8fa01c Mon Sep 17 00:00:00 2001
From: Luna
Date: Sat, 19 Mar 2022 01:50:28 -0300
Subject: add Db.runMulti

---
 sqlite.zig | 25 ++++++++++++++++++++++++-
 1 file changed, 24 insertions(+), 1 deletion(-)

diff --git a/sqlite.zig b/sqlite.zig
index 619e0b8..57d1be2 100644
--- a/sqlite.zig
+++ b/sqlite.zig
@@ -876,6 +876,24 @@ pub const Db = struct {
             return errors.errorFromResultCode(result);
         }
     }
+
+    /// This is a convenience function to run statements that do not need
+    /// bindings to values, but have multiple commands inside.
+    ///
+    /// Exmaple: 'create table a(); create table b();'
+    pub fn runMulti(self: *Self, comptime query: []const u8, options: QueryOptions) !void {
+        while (true) {
+            var sql_tail_ptr: ?[*:0]const u8 = null;
+            options.sql_tail_ptr = &sql_tail_ptr;
+
+            // continuously prepare and execute
+            var stmt = try self.prepareWithDiags(query, options);
+            defer stmt.deinit();
+            if (sql_tail_ptr == null) break;
+
+            try stmt.exec(options, .{});
+        }
+    }
 };
 
 /// Savepoint is a helper type for managing savepoints.
@@ -996,6 +1014,11 @@ pub const Savepoint = struct {
 pub const QueryOptions = struct {
     /// if provided, diags will be populated in case of failures.
     diags: ?*Diagnostics = null,
+
+    /// if provided, sql_tail_ptr will point to the last uncompiled statement
+    /// in the prepare() call. this is useful for multiple-statements being
+    /// processed.
+    sql_tail_ptr: ?*?[*:0]const u8 = null,
 };
 
 /// Iterator allows iterating over a result set.
@@ -1532,7 +1555,7 @@ pub const DynamicStatement = struct {
                 @intCast(c_int, query.len),
                 flags,
                 &tmp,
-                null,
+                options.sql_tail_ptr,
             );
             if (result != c.SQLITE_OK) {
                 diags.err = getLastDetailedErrorFromDb(db.db);
-- 
cgit v1.2.3