k3s/vendor/github.com/rancher/wrangler/pkg/generic/handlers.go
2019-12-15 23:28:25 -07:00

96 lines
1.4 KiB
Go

package generic
import (
"fmt"
"reflect"
"strings"
"k8s.io/apimachinery/pkg/runtime"
)
type handlerEntry struct {
generation int
name string
handler Handler
}
type Handlers struct {
handlers []handlerEntry
}
func (h *Handlers) Handle(key string, obj runtime.Object) (runtime.Object, error) {
var (
errs errors
)
for _, handler := range h.handlers {
newObj, err := handler.handler(key, obj)
if err != nil {
errs = append(errs, &handlerError{
HandlerName: handler.name,
Err: err,
})
}
if newObj != nil {
obj = newObj
}
}
return obj, errs.ToErr()
}
type errors []error
func (e errors) Error() string {
buf := strings.Builder{}
for _, err := range e {
if buf.Len() > 0 {
buf.WriteString(", ")
}
buf.WriteString(err.Error())
}
return buf.String()
}
func (e errors) ToErr() error {
switch len(e) {
case 0:
return nil
case 1:
return e[0]
default:
return e
}
}
func (e errors) Cause() error {
if len(e) > 0 {
return e[0]
}
return nil
}
type handlerError struct {
HandlerName string
Err error
}
func (h handlerError) Error() string {
return fmt.Sprintf("handler %s: %v", h.HandlerName, h.Err)
}
func (h handlerError) Cause() error {
return h.Err
}
func ToName(h interface{}) string {
if str, ok := h.(fmt.Stringer); ok {
return str.String()
}
s := reflect.ValueOf(h).Type().String()
if len(s) > 1 && s[0] == '*' {
return s[1:]
}
return s
}