snapshot of https://github.com/ent/ent.git (master, 2026-09-15; blobs > 131072B omitted)

dialect/gremlin/graph/valuemap.go

@@ -0,0 +1,57 @@
+// 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
+}