74 lines
1.1 KiB
Go
74 lines
1.1 KiB
Go
|
package wp
|
||
|
|
|
||
|
|
import (
|
||
|
|
"errors"
|
||
|
|
)
|
||
|
|
|
||
|
|
var fnMap map[string]map[string]any
|
||
|
|
var fnHook map[string]map[string]any
|
||
|
|
|
||
|
|
func GetFn[T any](fnType string, name string) []T {
|
||
|
|
v, ok := fnMap[fnType]
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
vv, ok := v[name]
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return vv.([]T)
|
||
|
|
}
|
||
|
|
func GetFnHook[T any](fnType string, name string) []T {
|
||
|
|
v, ok := fnHook[fnType]
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
vv, ok := v[name]
|
||
|
|
if !ok {
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
return vv.([]T)
|
||
|
|
}
|
||
|
|
|
||
|
|
func PushFn[T any](fnType string, name string, fns ...T) error {
|
||
|
|
v, ok := fnMap[fnType]
|
||
|
|
if !ok {
|
||
|
|
v = make(map[string]any)
|
||
|
|
fnMap[fnType] = v
|
||
|
|
v[name] = fns
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
vv, ok := v[name]
|
||
|
|
if !ok || vv == nil {
|
||
|
|
v[name] = fns
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
s, ok := vv.([]T)
|
||
|
|
if ok {
|
||
|
|
s = append(s, fns...)
|
||
|
|
v[name] = s
|
||
|
|
}
|
||
|
|
return errors.New("error fn type")
|
||
|
|
}
|
||
|
|
|
||
|
|
func PushFnHook[T any](fnType string, name string, fns ...T) error {
|
||
|
|
v, ok := fnHook[fnType]
|
||
|
|
if !ok {
|
||
|
|
v = make(map[string]any)
|
||
|
|
fnHook[fnType] = v
|
||
|
|
v[name] = fns
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
vv, ok := v[name]
|
||
|
|
if !ok || vv == nil {
|
||
|
|
v[name] = fns
|
||
|
|
return nil
|
||
|
|
}
|
||
|
|
s, ok := vv.([]T)
|
||
|
|
if ok {
|
||
|
|
s = append(s, fns...)
|
||
|
|
v[name] = s
|
||
|
|
}
|
||
|
|
return errors.New("error fn type")
|
||
|
|
}
|