Monorepo for Tangled
Something went wrong. Try again.
Go
1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556package executor
import ( "context" "fmt" "sync"
"tangled.org/core/spindle/engine" "tangled.org/core/spindle/models" "tangled.org/core/spindle/storage")
// wraps a real engine so StartWorkflows gets the slot ReserveSeat already// acquired, not a second one. everything else delegates, the execution// path runs exactly like standalonetype reservedEngine struct { models.Engine slot engine.WorkflowSlot once sync.Once}
type reservedCacheEngine struct { *reservedEngine runner engine.CacheRunner}
func newReservedEngine(inner models.Engine, slot engine.WorkflowSlot) models.Engine { reserved := &reservedEngine{Engine: inner, slot: slot} if runner, ok := inner.(engine.CacheRunner); ok { return &reservedCacheEngine{reservedEngine: reserved, runner: runner} } return reserved}
func (e *reservedCacheEngine) RestoreCache(ctx context.Context, wid models.WorkflowId, wf *models.Workflow, store storage.Storage, caches []models.CacheBinding, wfLogger models.WorkflowLogger) error { return e.runner.RestoreCache(ctx, wid, wf, store, caches, wfLogger)}
func (e *reservedCacheEngine) SaveCache(ctx context.Context, wid models.WorkflowId, wf *models.Workflow, store storage.Storage, caches []models.CacheBinding, wfLogger models.WorkflowLogger) error { return e.runner.SaveCache(ctx, wid, wf, store, caches, wfLogger)}
// hands back the pre-acquired slot exactly once, a second acquire would// double-count itfunc (e *reservedEngine) AcquireWorkflowSlot(ctx context.Context, wid models.WorkflowId, wf *models.Workflow, _ engine.AcquireMode) (engine.WorkflowSlot, error) { var slot engine.WorkflowSlot e.once.Do(func() { slot = e.slot e.slot = nil }) if slot == nil { return nil, fmt.Errorf("reserved slot already consumed") } return slot, nil}