49 lines
1,021 B
Go
49 lines
1,021 B
Go
package registry
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"maps"
|
|
"slices"
|
|
)
|
|
|
|
type (
|
|
factoryFunc[T any] func(context.Context, map[string]any) (T, error)
|
|
|
|
registry[T any] struct {
|
|
typ string
|
|
data map[string]factoryFunc[T]
|
|
}
|
|
)
|
|
|
|
func (r registry[T]) Add(name string, factory func(map[string]any) (T, error)) {
|
|
r.AddCtx(name, func(_ context.Context, cc map[string]any) (T, error) {
|
|
return factory(cc)
|
|
})
|
|
}
|
|
|
|
func (r registry[T]) AddCtx(name string, factory factoryFunc[T]) {
|
|
if _, exists := r.data[name]; exists {
|
|
panic(fmt.Sprintf("cannot register duplicate %s type: %s", r.typ, name))
|
|
}
|
|
r.data[name] = factory
|
|
}
|
|
|
|
func (r registry[T]) Get(name string) (factoryFunc[T], error) {
|
|
factory, exists := r.data[name]
|
|
if !exists {
|
|
return nil, fmt.Errorf("invalid %s type: %s", r.typ, name)
|
|
}
|
|
return factory, nil
|
|
}
|
|
|
|
func (r registry[T]) Types() []string {
|
|
return slices.Collect(maps.Keys(r.data))
|
|
}
|
|
|
|
func New[T any](typ string) registry[T] {
|
|
return registry[T]{
|
|
typ: typ,
|
|
data: make(map[string]factoryFunc[T]),
|
|
}
|
|
}
|