Drizzle ORM 查询优化：实现高效的分页计数功能

在数据库操作中，分页查询并获取总记录数是一个常见需求。Drizzle ORM 作为一款现代化的 TypeScript ORM 工具，其查询 API 设计简洁高效，但在处理分页计数场景时仍存在优化空间。

传统实现方式的痛点

开发者在使用 Drizzle ORM 进行分页查询时，通常需要同时执行两个操作：

1. 获取当前页的数据列表
2. 计算符合条件的总记录数

传统实现方式需要编写重复的查询条件，代码冗余且容易出错。例如：

const [data, count] = await Promise.all([
  db.query.users.findMany({
    where: { active: true },
    limit: 10,
    offset: 0
  }),
  db.select({ count: count() })
    .from(users)
    .where(eq(users.active, true))
]);

这种方式虽然功能完整，但在实际项目中会带来以下问题：

• 查询条件需要重复编写
• 代码可读性降低
• 维护成本增加

解决方案探索

Drizzle ORM 团队已经意识到了这一需求，并提供了 db.$count 方法作为基础解决方案。这个方法专门用于简化行数统计操作，可以与关系查询构建器(RQB)API配合使用。

db.$count 的基本用法如下：

const total = await db.$count()
  .from(users)
  .where(eq(users.active, true));

虽然这解决了简单的计数问题，但对于完整的分页计数场景，开发者仍需要结合 findMany 使用。更优雅的解决方案是期待 Drizzle ORM 提供原生的 findManyAndCount 方法。

最佳实践建议

在当前版本下，我们可以通过以下方式优化分页计数实现：

1. 封装工具函数：创建一个通用的分页查询工具函数，统一处理查询条件和结果组装

async function paginateWithCount(
  queryBuilder: any,
  options: { where?: any; limit: number; offset: number }
) {
  const [data, total] = await Promise.all([
    queryBuilder.findMany(options),
    db.$count()
      .from(queryBuilder.meta.table)
      .where(options.where)
  ]);
  return { data, total };
}

2. 类型安全扩展：通过 TypeScript 类型扩展增强查询体验

declare module 'drizzle-orm' {
  interface DrizzleQueryBuilder {
    findManyAndCount: (options: FindManyOptions) => Promise<[any[], number]>;
  }
}

3. 性能优化：对于大数据量表，考虑使用近似计数或缓存策略

未来展望

随着 Drizzle ORM 的持续发展，我们期待官方能够提供更多便捷的查询方法，如：

• 原生的 findManyAndCount 方法
• 简化主键查询的 findByPK 方法
• 存在性检查的 exists 方法

这些改进将进一步提升开发体验，减少样板代码，使 Drizzle ORM 在各种应用场景中更加高效易用。

对于需要频繁处理分页计数的项目，建议关注 Drizzle ORM 的更新日志，及时获取最新的 API 改进信息，同时合理封装现有功能以提升开发效率。