如何用Ent构建包含Customer与Order表字段的动态查询？

我正尝试使用Ent构建动态查询，查询字段会随用户输入变化。现有Customer、Account、Order三张关联表，需要同时返回Customer表与Order表的字段，表结构及目标SQL如下：

表结构 #

Customer Table #

Account Table #

Order Table #

目标SQL #

SELECT customer_Id, first_name, last_name, order_num, product_num   
FROM customer c  
INNER JOIN account a ON c.account_id = a.account_id  
INNER JOIN `order` o ON a.payment_id = o.payment_id  
WHERE c.customer_id = 123 OR c.first_name = 'Mike' OR c.last_name = 'Lee' OR c.account_id = '321'

现有代码与问题 #

我已经用ent.Predicate构建了动态条件，但当前两种查询方式都无法满足需求：

var predicateCustomer []predicate.Customer  

if searchFields.CustomerId != "" {  
    predicateCustomer = append(predicateCustomer, customer.CustomerIdEQ(searchFields.CustomerId))  
}  

if searchFields.FirstName != "" {  
    predicateCustomer = append(predicateCustomer, customer.FirstNameEQ(searchFields.FirstName))  
}  

if searchFields.LastName != "" {  
    predicateCustomer = append(predicateCustomer, customer.LastNameEQ(searchFields.LastName))  
}  

if searchFields.AccountId != "" {  
      predicateCustomer = append(predicateCustomer, customer.AccountIdEQ(searchFields.AccountId))  
}  

// 方法1：仅返回Customer表字段
item, err := e.client.Customer.Query().
    Where(predicateCustomer...).
    Limit(30).
    Offset(0).
    All(ctx)

// 方法2：仅返回Order表字段
item, err := e.client.Customer.Query().
    Where(predicateCustomer...).QueryAccount().QueryOrder().
    Limit(30).
    Offset(0).
    All(ctx)

• 方法1仅返回Customer表的字段，无法获取Order数据；
• 方法2仅返回Order表的字段，丢失了Customer信息。

1. 修正动态条件的逻辑（OR组合） #

首先注意：你当前的动态条件会被Ent默认用AND连接，和目标SQL的OR逻辑不符，需要用predicate.Or包裹所有条件：

var conditions []predicate.Customer

if searchFields.CustomerId != "" {
    conditions = append(conditions, customer.CustomerIdEQ(searchFields.CustomerId))
}
if searchFields.FirstName != "" {
    conditions = append(conditions, customer.FirstNameEQ(searchFields.FirstName))
}
if searchFields.LastName != "" {
    conditions = append(conditions, customer.LastNameEQ(searchFields.LastName))
}
if searchFields.AccountId != "" {
    conditions = append(conditions, customer.AccountIdEQ(searchFields.AccountId))
}

// 用OR组合所有筛选条件
var customerPredicate predicate.Customer
if len(conditions) > 0 {
    customerPredicate = predicate.Or(conditions...)
}

2. 预加载关联数据，通过实体对象访问字段 #

利用Ent的With方法预加载关联的Account和Order数据，查询Customer时会同时拉取关联表的信息，之后可以通过实体的Edges字段访问：

// 查询Customer并预加载Account及关联的Order
customers, err := e.client.Customer.Query().
    Where(customerPredicate).
    // 预加载Account，同时预加载Account关联的Order
    WithAccount(func(a *ent.AccountQuery) {
        a.WithOrder()
    }).
    Limit(30).
    Offset(0).
    All(ctx)

if err != nil {
    // 错误处理逻辑
}

// 遍历结果，提取Customer和Order字段
for _, cust := range customers {
    // Customer字段
    custID := cust.CustomerId
    firstName := cust.FirstName
    lastName := cust.LastName

    // 获取关联的Account
    if account := cust.Edges.Account; account != nil {
        // 获取Account关联的Order列表
        for _, order := range account.Edges.Order {
            orderNum := order.OrderNum
            productNum := order.ProductNum
            // 处理业务逻辑
        }
    }
}

3. 自定义结构体映射扁平结果（贴近目标SQL） #

如果需要直接得到类似SQL查询的扁平结构，可以用Join+Scan将结果映射到自定义结构体：

步骤1：定义结构体

type CustomerOrder struct {
    CustomerID  string `json:"customer_id"`
    FirstName   string `json:"first_name"`
    LastName    string `json:"last_name"`
    OrderNum    string `json:"order_num"`
    ProductNum  string `json:"product_num"`
}

步骤2：执行查询并扫描结果

var results []CustomerOrder

err := e.client.Customer.Query().
    Where(customerPredicate).
    // 关联Account表
    Join("INNER JOIN account AS a ON customer.account_id = a.account_id").
    // 关联Order表（注意order是SQL关键字，需要加反引号）
    Join("INNER JOIN `order` AS o ON a.payment_id = o.payment_id").
    // 指定需要查询的字段
    Select(
        customer.FieldCustomerId,
        customer.FieldFirstName,
        customer.FieldLastName,
        ent.Expr("o.order_num"),
        ent.Expr("o.product_num"),
    ).
    // 将结果扫描到自定义结构体
    Scan(ctx, &results)

if err != nil {
    // 错误处理逻辑
}

// results中就是同时包含Customer和Order字段的扁平结果集

内容的提问来源于stack exchange，提问作者wltz