diff options
Diffstat (limited to 'database.go')
| -rw-r--r-- | database.go | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/database.go b/database.go index c2d69c1..c71ff78 100644 --- a/database.go +++ b/database.go @@ -1,6 +1,7 @@ package main import ( + "database/sql" "embed" "fmt" "io/fs" @@ -43,3 +44,25 @@ func readMigrations() { return nil }) } + +func Transaction(db *sql.DB, f func(*sql.Tx) error) error { + tx, err := db.Begin() + if err != nil { + return err + } + defer tx.Rollback() + + err = f(tx) + if err != nil { + return err + } + + return tx.Commit() +} + +func Migrate(db *sql.DB, index int) error { + return Transaction(db, func(tx *sql.Tx) error { + _, err := tx.Exec(migrations[index]) + return err + }) +} |
