mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
5d988870bc
Summary: Pull Request resolved: https://github.com/facebookexternal/fbc/pull/1552 Pull Request resolved: https://github.com/facebookincubator/ent/pull/93 avoids complete ent load on id query Reviewed By: a8m Differential Revision: D17906002 fbshipit-source-id: 7749a0d7c707858eab368978a1d220b581ea6dbb
43 lines
1.2 KiB
Cheetah
43 lines
1.2 KiB
Cheetah
{{/*
|
|
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.
|
|
*/}}
|
|
|
|
{{ define "dialect/gremlin/select" }}
|
|
{{ $builder := pascal $.Scope.Builder }}
|
|
{{ $receiver := receiver $builder }}
|
|
|
|
func ({{ $receiver }} *{{ $builder }}) gremlinScan(ctx context.Context, v interface{}) error {
|
|
var (
|
|
traversal *dsl.Traversal
|
|
res = &gremlin.Response{}
|
|
)
|
|
if len({{ $receiver }}.fields) == 1 {
|
|
if {{ $receiver }}.fields[0] != {{ $.Package }}.FieldID {
|
|
traversal = {{ $receiver }}.gremlin.Values({{ $receiver }}.fields...)
|
|
} else {
|
|
traversal = {{ $receiver }}.gremlin.ID()
|
|
}
|
|
} else {
|
|
fields := make([]interface{}, len({{ $receiver }}.fields))
|
|
for i, f := range {{ $receiver }}.fields {
|
|
fields[i] = f
|
|
}
|
|
traversal = {{ $receiver }}.gremlin.ValueMap(fields...)
|
|
}
|
|
query, bindings := traversal.Query()
|
|
if err := {{ $receiver }}.driver.Exec(ctx, query, bindings, res); err != nil {
|
|
return err
|
|
}
|
|
if len({{ $receiver }}.fields) == 1 {
|
|
return res.ReadVal(v)
|
|
}
|
|
vm, err := res.ReadValueMap()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
return vm.Decode(v)
|
|
}
|
|
{{ end }}
|