mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
c3955a08f1
Summary: Pull Request resolved: https://github.com/facebookincubator/ent/pull/38 Only `IsNil` and `NotNil` predicates are supported this moment Reviewed By: alexsn Differential Revision: D17444976 fbshipit-source-id: 37336fa0bc7749af995933baee2e23bb7366dd78
69 lines
2.2 KiB
Cheetah
69 lines
2.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/sql/decode/one" }}
|
|
{{ $receiver := $.Receiver }}
|
|
|
|
// FromRows scans the sql response data into {{ $.Name }}.
|
|
func ({{ $receiver }} *{{ $.Name }}) FromRows(rows *sql.Rows) error {
|
|
{{- $scan := print "v" $receiver }}
|
|
var {{ $scan }} struct {
|
|
ID {{ if $.ID.IsString }}int{{ else }}{{ $.ID.Type }}{{ end }}
|
|
{{ range $_, $f := $.Fields }}
|
|
{{- pascal $f.Name }} {{ $f.NullType }}
|
|
{{ end }}
|
|
}
|
|
// the order here should be the same as in the `{{ $.Package }}.Columns`.
|
|
if err := rows.Scan(
|
|
&{{ $scan }}.ID,
|
|
{{- range $_, $f := $.Fields }}
|
|
&{{ $scan }}.{{- pascal $f.Name }},
|
|
{{- end }}
|
|
); err != nil {
|
|
return err
|
|
}
|
|
{{ $receiver }}.ID = {{ if $.ID.IsString }}strconv.Itoa({{ $scan }}.ID){{ else }}{{ $scan }}.ID{{ end }}
|
|
{{- range $_, $f := $.Fields }}
|
|
{{- if $f.IsJSON }}
|
|
if value := {{ $scan }}.{{ pascal $f.Name }}; len(value) > 0 {
|
|
if err := json.Unmarshal(value, &{{ $receiver }}.{{ pascal $f.Name }}); err != nil {
|
|
return fmt.Errorf("unmarshal field {{ $f.Name }}: %v", err)
|
|
}
|
|
}
|
|
{{- else if $f.Nillable }}
|
|
{{- if $f.IsTime }}
|
|
{{ $receiver }}.{{ pascal $f.Name }} = &{{ $scan }}.{{ pascal $f.Name }}
|
|
{{- else }}
|
|
if {{ $scan }}.{{- pascal $f.Name }}.Valid {
|
|
{{ $receiver }}.{{ pascal $f.Name }} = new({{ $f.Type }})
|
|
*{{ $receiver }}.{{ pascal $f.Name }} = {{ printf "%s.%s" $scan (pascal $f.Name) | $f.NullTypeField }}
|
|
}
|
|
{{- end }}
|
|
{{- else }}
|
|
{{ $receiver }}.{{ pascal $f.Name }} = {{ printf "%s.%s" $scan (pascal $f.Name) | $f.NullTypeField }}
|
|
{{- end }}
|
|
{{- end }}
|
|
return nil
|
|
}
|
|
{{ end }}
|
|
|
|
{{ define "dialect/sql/decode/many" }}
|
|
{{ $receiver := $.Receiver }}
|
|
{{ $slice := $.Scope.Slice }}
|
|
|
|
// FromRows scans the sql response data into {{ $slice }}.
|
|
func ({{ $receiver }} *{{ $slice }}) FromRows(rows *sql.Rows) error {
|
|
for rows.Next() {
|
|
{{- $scan := print "v" $receiver }}
|
|
{{ $scan }} := &{{ $.Name }}{}
|
|
if err := {{ $scan }}.FromRows(rows); err != nil {
|
|
return err
|
|
}
|
|
*{{ $receiver }} = append(*{{ $receiver }}, {{ $scan }})
|
|
}
|
|
return nil
|
|
}
|
|
{{ end }} |