Kysely中使用VALUES子句实现数据过滤查询

在数据库操作中，我们经常需要处理一组预定义值与数据库中现有数据的对比查询。本文将介绍如何在Kysely这个TypeScript SQL查询构建器中，优雅地实现这种查询模式。

问题场景

假设我们需要检查一组Spotify ID（如'123'、'456'、'789'）中哪些ID尚未存在于数据库的search_result表中。在原生SQL中，我们通常会使用VALUES子句创建一个临时表，然后通过LEFT JOIN和IS NULL条件来筛选出不存在的数据：

SELECT t.spotify_id
FROM (VALUES ('123'),('456'),('789')) AS t(spotify_id)
LEFT JOIN search_result sr ON sr.spotify_id = t.spotify_id
WHERE sr.spotify_id IS NULL;

Kysely实现方案

在Kysely中，我们需要采用稍微不同的方法来实现相同的功能。以下是完整的解决方案：

1. 创建VALUES临时表

首先，我们需要创建一个扩展Kysely的方法来方便地生成VALUES子句：

function values<R extends Record<string, unknown>, A extends string>(
  records: R[],
  alias: A
): ValuesNode<R, A> {
  return new ValuesNode(records, alias)
}

class ValuesNode<R extends Record<string, unknown>, A extends string> {
  readonly #records: R[]
  readonly #alias: A

  constructor(records: R[], alias: A) {
    this.#records = records
    this.#alias = alias
  }

  toOperationNode(): ValuesNodeOperation {
    return {
      kind: 'ValuesNode',
      records: this.#records,
      alias: this.#alias,
    }
  }
}

2. 构建完整查询

有了上述辅助方法后，我们可以构建完整的查询：

const batch = ['123', '456', '789'];

const valuesTable = values(
  batch.map((id) => ({ spotify_id: id })),
  't' // 表别名
);

const query = db
  .with('t', (db) => db.selectFrom(valuesTable).selectAll())
  .selectFrom('t')
  .leftJoin('search_result as sr', (join) =>
    join.onRef('t.spotify_id', '=', 'sr.spotify_id')
  )
  .where('sr.spotify_id', 'is', null)
  .select('t.spotify_id');

实现原理

这种方法的核心在于：

1. 使用VALUES子句创建一个临时内存表，包含我们想要检查的所有Spotify ID
2. 通过WITH子句将这个临时表命名为't'
3. 使用LEFT JOIN将临时表与数据库中的search_result表关联
4. 通过WHERE条件筛选出search_result表中不存在的记录

优势分析

相比直接在SQL中拼接字符串，这种Kysely实现方式具有以下优势：

1. 类型安全：所有字段都有明确的TypeScript类型定义
2. 防止SQL注入：通过参数化查询自动处理
3. 可维护性：代码结构清晰，易于理解和修改
4. 可组合性：可以方便地与其他Kysely查询组合使用

总结

在Kysely中处理预定义值与数据库记录的对比查询，虽然与原生SQL略有不同，但通过扩展Kysely并利用其强大的查询构建能力，我们依然能够实现优雅且类型安全的解决方案。这种方法不仅适用于Spotify ID的检查，也可以推广到其他类似的数据过滤场景中。