Improve migration handling and error forwarding in deferred statements
All checks were successful
Build / build (push) Successful in 42s

This commit is contained in:
Sebastiaan de Schaetzen 2025-03-16 11:38:31 +01:00
parent 2ff3477812
commit 2eacf6fbc4
4 changed files with 23 additions and 4 deletions

View File

@ -74,6 +74,9 @@ func performSingleMigration(d *Db, migrationScript []byte, targetVersion int) er
for _, statement := range statements { for _, statement := range statements {
statement = strings.TrimSpace(statement) statement = strings.TrimSpace(statement)
if statement == "" {
continue
}
err = tx.Query(statement).Exec() err = tx.Query(statement).Exec()
if err != nil { if err != nil {
return fmt.Errorf("error performing migration: %v", err) return fmt.Errorf("error performing migration: %v", err)

View File

@ -58,7 +58,7 @@ func (q *Query) Exec() (rerr error) {
defer q.unlock() defer q.unlock()
if q.stmt != nil { if q.stmt != nil {
defer func() { rerr = q.stmt.Finalize() }() defer func() { forwardError(q.stmt.Finalize(), &rerr) }()
} }
if q.err != nil { if q.err != nil {
return q.err return q.err
@ -84,7 +84,7 @@ func (q *Query) ScanSingle(results ...any) (rerr error) {
defer q.unlock() defer q.unlock()
// Scan rows // Scan rows
if q.stmt != nil { if q.stmt != nil {
defer func() { rerr = q.stmt.Finalize() }() defer func() { forwardError(q.stmt.Finalize(), &rerr) }()
} }
if q.err != nil { if q.err != nil {
return q.err return q.err

View File

@ -73,11 +73,20 @@ func TestUpdateQuery(t *testing.T) {
tx := db.MustBegin() tx := db.MustBegin()
defer tx.MustRollback() defer tx.MustRollback()
tx.Query("insert into mytable(key, value) values ('lorem', 'bar')").MustExec() tx.Query("insert into mytable(key, value) values ('lorem', 'bar')").MustExec()
tx.Query("update mytable set value = 'ipsum' where key = 'lorem'").MustExec() value := "ipsum"
key := "lorem"
tx.Query("update mytable set value = ? where key = ?").Bind(value, key).MustExec()
tx.MustCommit() tx.MustCommit()
}() }()
var value string var value string
db.Query("select value from mytable where value = 'ipsum'").MustScanSingle(&value) db.Query("select value from mytable where key = 'lorem'").MustScanSingle(&value)
require.Equal(t, "ipsum", value) require.Equal(t, "ipsum", value)
} }
func TestUpdateQueryWithWrongArguments(t *testing.T) {
db := openTestDb(t)
value := "ipsum"
err := db.Query("insert into mytable(key, value) values ('lorem', ?)").Bind(&value).Exec()
require.Error(t, err)
}

7
util.go Normal file
View File

@ -0,0 +1,7 @@
package mysqlite
func forwardError(from error, to *error) {
if from != nil {
*to = from
}
}