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 ExtractorConstructor func() (contracts.Extractor, error)
type ExtractorRegistry struct {
constructors map[string]ExtractorConstructor
specs map[string]ModuleSpec
}
func NewExtractorRegistry() *ExtractorRegistry {
return &ExtractorRegistry{
constructors: make(map[string]ExtractorConstructor),
specs: make(map[string]ModuleSpec),
}
}
func (r *ExtractorRegistry) Register(key string, constructor ExtractorConstructor) error {
return r.RegisterWithSpec(defaultModuleSpec(key, StageExtract), constructor)
}
func (r *ExtractorRegistry) RegisterWithSpec(spec ModuleSpec, constructor ExtractorConstructor) error {
if r == nil {
return fmt.Errorf("extractor registry must not be nil")
}
normalizedKey := strings.TrimSpace(key)
if normalizedKey == "" {
return fmt.Errorf("extractor key must not be empty")
normalizedSpec := normalizeModuleSpec(spec)
if err := validateModuleSpec("extractor", StageExtract, normalizedSpec); err != nil {
return err
}
if constructor == nil {
return fmt.Errorf("extractor constructor for %q must not be nil", normalizedKey)
return fmt.Errorf("extractor constructor for %q must not be nil", normalizedSpec.Key)
}
if _, ok := r.constructors[normalizedKey]; ok {
return fmt.Errorf("extractor %q is already registered", normalizedKey)
if _, ok := r.constructors[normalizedSpec.Key]; ok {
return fmt.Errorf("extractor %q is already registered", normalizedSpec.Key)
}
r.constructors[normalizedKey] = constructor
if r.constructors == nil {
r.constructors = make(map[string]ExtractorConstructor)
}
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 *ExtractorRegistry) Build(key string) (contracts.Extractor, error) {
return extractor, nil
}
func (r *ExtractorRegistry) 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 *ExtractorRegistry) 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)
}