Add pipeline module registries

This commit is contained in:
2026-07-03 15:30:44 +00:00
parent 75a0a9fa79
commit 62bc8983c7
15 changed files with 1446 additions and 28 deletions

View File

@@ -2,7 +2,6 @@ package pipeline
import (
"fmt"
"sort"
"strings"
"gitea.maximumdirect.net/eric/notarius/internal/framework/contracts"
@@ -12,31 +11,44 @@ type InputAdapterConstructor func() (contracts.InputAdapter, error)
type InputAdapterRegistry struct {
constructors map[string]InputAdapterConstructor
specs map[string]ModuleSpec
}
func NewInputAdapterRegistry() *InputAdapterRegistry {
return &InputAdapterRegistry{
constructors: make(map[string]InputAdapterConstructor),
specs: make(map[string]ModuleSpec),
}
}
func (r *InputAdapterRegistry) Register(key string, constructor InputAdapterConstructor) error {
return r.RegisterWithSpec(defaultModuleSpec(key, StageInput), constructor)
}
func (r *InputAdapterRegistry) RegisterWithSpec(spec ModuleSpec, constructor InputAdapterConstructor) error {
if r == nil {
return fmt.Errorf("input adapter registry must not be nil")
}
normalizedKey := strings.TrimSpace(key)
if normalizedKey == "" {
return fmt.Errorf("input adapter key must not be empty")
normalizedSpec := normalizeModuleSpec(spec)
if err := validateModuleSpec("input adapter", StageInput, normalizedSpec); err != nil {
return err
}
if constructor == nil {
return fmt.Errorf("input adapter constructor for %q must not be nil", normalizedKey)
return fmt.Errorf("input adapter constructor for %q must not be nil", normalizedSpec.Key)
}
if _, ok := r.constructors[normalizedKey]; ok {
return fmt.Errorf("input adapter %q is already registered", normalizedKey)
if _, ok := r.constructors[normalizedSpec.Key]; ok {
return fmt.Errorf("input adapter %q is already registered", normalizedSpec.Key)
}
r.constructors[normalizedKey] = constructor
if r.constructors == nil {
r.constructors = make(map[string]InputAdapterConstructor)
}
if r.specs == nil {
r.specs = make(map[string]ModuleSpec)
}
r.constructors[normalizedSpec.Key] = constructor
r.specs[normalizedSpec.Key] = cloneModuleSpec(normalizedSpec)
return nil
}
@@ -69,15 +81,22 @@ func (r *InputAdapterRegistry) Build(key string) (contracts.InputAdapter, error)
return adapter, nil
}
func (r *InputAdapterRegistry) Spec(key string) (ModuleSpec, bool) {
if r == nil {
return ModuleSpec{}, false
}
spec, ok := r.specs[strings.TrimSpace(key)]
if !ok {
return ModuleSpec{}, false
}
return cloneModuleSpec(spec), true
}
func (r *InputAdapterRegistry) RegisteredKeys() []string {
if r == nil {
return nil
}
keys := make([]string, 0, len(r.constructors))
for key := range r.constructors {
keys = append(keys, key)
}
sort.Strings(keys)
return keys
return sortedRegistryKeys(r.constructors)
}