add execM

Author: codehz

src/easy_sqlite3.nim

@@ -1,4 +1,4 @@
 import easy_sqlite3/[bindings,macros]
 export macros
 
-export raw, len, toOpenArray, SQLiteError, SQLiteBlob, Statement, Database, OpenFlag, enableSharedCache, initDatabase, exec, changes, lastInsertRowid, `[]=`, reset, step, withColumnBlob, getParameterIndex, getColumnType, getColumn, unpack, `=destroy`
+export raw, len, toOpenArray, SQLiteError, SQLiteBlob, Statement, Database, OpenFlag, enableSharedCache, initDatabase, exec, execM, changes, lastInsertRowid, `[]=`, reset, step, withColumnBlob, getParameterIndex, getColumnType, getColumn, unpack, `=destroy`

src/easy_sqlite3/bindings.nim

@@ -421,7 +421,8 @@ proc changes*(st: var Database): int =
 proc changes*(st: ref Statement): int =
   sqlite3_changes sqlite3_db_handle st.raw
 
-proc initStatement*(db: var Database, sql: string, flags: PrepareFlags = {}): Statement =
+proc newStatement*(db: var Database, sql: string, flags: PrepareFlags = {}): ref Statement =
+  new result
   sqliteCheck db.raw, sqlite3_prepare_v3(db.raw, sql, sql.len, flags,
       addr result.raw, nil)
 
@@ -430,17 +431,15 @@ proc fetchStatement*(db: var Database, sql: string): ref Statement =
   db.stmtcache.withValue(rhash, value) do:
     return value[]
   do:
-    new result
-    result[] = db.initStatement(sql, {sp_persistent})
+    result = db.newStatement(sql, {sp_persistent})
     db.stmtcache[rhash] = result
 
 proc fetchStatement*(db: var Database, sql: static[string]): ref Statement =
   const chash = sql.compileTimeHash
   db.stmtcache.withValue(chash, value) do:
     return value[]
   do:
-    new result
-    result[] = db.initStatement(sql, {sp_persistent})
+    result = db.newStatement(sql, {sp_persistent})
     db.stmtcache[chash] = result
 
 proc getParameterIndex*(st: ref Statement, name: string): int =
@@ -518,8 +517,21 @@ proc unpack*[T: tuple](st: ref Statement, _: typedesc[T]): T =
 
 {.pop.}
 
-proc exec*(db: var Database, sql: string): int {.discardable.} =
-  var st = db.fetchStatement(sql)
-  defer: st.reset()
+proc exec*(db: var Database, sql: string, cache: static bool = true): int {.discardable.} =
+  when cache:
+    let st = db.fetchStatement(sql)
+    defer: st.reset()
+  else:
+    let st = db.newStatement(sql)
   if st.step():
     result = st.getColumn(0, int)
+
+proc execM*(db: var Database, sqls: varargs[string]) {.discardable.} =
+  discard db.exec "BEGIN IMMEDIATE"
+  try:
+    for sql in sqls:
+      discard db.exec(sql, cache = false)
+    discard db.exec "COMMIT"
+  except:
+    discard db.exec "ROLLBACK"
+    raise getCurrentException()

tests/test_basic.nim

@@ -5,10 +5,6 @@ import easy_sqlite3/[memfs, logfs]
 
 proc select_1(arg: int): tuple[value: int] {.importdb: "SELECT $arg".}
 
-proc create_table() {.importdb: """
-  CREATE TABLE mydata(name TEXT PRIMARY KEY NOT NULL, value INT NOT NULL) WITHOUT ROWID;
-""".}
-
 proc insert_data(name: string, value: int) {.importdb: """
   INSERT INTO mydata(name, value) VALUES ($name, $value);
 """.}
@@ -31,8 +27,10 @@ test "full":
     "D": 3,
   }.toTable
   var db = initDatabase("test")
-  db.exec "PRAGMA journal_mode=DELETE"
-  db.create_table()
+  db.execM(
+    "PRAGMA journal_mode=DELETE",
+    "CREATE TABLE mydata(name TEXT PRIMARY KEY NOT NULL, value INT NOT NULL) WITHOUT ROWID;"
+  )
   db.transaction:
     for name, value in dataset:
       db.insert_data name, value