Drizzle ORM 动态查询条件构建的注意事项

在使用 Drizzle ORM 进行动态查询构建时，开发者经常会遇到条件组合的问题。本文将通过一个实际案例，深入分析如何正确构建动态查询条件，特别是处理多条件组合时的注意事项。

问题背景

在开发过程中，我们经常需要根据用户输入动态构建查询条件。例如，在一个类型查询功能中，用户可能选择一种或多种类型进行筛选。初始实现可能会尝试通过连续调用 .where() 方法来添加条件，但这会导致意外的查询行为。

错误实现分析

原始代码中，开发者尝试通过以下方式构建查询：

let query = db.select(...).from(...).leftJoin(...).$dynamic();

if (input?.types?.includes('A')) {
  query = query.where(isNotNull(typesJoinTable.alphaId));
}

if (input?.types?.includes('B')) {
  query = query.where(isNotNull(typesJoinTable.betaId));
}

if (input?.types?.includes('C')) {
  query = query.where(isNotNull(typesJoinTable.charlieId));
}

query = query.where(isNotNull(typesTable.name));

这种实现方式存在两个主要问题：

1. 连续调用 .where() 并不会自动将条件用 AND 连接
2. 当需要 OR 逻辑时，这种实现无法满足需求

正确解决方案

方案一：使用 AND 组合条件

当需要所有条件同时满足时（AND 逻辑），应该将所有条件收集到一个数组中，然后使用 and() 函数组合：

const where: SQL[] = [];

if (input?.types?.includes('A')) {
  where.push(isNotNull(typesJoinTable.alphaId));
}

if (input?.types?.includes('B')) {
  where.push(isNotNull(typesJoinTable.betaId));
}

if (input?.types?.includes('C')) {
  where.push(isNotNull(typesJoinTable.charlieId));
}

where.push(isNotNull(typesTable.name));

let query = db.select(...)
  .from(...)
  .leftJoin(...)
  .where(and(...where));

方案二：混合使用 AND 和 OR

当需要部分条件使用 OR 逻辑时，可以分层组合条件：

const where: SQL[] = [];
const orConditions: SQL[] = [];

if (input?.types?.includes('A')) {
  orConditions.push(isNotNull(typesJoinTable.alphaId));
}

if (input?.types?.includes('B')) {
  orConditions.push(isNotNull(typesJoinTable.betaId));
}

if (input?.types?.includes('C')) {
  orConditions.push(isNotNull(typesJoinTable.charlieId));
}

if (orConditions.length > 0) {
  where.push(or(...orConditions));
}

where.push(isNotNull(typesTable.name));

let query = db.select(...)
  .from(...)
  .leftJoin(...)
  .where(and(...where));

最佳实践建议

1. 明确条件逻辑：在构建查询前，先明确各条件间是 AND 还是 OR 关系
2. 使用条件收集模式：先将条件收集到数组中，再统一应用到查询
3. 分层组合条件：对于复杂的逻辑组合，可以分层处理不同层级的条件
4. 检查生成的SQL：通过日志检查最终生成的SQL是否符合预期

总结

Drizzle ORM 提供了灵活的动态查询构建能力，但需要开发者明确条件间的逻辑关系。通过合理使用 and() 和 or() 函数，可以构建出符合业务需求的复杂查询。记住，连续调用 .where() 并不会自动组合条件，正确的做法是显式地使用条件组合函数。