58 lines
1.3 KiB
Go
58 lines
1.3 KiB
Go
// Copyright 2019-present Facebook Inc. All rights reserved.
|
|
// This source code is licensed under the Apache 2.0 license found
|
|
// in the LICENSE file in the root directory of this source tree.
|
|
|
|
package graph
|
|
|
|
import (
|
|
"errors"
|
|
"fmt"
|
|
"reflect"
|
|
|
|
"github.com/mitchellh/mapstructure"
|
|
)
|
|
|
|
// ValueMap models a .valueMap() gremlin response.
|
|
type ValueMap []map[string]any
|
|
|
|
// Decode decodes a value map into v.
|
|
func (m ValueMap) Decode(v any) error {
|
|
rv := reflect.ValueOf(v)
|
|
if rv.Kind() != reflect.Ptr {
|
|
return errors.New("cannot unmarshal into a non pointer")
|
|
}
|
|
if rv.IsNil() {
|
|
return errors.New("cannot unmarshal into a nil pointer")
|
|
}
|
|
|
|
if rv.Elem().Kind() != reflect.Slice {
|
|
v = &[]any{v}
|
|
}
|
|
return m.decode(v)
|
|
}
|
|
|
|
func (m ValueMap) decode(v any) error {
|
|
cfg := mapstructure.DecoderConfig{
|
|
DecodeHook: func(f, t reflect.Kind, data any) (any, error) {
|
|
if f == reflect.Slice && t != reflect.Slice {
|
|
rv := reflect.ValueOf(data)
|
|
if rv.Len() == 1 {
|
|
data = rv.Index(0).Interface()
|
|
}
|
|
}
|
|
return data, nil
|
|
},
|
|
Result: v,
|
|
TagName: "json",
|
|
}
|
|
|
|
dec, err := mapstructure.NewDecoder(&cfg)
|
|
if err != nil {
|
|
return fmt.Errorf("creating structure decoder: %w", err)
|
|
}
|
|
if err := dec.Decode(m); err != nil {
|
|
return fmt.Errorf("decoding value map: %w", err)
|
|
}
|
|
return nil
|
|
}
|