mirror of
https://github.com/ent/ent.git
synced 2026-05-24 09:31:56 +03:00
37 lines
1.2 KiB
Cheetah
37 lines
1.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/group" }}
|
|
{{ $builder := pascal $.Scope.Builder }}
|
|
{{ $receiver := receiver $builder }}
|
|
|
|
func ({{ $receiver }} *{{ $builder }}) sqlScan(ctx context.Context, v interface{}) error {
|
|
for _, f := range {{ $receiver }}.fields {
|
|
if !{{ $.Package }}.ValidColumn(f) {
|
|
return &ValidationError{Name: f, err: fmt.Errorf("invalid field %q for group-by", f)}
|
|
}
|
|
}
|
|
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 }}
|