mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
9ab0a28ffe
Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/27 Reviewed By: alexsn Differential Revision: D17164536 fbshipit-source-id: f5caa80b05aff1ef8ec30d7a53256b00380d018c
38 lines
1.1 KiB
Cheetah
38 lines
1.1 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 {
|
|
traversal = {{ $receiver }}.gremlin.Values({{ $receiver }}.fields...)
|
|
} 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 }} |