stash/pkg/models/transaction.go
SmallCoccinelle a9e2a590b2
Lint checks phase 2 (#1747)
* Log 3 unchecked errors

Rather than ignore errors, log them at
the WARNING log level.

The server has been functioning without these, so assume they are not at
the ERROR level.

* Log errors in concurrency test

If we can't initialize the configuration, treat the test as a failure.

* Undo the errcheck on configurations for now.

* Handle unchecked errors in pkg/manager

* Resolve unchecked errors

* Handle DLNA/DMS unchecked errors

* Handle error checking in concurrency test

Generalize config initialization, so we can initialize a configuration
without writing it to disk.

Use this in the test case, since otherwise the test fails to write.

* Handle the remaining unchecked errors

* Heed gosimple in update test

* Use one-line if-initializer statements

While here, fix a wrong variable capture error.

* testing.T doesn't support %w

use %v instead which is supported.

* Remove unused query builder functions

The Int/String criterion handler functions are now generalized.

Thus, there's no need to keep these functions around anymore.

* Mark filterBuilder.addRecursiveWith nolint

The function is useful in the future and no other refactors are looking
nice.

Keep the function around, but tell the linter to ignore it.

* Remove utils.Btoi

There are no users of this utility function

* Return error on scan failure

If we fail to scan the row when looking for the
unique checksum index, then report the error upwards.

* Fix comments on exported functions

* Fix typos

* Fix startup error
2021-09-23 17:15:50 +10:00

82 lines
1.7 KiB
Go

package models
import (
"context"
"github.com/stashapp/stash/pkg/logger"
)
type Transaction interface {
Begin() error
Rollback() error
Commit() error
Repository() Repository
}
type ReadTransaction interface {
Begin() error
Rollback() error
Commit() error
Repository() ReaderRepository
}
type TransactionManager interface {
WithTxn(ctx context.Context, fn func(r Repository) error) error
WithReadTxn(ctx context.Context, fn func(r ReaderRepository) error) error
}
func WithTxn(txn Transaction, fn func(r Repository) error) error {
err := txn.Begin()
if err != nil {
return err
}
defer func() {
if p := recover(); p != nil {
// a panic occurred, rollback and repanic
if err := txn.Rollback(); err != nil {
logger.Warnf("error while trying to roll back transaction: %v", err)
}
panic(p)
} else if err != nil {
// something went wrong, rollback
if err := txn.Rollback(); err != nil {
logger.Warnf("error while trying to roll back transaction: %v", err)
}
} else {
// all good, commit
err = txn.Commit()
}
}()
err = fn(txn.Repository())
return err
}
func WithROTxn(txn ReadTransaction, fn func(r ReaderRepository) error) error {
err := txn.Begin()
if err != nil {
return err
}
defer func() {
if p := recover(); p != nil {
// a panic occurred, rollback and repanic
if err := txn.Rollback(); err != nil {
logger.Warnf("error while trying to roll back RO transaction: %v", err)
}
panic(p)
} else if err != nil {
// something went wrong, rollback
if err := txn.Rollback(); err != nil {
logger.Warnf("error while trying to roll back RO transaction: %v", err)
}
} else {
// all good, commit
err = txn.Commit()
}
}()
err = fn(txn.Repository())
return err
}