43 lines
933 B
Go
43 lines
933 B
Go
|
// Copyright (C) 2019 Storj Labs, Inc.
|
||
|
|
// See LICENSE for copying information.
|
||
|
|
|
||
|
|
package satellitedb
|
||
|
|
|
||
|
|
import (
|
||
|
|
"context"
|
||
|
|
"sync"
|
||
|
|
|
||
|
|
"storj.io/storj/satellite/console"
|
||
|
|
)
|
||
|
|
|
||
|
|
// BeginTransaction is a method for opening transaction
|
||
|
|
func (m *lockedConsole) BeginTx(ctx context.Context) (console.DBTx, error) {
|
||
|
|
m.Lock()
|
||
|
|
db, err := m.db.BeginTx(ctx)
|
||
|
|
|
||
|
|
txlocked := &lockedConsole{&sync.Mutex{}, db}
|
||
|
|
return &lockedTx{m, txlocked, db, sync.Once{}}, err
|
||
|
|
}
|
||
|
|
|
||
|
|
// lockedTx extends Database with transaction scope
|
||
|
|
type lockedTx struct {
|
||
|
|
parent *lockedConsole
|
||
|
|
*lockedConsole
|
||
|
|
tx console.DBTx
|
||
|
|
once sync.Once
|
||
|
|
}
|
||
|
|
|
||
|
|
// Commit is a method for committing and closing transaction
|
||
|
|
func (db *lockedTx) Commit() error {
|
||
|
|
err := db.tx.Commit()
|
||
|
|
db.once.Do(db.parent.Unlock)
|
||
|
|
return err
|
||
|
|
}
|
||
|
|
|
||
|
|
// Rollback is a method for rollback and closing transaction
|
||
|
|
func (db *lockedTx) Rollback() error {
|
||
|
|
err := db.tx.Rollback()
|
||
|
|
db.once.Do(db.parent.Unlock)
|
||
|
|
return err
|
||
|
|
}
|