小调整

model/query.go

@@ -9,6 +9,19 @@ import (
 	"strings"
 )
 
+type count[T Model] struct {
+	t T
+	N int `json:"n,omitempty" db:"n" gorm:"n"`
+}
+
+func (c count[T]) PrimaryKey() string {
+	return c.t.PrimaryKey()
+}
+
+func (c count[T]) Table() string {
+	return c.t.Table()
+}
+
 func pagination[T Model](db dbQuery, ctx context.Context, q QueryCondition, page, pageSize int) (r []T, total int, err error) {
 	if page < 1 || pageSize < 1 {
 		return
@@ -21,6 +34,7 @@ func pagination[T Model](db dbQuery, ctx context.Context, q QueryCondition, page
 		In:     q.In,
 		Group:  q.Group,
 		From:   q.From,
+		Fields: "count(*) n",
 	}
 	if q.Group != "" {
 		qx.Fields = q.Fields
@@ -34,11 +48,11 @@ func pagination[T Model](db dbQuery, ctx context.Context, q QueryCondition, page
 		qx = QueryCondition{
 			From:   qx.From,
 			In:     qx.In,
+			Fields: "count(*) n",
 		}
 	}
-
-	n, err := GetField[T](ctx, "count(*)", qx)
-	total = str.ToInt[int](n)
+	n, err := gets[count[T]](db, ctx, qx)
+	total = n.N
 	if err != nil || total < 1 {
 		return
 	}