store the database handle in Iterator and Statement

1 files changed, 4 insertions, 0 deletions

diff --git a/sqlite.zig b/sqlite.zig
index e1e3216..feb0295 100644
--- a/sqlite.zig
+++ b/sqlite.zig
@@ -456,6 +456,7 @@ pub fn Iterator(comptime Type: type) type {
 
         const TypeInfo = @typeInfo(Type);
 
+        db: *c.sqlite3,
         stmt: *c.sqlite3_stmt,
 
         // next scans the next row using the prepared statement.
@@ -843,6 +844,7 @@ pub fn Statement(comptime opts: StatementOptions, comptime query: ParsedQuery) t
     return struct {
         const Self = @This();
 
+        db: *c.sqlite3,
         stmt: *c.sqlite3_stmt,
 
         fn prepare(db: *Db, flags: c_uint) !Self {
@@ -865,6 +867,7 @@ pub fn Statement(comptime opts: StatementOptions, comptime query: ParsedQuery) t
             };
 
             return Self{
+                .db = db.db,
                 .stmt = stmt,
             };
         }
@@ -1010,6 +1013,7 @@ pub fn Statement(comptime opts: StatementOptions, comptime query: ParsedQuery) t
             self.bind(values);
 
             var res: Iterator(Type) = undefined;
+            res.db = self.db;
             res.stmt = self.stmt;
 
             return res;