32 lines
959 B
Go
32 lines
959 B
Go
package tasks
|
|
|
|
import (
|
|
"fmt"
|
|
)
|
|
|
|
type TaskHandlerRegistry map[TaskType]func(map[string]interface{}) (TaskHandler, error)
|
|
|
|
var registry TaskHandlerRegistry = make(map[TaskType]func(map[string]interface{}) (TaskHandler, error))
|
|
|
|
func (r TaskHandlerRegistry) Add(name TaskType, factory func(map[string]interface{}) (TaskHandler, error)) {
|
|
if _, exists := r[name]; exists {
|
|
panic(fmt.Sprintf("cannot register duplicate charger type: %s", name))
|
|
}
|
|
r[name] = factory
|
|
}
|
|
|
|
// func (r TaskHandlerRegistry) Get(name string) (func(map[string]interface{}) (TaskHandler, error), error) {
|
|
// factory, exists := r[name]
|
|
// if !exists {
|
|
// return nil, fmt.Errorf("charger type not registered: %s", name)
|
|
// }
|
|
// return factory, nil
|
|
// }
|
|
|
|
func Get(name TaskType) (func(map[string]interface{}) (TaskHandler, error), error) {
|
|
factory, exists := registry[name]
|
|
if !exists {
|
|
return nil, fmt.Errorf("charger type not registered: %s", name)
|
|
}
|
|
return factory, nil
|
|
}
|