聚合

分组

对 users 按 name 和 age 字段分组，并计算 age 的总和。

  1. package main
  3. import (
  4.     "context"
  6.     "<project>/ent"
  7.     "<project>/ent/user"
  8. )
  10. func Do(ctx context.Context, client *ent.Client) {
  11.     var v []struct {
  12.         Name  string `json:"name"`
  13.         Age   int    `json:"age"`
  14.         Sum   int    `json:"sum"`
  15.         Count int    `json:"count"`
  16.     }
  17.     err := client.User.Query().
  18.         GroupBy(user.FieldName, user.FieldAge).
  19.         Aggregate(ent.Count(), ent.Sum(user.FieldAge)).
  20.         Scan(ctx, &v)
  21. }

按单个字段分组.

  1. package main
  3. import (
  4.     "context"
  6.     "<project>/ent"
  7.     "<project>/ent/user"
  8. )
  10. func Do(ctx context.Context, client *ent.Client) {
  11.     names, err := client.User.
  12.         Query().
  13.         GroupBy(user.FieldName).
  14.         Strings(ctx)
  15. }

根据边进行分组

如果您想按照自己的逻辑进行聚合，可以使用自定义聚合函数。

下面展示了：如何根据用户的 id 和 name 进行分组，并计算其宠物的平均 age。

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  7.     "<project>/ent"
  8.     "<project>/ent/pet"
  9.     "<project>/ent/user"
  10. )
  12. func Do(ctx context.Context, client *ent.Client) {
  13.     var users []struct {
  14.         ID      int
  15.         Name    string
  16.         Average float64
  17.     }
  18.     err := client.User.Query().
  19.         GroupBy(user.FieldID, user.FieldName).
  20.         Aggregate(func(s *sql.Selector) string {
  21.             t := sql.Table(pet.Table)
  22.             s.Join(t).On(s.C(user.FieldID), t.C(pet.OwnerColumn))
  23.             return sql.As(sql.Avg(t.C(pet.FieldAge)), "average")
  24.         }).
  25.         Scan(ctx, &users)
  26. }

Having + Group By

自定义 SQL 修饰符 可以帮助您控制所有查询。 下面展示了如何检索每个角色最老 (即 age 最大) 的用户。

  1. package main
  3. import (
  4.     "context"
  5.     "log"
  7.     "entgo.io/ent/dialect/sql"
  8.     "<project>/ent"
  9.     "<project>/ent/user"
  10. )
  12. func Do(ctx context.Context, client *ent.Client) {
  13.     var users []struct {
  14.         Id      Int
  15.         Age     Int
  16.         Role    string
  17.     }
  18.     err := client.User.Query().
  19.         Modify(func(s *sql.Selector) {
  20.             s.GroupBy(user.Role)
  21.             s.Having(
  22.                 sql.EQ(
  23.                     user.FieldAge,
  24.                     sql.Raw(sql.Max(user.FieldAge)),
  25.                 ),
  26.             )
  27.         }).
  28.         ScanX(ctx, &users)
  29. }

注意： The sql.Raw 至关重要。 It tells the predicate that sql.Max is not an argument.

上述代码会生成以下SQL查询：

  1. SELECT * FROM user GROUP BY user.role HAVING user.age = MAX(user.age)