You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
49 lines
1.0 KiB
49 lines
1.0 KiB
3 years ago
|
package clause
|
||
|
|
||
|
// GroupBy group by clause
|
||
|
type GroupBy struct {
|
||
|
Columns []Column
|
||
|
Having []Expression
|
||
|
}
|
||
|
|
||
|
// Name from clause name
|
||
|
func (groupBy GroupBy) Name() string {
|
||
|
return "GROUP BY"
|
||
|
}
|
||
|
|
||
|
// Build build group by clause
|
||
|
func (groupBy GroupBy) Build(builder Builder) {
|
||
|
for idx, column := range groupBy.Columns {
|
||
|
if idx > 0 {
|
||
|
builder.WriteByte(',')
|
||
|
}
|
||
|
|
||
|
builder.WriteQuoted(column)
|
||
|
}
|
||
|
|
||
|
if len(groupBy.Having) > 0 {
|
||
|
builder.WriteString(" HAVING ")
|
||
|
Where{Exprs: groupBy.Having}.Build(builder)
|
||
|
}
|
||
|
}
|
||
|
|
||
|
// MergeClause merge group by clause
|
||
|
func (groupBy GroupBy) MergeClause(clause *Clause) {
|
||
|
if v, ok := clause.Expression.(GroupBy); ok {
|
||
|
copiedColumns := make([]Column, len(v.Columns))
|
||
|
copy(copiedColumns, v.Columns)
|
||
|
groupBy.Columns = append(copiedColumns, groupBy.Columns...)
|
||
|
|
||
|
copiedHaving := make([]Expression, len(v.Having))
|
||
|
copy(copiedHaving, v.Having)
|
||
|
groupBy.Having = append(copiedHaving, groupBy.Having...)
|
||
|
}
|
||
|
clause.Expression = groupBy
|
||
|
|
||
|
if len(groupBy.Columns) == 0 {
|
||
|
clause.Name = ""
|
||
|
} else {
|
||
|
clause.Name = groupBy.Name()
|
||
|
}
|
||
|
}
|