go-sample-webpage/vendor/github.com/lestrrat-go/jwx/internal/iter/mapiter.go

36 lines
912 B
Go
Raw Normal View History

2021-11-04 01:14:51 +00:00
package iter
import (
"context"
"github.com/lestrrat-go/iter/mapiter"
"github.com/pkg/errors"
)
// MapVisitor is a specialized visitor for our purposes.
// Whereas mapiter.Visitor supports any type of key, this
// visitor assumes the key is a string
type MapVisitor interface {
Visit(string, interface{}) error
}
type MapVisitorFunc func(string, interface{}) error
func (fn MapVisitorFunc) Visit(s string, v interface{}) error {
return fn(s, v)
}
func WalkMap(ctx context.Context, src mapiter.Source, visitor MapVisitor) error {
return mapiter.Walk(ctx, src, mapiter.VisitorFunc(func(k, v interface{}) error {
return visitor.Visit(k.(string), v)
}))
}
func AsMap(ctx context.Context, src mapiter.Source) (map[string]interface{}, error) {
var m map[string]interface{}
if err := mapiter.AsMap(ctx, src, &m); err != nil {
return nil, errors.Wrap(err, `mapiter.AsMap failed`)
}
return m, nil
}