fix

ecodeclub · Jun 16, 2024 · bcffa07 · bcffa07
1 parent a0ba2eb
commit bcffa07
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/internal/merger/type.go b/internal/merger/type.go
@@ -22,6 +22,7 @@ import (
 	"strings"
 	"time"
 
+	"github.com/ecodeclub/ekit/slice"
 	"github.com/ecodeclub/eorm/internal/merger/internal/errs"
 	"github.com/ecodeclub/eorm/internal/rows"
 )
@@ -67,6 +68,11 @@ func (c ColumnInfo) SelectName() string {
 func (c ColumnInfo) Validate() bool {
 	// ColumnInfo.Name中不能包含括号,也就是聚合函数, name = `id`, 而不是name = count(`id`)
 	// 聚合函数需要写在aggregateFunc字段中
+	aggregateFuncs := []string{"MAX", "MIN", "AVG", "SUM", "COUNT"}
+	if c.AggregateFunc != "" &&
+		!slice.ContainsFunc(aggregateFuncs, func(src string) bool { return strings.EqualFold(src, c.AggregateFunc) }) {
+		return false
+	}
 	return !strings.Contains(c.Name, "(")
 }