Stanislav N. aka pztrn
2c13e3f380
Some checks failed
Linting and tests / Linting (push) Failing after 37s
82 lines
1.8 KiB
Go
82 lines
1.8 KiB
Go
package database
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"log/slog"
|
|
|
|
"bunker/server/internal/services/core"
|
|
|
|
"github.com/jmoiron/sqlx"
|
|
)
|
|
|
|
type transaction struct {
|
|
transaction *sqlx.Tx
|
|
logger *slog.Logger
|
|
}
|
|
|
|
func (d *database) Transaction(ctx context.Context) (core.DatabaseTransaction, error) {
|
|
txn, err := d.db.BeginTxx(ctx, nil)
|
|
if err != nil {
|
|
return nil, fmt.Errorf("%w: starting transaction: %w", core.ErrDatabase, err)
|
|
}
|
|
|
|
txHandler := &transaction{
|
|
transaction: txn,
|
|
logger: d.logger.With("module", "transactioner"),
|
|
}
|
|
|
|
return txHandler, nil
|
|
}
|
|
|
|
func (t *transaction) Apply(steps ...core.TransactionFunc) error {
|
|
for stepNumber, stepFunc := range steps {
|
|
if err := stepFunc(t.transaction); err != nil {
|
|
t.logger.Error(
|
|
"Error occurred.",
|
|
"step", stepNumber,
|
|
"error", err.Error(),
|
|
"module", "core/database",
|
|
"subsystem", "transaction",
|
|
)
|
|
|
|
if rollbackErr := t.transaction.Rollback(); rollbackErr != nil {
|
|
t.logger.Error(
|
|
"Transaction rollback failed.",
|
|
"error", err.Error(),
|
|
"module", "core/database",
|
|
"subsystem", "transaction",
|
|
)
|
|
|
|
return fmt.Errorf("%w: transaction rollback: %w", core.ErrDatabase, rollbackErr)
|
|
}
|
|
|
|
return err
|
|
}
|
|
}
|
|
|
|
if err := t.transaction.Commit(); err != nil {
|
|
t.logger.Error(
|
|
"Transaction commit failed.",
|
|
"error", err.Error(),
|
|
"module", "core/database",
|
|
"subsystem", "transaction",
|
|
)
|
|
|
|
if rollbackErr := t.transaction.Rollback(); rollbackErr != nil {
|
|
t.logger.Error(
|
|
"Transaction rollback failed.",
|
|
"error", err.Error(),
|
|
"module", "core/database",
|
|
"subsystem", "transaction",
|
|
)
|
|
|
|
return fmt.Errorf("%w: transaction rollback: %w", core.ErrDatabase, rollbackErr)
|
|
}
|
|
|
|
return fmt.Errorf("%w: transaction commit: %w", core.ErrDatabase, err)
|
|
}
|
|
|
|
return nil
|
|
}
|