mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
88 lines
2.8 KiB
Cheetah
88 lines
2.8 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.
|
|
*/}}
|
|
|
|
{{/* gotype: entgo.io/ent/entc/gen.Type */}}
|
|
|
|
{{ define "dialect/gremlin/decode/one" }}
|
|
{{ $receiver := $.Receiver }}
|
|
|
|
// FromResponse scans the gremlin response data into {{ $.Name }}.
|
|
func ({{ $receiver }} *{{ $.Name }}) FromResponse(res *gremlin.Response) error {
|
|
vmap, err := res.ReadValueMap()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{- $scan := print "scan" $receiver }}
|
|
var {{ $scan }} struct {
|
|
ID {{ $.ID.Type }} `json:"id,omitempty"`
|
|
{{ range $f := $.Fields }}
|
|
{{- if $f.IsBlob }}{{ continue }}{{ end }}
|
|
{{- $f.StructField }} {{ if and $f.IsTime (not $f.HasGoType) }}int64{{ else }}{{ if $f.NillableValue }}*{{ end }}{{ $f.Type }}{{ end }} `json:"{{ $f.StorageKey }},omitempty"`
|
|
{{ end }}
|
|
}
|
|
if err := vmap.Decode(&{{ $scan }}); err != nil {
|
|
return err
|
|
}
|
|
{{ $receiver }}.ID = {{ $scan }}.ID
|
|
{{- range $i, $f := $.Fields }}
|
|
{{- if $f.IsBlob }}{{ continue }}{{ end }}
|
|
{{- if and $f.IsTime (not $f.HasGoType) }}
|
|
{{- if $f.Nillable }}
|
|
v{{ $i }} := time.Unix(0, {{ $scan }}.{{ $f.StructField }})
|
|
{{ $receiver }}.{{ $f.StructField }} = &v{{ $i }}
|
|
{{- else }}
|
|
{{ $receiver }}.{{ $f.StructField }} = time.Unix(0, {{ $scan }}.{{ $f.StructField }})
|
|
{{- end }}
|
|
{{- else }}
|
|
{{ $receiver }}.{{ $f.StructField }} = {{ $scan }}.{{ $f.StructField }}
|
|
{{- end }}
|
|
{{- end }}
|
|
return nil
|
|
}
|
|
{{ end }}
|
|
|
|
{{ define "dialect/gremlin/decode/many" }}
|
|
{{ $receiver := $.Receiver }}
|
|
{{ $slice := $.Scope.Slice }}
|
|
|
|
// FromResponse scans the gremlin response data into {{ $slice }}.
|
|
func ({{ $receiver }} *{{ $slice }}) FromResponse(res *gremlin.Response) error {
|
|
vmap, err := res.ReadValueMap()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
{{- $scan := print "scan" $receiver }}
|
|
var {{ $scan }} []struct {
|
|
ID {{ $.ID.Type }} `json:"id,omitempty"`
|
|
{{ range $f := $.Fields }}
|
|
{{- if $f.IsBlob }}{{ continue }}{{ end }}
|
|
{{- $f.StructField }} {{ if and $f.IsTime (not $f.HasGoType) }}int64{{ else }}{{ if $f.NillableValue }}*{{ end }}{{ $f.Type }}{{ end }} `json:"{{ $f.StorageKey }},omitempty"`
|
|
{{ end }}
|
|
}
|
|
if err := vmap.Decode(&{{ $scan }}); err != nil {
|
|
return err
|
|
}
|
|
for _, v := range {{ $scan }} {
|
|
node := &{{ $.Name }}{ID: v.ID}
|
|
{{- range $i, $f := $.Fields }}
|
|
{{- if $f.IsBlob }}{{ continue }}{{ end }}
|
|
{{- if and $f.IsTime (not $f.HasGoType) }}
|
|
{{- if $f.Nillable }}
|
|
v{{ $i }} := time.Unix(0, v.{{ $f.StructField }})
|
|
node.{{ $f.StructField }} = &v{{ $i }}
|
|
{{- else }}
|
|
node.{{ $f.StructField }} = time.Unix(0, v.{{ $f.StructField }})
|
|
{{- end }}
|
|
{{- else }}
|
|
node.{{ $f.StructField }} = v.{{ $f.StructField }}
|
|
{{- end }}
|
|
{{- end }}
|
|
*{{ $receiver }} = append(*{{ $receiver }}, node)
|
|
}
|
|
return nil
|
|
}
|
|
{{ end }}
|