kevo/pkg/transaction/creator.go

package transaction

import (
	"github.com/KevoDB/kevo/pkg/engine"
	"github.com/KevoDB/kevo/pkg/engine/interfaces"
)

// TransactionCreatorImpl implements the interfaces.TransactionCreator interface
type TransactionCreatorImpl struct{}

// CreateTransaction creates a new transaction
func (tc *TransactionCreatorImpl) CreateTransaction(e interface{}, readOnly bool) (interfaces.Transaction, error) {
	// Convert the interface to the engine.Engine type
	eng, ok := e.(*engine.Engine)
	if !ok {
		return nil, ErrInvalidEngine
	}

	// Determine transaction mode
	var mode TransactionMode
	if readOnly {
		mode = ReadOnly
	} else {
		mode = ReadWrite
	}

	// Create a new transaction
	tx, err := NewTransaction(eng, mode)
	if err != nil {
		return nil, err
	}

	// Return the transaction as an interfaces.Transaction
	return tx, nil
}

// TransactionCreatorWrapper wraps our TransactionCreatorImpl to implement the LegacyTransactionCreator interface
type TransactionCreatorWrapper struct {
	impl *TransactionCreatorImpl
}

// CreateTransaction creates a transaction for the legacy system
func (w *TransactionCreatorWrapper) CreateTransaction(e interface{}, readOnly bool) (engine.LegacyTransaction, error) {
	tx, err := w.impl.CreateTransaction(e, readOnly)
	if err != nil {
		return nil, err
	}

	// Cast to the legacy interface
	// Our Transaction implementation already has all the required methods
	legacyTx, ok := tx.(engine.LegacyTransaction)
	if !ok {
		return nil, ErrInvalidEngine
	}

	return legacyTx, nil
}

// For backward compatibility, register with the old mechanism too
// This can be removed once all code is migrated
func init() {
	// Register the wrapped transaction creator with the engine compatibility layer
	engine.RegisterTransactionCreator(&TransactionCreatorWrapper{
		impl: &TransactionCreatorImpl{},
	})
}