Files
ent/entc/gen/template/dialect/sql/group.tmpl
Ariel Mashraki 4323141fe2 ent: add license and copyright to template files
Reviewed By: alexsn

Differential Revision: D17149292

fbshipit-source-id: 837de5fad988de1e54438b47584701f2fc35326d
2019-09-01 03:03:15 -07:00

31 lines
1.0 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/group" }}
{{ $builder := pascal $.Scope.Builder }}
{{ $receiver := receiver $builder }}
func ({{ $receiver }} *{{ $builder }}) sqlScan(ctx context.Context, v interface{}) error {
rows := &sql.Rows{}
query, args := {{ $receiver }}.sqlQuery().Query()
if err := {{ $receiver }}.driver.Query(ctx, query, args, rows); err != nil {
return err
}
defer rows.Close()
return sql.ScanSlice(rows, v)
}
func ({{ $receiver }} *{{ $builder }}) sqlQuery() *sql.Selector {
selector := {{ $receiver }}.sql
columns := make([]string, 0, len({{ $receiver }}.fields) + len({{ $receiver}}.fns))
columns = append(columns, {{ $receiver }}.fields...)
for _, fn := range {{ $receiver }}.fns {
columns = append(columns, fn.SQL(selector))
}
return selector.Select(columns...).GroupBy({{ $receiver }}.fields...)
}
{{ end }}