36 lines
912 B
Go
36 lines
912 B
Go
|
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
|
||
|
|
}
|