28 lines
504 B
Go
28 lines
504 B
Go
|
package database
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"database/sql"
|
||
|
|
"fmt"
|
||
|
|
)
|
||
|
|
|
||
|
|
func WithTx(ctx context.Context, db *sql.DB, fn func(tx *sql.Tx) error) error {
|
||
|
|
tx, err := db.BeginTx(ctx, nil)
|
||
|
|
if err != nil {
|
||
|
|
return fmt.Errorf("begin tx: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := fn(tx); err != nil {
|
||
|
|
if rbErr := tx.Rollback(); rbErr != nil {
|
||
|
|
return fmt.Errorf("rollback failed: %v (original: %w)", rbErr, err)
|
||
|
|
}
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
if err := tx.Commit(); err != nil {
|
||
|
|
return fmt.Errorf("commit tx: %w", err)
|
||
|
|
}
|
||
|
|
|
||
|
|
return nil
|
||
|
|
}
|