add db.Execute

delaneyj · Jul 3, 2024 · 016fe53 · 016fe53
1 parent f419860
commit 016fe53
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/Taskfile.yml b/Taskfile.yml
@@ -3,7 +3,7 @@
 version: "3"
 
 vars:
-  VERSION: 0.2.8
+  VERSION: 0.2.9
 
 interval: 200ms
 

diff --git a/database.go b/database.go
@@ -41,6 +41,16 @@ func NewDatabase(ctx context.Context, dbFilename string, migrations []string) (*
 	return db, nil
 }
 
+func (db *Database) Execute(ctx context.Context, query string, opts *sqlitex.ExecOptions) error {
+	conn, err := db.writePool.Take(ctx)
+	if err != nil {
+		return fmt.Errorf("failed to take write connection: %w", err)
+	}
+	defer db.writePool.Put(conn)
+
+	return sqlitex.Execute(conn, query, opts)
+}
+
 func (db *Database) Reset(ctx context.Context, shouldClear bool) (err error) {
 	if err := db.Close(); err != nil {
 		return fmt.Errorf("could not close database: %w", err)