54 lines
923 B
Go
54 lines
923 B
Go
|
package check
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
"fmt"
|
||
|
|
"strings"
|
||
|
|
"sync"
|
||
|
|
)
|
||
|
|
|
||
|
|
var (
|
||
|
|
ErrModuleNameConflict = errors.New("module name conflict")
|
||
|
|
ErrNoSuchModule = errors.New("no module of given name known")
|
||
|
|
)
|
||
|
|
|
||
|
|
func NewRegistry() *Registry {
|
||
|
|
return &Registry{
|
||
|
|
mods: make(map[string]*Module),
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
type (
|
||
|
|
Registry struct {
|
||
|
|
lock sync.RWMutex
|
||
|
|
mods map[string]*Module
|
||
|
|
}
|
||
|
|
)
|
||
|
|
|
||
|
|
func (r *Registry) Register(module *Module) error {
|
||
|
|
r.lock.Lock()
|
||
|
|
defer r.lock.Unlock()
|
||
|
|
|
||
|
|
modName := strings.ToLower(module.Name())
|
||
|
|
|
||
|
|
if _, ok := r.mods[modName]; ok {
|
||
|
|
return fmt.Errorf("%w: %s", ErrModuleNameConflict, modName)
|
||
|
|
}
|
||
|
|
|
||
|
|
r.mods[modName] = module
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
|
||
|
|
func (r *Registry) Lookup(modName string) (CheckerLookup, error) {
|
||
|
|
r.lock.RLock()
|
||
|
|
defer r.lock.RUnlock()
|
||
|
|
|
||
|
|
modName = strings.ToLower(modName)
|
||
|
|
|
||
|
|
if mod, ok := r.mods[modName]; !ok {
|
||
|
|
return nil, fmt.Errorf("%w: %s", ErrNoSuchModule, modName)
|
||
|
|
} else {
|
||
|
|
return mod, nil
|
||
|
|
}
|
||
|
|
}
|