Files
ent/entc/gen/template/dialect/sql/group.tmpl
Ariel Mashraki fd0a7f9f02 all: facebookincubator/ent => facebook/ent (#660)
ent repository is going to be migrated to facebook organization
2020-08-18 11:05:08 +03:00

32 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(selector))
}
return selector.Select(columns...).GroupBy({{ $receiver }}.fields...)
}
{{ end }}