Kysely 项目中实现多计数与去重计数的查询方法

Kysely 是一个类型安全的 SQL 查询构建器，它提供了强大的功能来构建复杂的 SQL 查询。在实际开发中，我们经常需要同时获取表的总记录数和某个字段的去重计数，这在数据分析场景中尤为常见。

基本查询需求

假设我们有一个用户表 tbl_user，需要同时查询：

1. 表中的总用户数
2. 不同职位(position)的总数

传统SQL写法如下：

SELECT 
  COUNT(*) AS total_users, 
  COUNT(DISTINCT position) AS total_positions 
FROM tbl_user;

Kysely 的实现方式

Kysely 提供了内置的方法来实现这种多计数查询。核心在于使用 fn 函数构建器来创建聚合函数表达式：

const result = await db
  .selectFrom("tbl_user")
  .select(({ fn }) => [
    fn.countAll().as("total_users"),
    fn.count("position").distinct().as("total_positions"),
  ])
  .execute();

技术解析

1. fn.countAll()
   对应SQL中的 COUNT(*)，计算表中的所有记录数

2. fn.count().distinct()
   这种链式调用方式对应SQL中的 COUNT(DISTINCT column)，实现了对指定列的去重计数

3. .as() 方法
   为每个计数结果指定别名，方便在结果对象中访问

实际应用场景

这种多计数查询在以下场景特别有用：

• 数据概览仪表盘
• 统计报表生成
• 数据质量检查（如检查字段的唯一值数量）
• 分页前的数据总量统计

类型安全优势

Kysely 的类型系统能够自动推断查询结果的类型。对于上面的查询，TypeScript 会自动推断出返回结果的类型为：

{
  total_users: number;
  total_positions: number;
}

这避免了手动定义类型的麻烦，也减少了类型错误的可能性。

性能考虑

虽然这种查询方式简洁明了，但在大数据量表上使用时需要注意：

1. COUNT(DISTINCT) 操作可能比简单的 COUNT 更耗资源
2. 如果只需要计数而不需要具体数据，确保不要添加不必要的字段到SELECT中
3. 对计数字段添加适当的索引可以显著提高查询性能

通过Kysely的这种查询构建方式，开发者可以既保持代码的类型安全，又能以声明式的方法构建复杂的计数查询，大大提高了开发效率和代码的可维护性。