go-sample-webpage/vendor/github.com/lestrrat-go/jwx/internal/json/registry.go
2021-11-04 02:14:51 +01:00

53 lines
1 KiB
Go

package json
import (
"reflect"
"sync"
"github.com/pkg/errors"
)
type Registry struct {
mu *sync.RWMutex
data map[string]reflect.Type
}
func NewRegistry() *Registry {
return &Registry{
mu: &sync.RWMutex{},
data: make(map[string]reflect.Type),
}
}
func (r *Registry) Register(name string, object interface{}) {
if object == nil {
r.mu.Lock()
defer r.mu.Unlock()
delete(r.data, name)
return
}
typ := reflect.TypeOf(object)
r.mu.Lock()
defer r.mu.Unlock()
r.data[name] = typ
}
func (r *Registry) Decode(dec *Decoder, name string) (interface{}, error) {
r.mu.RLock()
defer r.mu.RUnlock()
if typ, ok := r.data[name]; ok {
ptr := reflect.New(typ).Interface()
if err := dec.Decode(ptr); err != nil {
return nil, errors.Wrapf(err, `failed to decode field %s`, name)
}
return reflect.ValueOf(ptr).Elem().Interface(), nil
}
var decoded interface{}
if err := dec.Decode(&decoded); err != nil {
return nil, errors.Wrapf(err, `failed to decode field %s`, name)
}
return decoded, nil
}