mirror of
https://github.com/stashapp/stash.git
synced 2025-12-17 04:14:39 +03:00
Restructure data layer (#2532)
* Add new txn manager interface * Add txn management to sqlite * Rename get to getByID * Add contexts to repository methods * Update query builders * Add context to reader writer interfaces * Use repository in resolver * Tighten interfaces * Tighten interfaces in dlna * Tighten interfaces in match package * Tighten interfaces in scraper package * Tighten interfaces in scan code * Tighten interfaces on autotag package * Remove ReaderWriter usage * Merge database package into sqlite
This commit is contained in:
38
pkg/txn/transaction.go
Normal file
38
pkg/txn/transaction.go
Normal file
@@ -0,0 +1,38 @@
|
||||
package txn
|
||||
|
||||
import "context"
|
||||
|
||||
type Manager interface {
|
||||
Begin(ctx context.Context) (context.Context, error)
|
||||
Commit(ctx context.Context) error
|
||||
Rollback(ctx context.Context) error
|
||||
}
|
||||
|
||||
type TxnFunc func(ctx context.Context) error
|
||||
|
||||
func WithTxn(ctx context.Context, m Manager, fn TxnFunc) error {
|
||||
var err error
|
||||
ctx, err = m.Begin(ctx)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
defer func() {
|
||||
if p := recover(); p != nil {
|
||||
// a panic occurred, rollback and repanic
|
||||
_ = m.Rollback(ctx)
|
||||
panic(p)
|
||||
}
|
||||
|
||||
if err != nil {
|
||||
// something went wrong, rollback
|
||||
_ = m.Rollback(ctx)
|
||||
} else {
|
||||
// all good, commit
|
||||
err = m.Commit(ctx)
|
||||
}
|
||||
}()
|
||||
|
||||
err = fn(ctx)
|
||||
return err
|
||||
}
|
||||
Reference in New Issue
Block a user