Improve migration handling and error forwarding in deferred statements
All checks were successful
Build / build (push) Successful in 42s
All checks were successful
Build / build (push) Successful in 42s
This commit is contained in:
parent
2ff3477812
commit
2eacf6fbc4
@ -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)
|
||||||
|
4
query.go
4
query.go
@ -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
|
||||||
|
@ -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)
|
||||||
|
}
|
||||||
|
Loading…
x
Reference in New Issue
Block a user