Jeremy Tregunna
9a98349115
Some checks failed
Go Tests / Run Tests (1.24.2) (push) Failing after 15m7s
112 lines
2.3 KiB
Go
112 lines
2.3 KiB
Go
package transaction
|
|
|
|
import (
|
|
"sync"
|
|
"sync/atomic"
|
|
|
|
"github.com/KevoDB/kevo/pkg/stats"
|
|
)
|
|
|
|
// Manager implements the TransactionManager interface
|
|
type Manager struct {
|
|
// Storage backend for transaction operations
|
|
storage StorageBackend
|
|
|
|
// Statistics collector
|
|
stats stats.Collector
|
|
|
|
// Transaction isolation lock
|
|
txLock sync.RWMutex
|
|
|
|
// Transaction counters
|
|
txStarted atomic.Uint64
|
|
txCompleted atomic.Uint64
|
|
txAborted atomic.Uint64
|
|
}
|
|
|
|
// NewManager creates a new transaction manager
|
|
func NewManager(storage StorageBackend, stats stats.Collector) *Manager {
|
|
return &Manager{
|
|
storage: storage,
|
|
stats: stats,
|
|
}
|
|
}
|
|
|
|
// BeginTransaction starts a new transaction
|
|
func (m *Manager) BeginTransaction(readOnly bool) (Transaction, error) {
|
|
// Track transaction start
|
|
if m.stats != nil {
|
|
m.stats.TrackOperation(stats.OpTxBegin)
|
|
}
|
|
m.txStarted.Add(1)
|
|
|
|
// Convert to transaction mode
|
|
mode := ReadWrite
|
|
if readOnly {
|
|
mode = ReadOnly
|
|
}
|
|
|
|
// Create a new transaction
|
|
tx := &TransactionImpl{
|
|
storage: m.storage,
|
|
mode: mode,
|
|
buffer: NewBuffer(),
|
|
rwLock: &m.txLock,
|
|
stats: m,
|
|
}
|
|
|
|
// Set transaction as active
|
|
tx.active.Store(true)
|
|
|
|
// Acquire appropriate lock
|
|
if mode == ReadOnly {
|
|
m.txLock.RLock()
|
|
tx.hasReadLock.Store(true)
|
|
} else {
|
|
m.txLock.Lock()
|
|
tx.hasWriteLock.Store(true)
|
|
}
|
|
|
|
return tx, nil
|
|
}
|
|
|
|
// GetRWLock returns the transaction isolation lock
|
|
func (m *Manager) GetRWLock() *sync.RWMutex {
|
|
return &m.txLock
|
|
}
|
|
|
|
// IncrementTxCompleted increments the completed transaction counter
|
|
func (m *Manager) IncrementTxCompleted() {
|
|
m.txCompleted.Add(1)
|
|
|
|
// Track the commit operation
|
|
if m.stats != nil {
|
|
m.stats.TrackOperation(stats.OpTxCommit)
|
|
}
|
|
}
|
|
|
|
// IncrementTxAborted increments the aborted transaction counter
|
|
func (m *Manager) IncrementTxAborted() {
|
|
m.txAborted.Add(1)
|
|
|
|
// Track the rollback operation
|
|
if m.stats != nil {
|
|
m.stats.TrackOperation(stats.OpTxRollback)
|
|
}
|
|
}
|
|
|
|
// GetTransactionStats returns transaction statistics
|
|
func (m *Manager) GetTransactionStats() map[string]interface{} {
|
|
stats := make(map[string]interface{})
|
|
|
|
stats["tx_started"] = m.txStarted.Load()
|
|
stats["tx_completed"] = m.txCompleted.Load()
|
|
stats["tx_aborted"] = m.txAborted.Load()
|
|
|
|
// Calculate active transactions
|
|
active := m.txStarted.Load() - m.txCompleted.Load() - m.txAborted.Load()
|
|
stats["tx_active"] = active
|
|
|
|
return stats
|
|
}
|