Kysely 动态构建 VALUES 子句的高级用法

问题背景

在使用 Kysely 这个 TypeScript SQL 查询构建器时，开发者经常需要从一组动态数据构建 VALUES 子句。标准的 VALUES 语法在 SQL 中允许我们创建临时表结构，这在许多场景下非常有用，比如批量插入数据或者作为子查询使用。

静态 VALUES 实现

Kysely 文档中展示了静态 VALUES 的实现方式：

const values = sql`(values 
  ('2024-08-10 21:45:00'::TIMESTAMP, '2024-08-11 21:45:00'::TIMESTAMP),
  ('2024-08-11 21:45:00'::TIMESTAMP, '2024-08-12 21:45:00'::TIMESTAMP)
)`.as('t(ts_from, ts_to)')

这种方式对于固定数据量很有效，但当数据是动态变化时就不太适用了。

动态 VALUES 的挑战

开发者尝试通过拼接字符串数组来实现动态 VALUES：

const intervals: [string, string][] = [["one", "two"], ["three", "four"]];
const intervalsStr = intervals.map(([from, to]) => `(${from}, ${to})`).join(",");

然后尝试将其嵌入 SQL 模板：

const values = sql`(values ${sql`${intervalsStr}`})`.as('t(a,b)');

但这种方法会导致 Kysely 将整个字符串视为单个参数，生成的 SQL 不是我们期望的形式。

解决方案

正确的做法是使用 Kysely 的 sql 模板标签函数动态构建每个值元组：

const intervals: [string, string][] = [["one", "two"], ["three", "four"]];

// 动态构建 VALUES 子句
const values = sql`(
  values ${sql.join(
    intervals.map(([from, to]) => sql`(${sql.val(from)}, ${sql.val(to)})`),
    sql`,`
  )}
)`.as('t(a, b)');

// 完整查询
const rows = await db
  .selectFrom(values)
  .select(["t.a", "t.b"])
  .execute();

实现原理

1. sql.join 函数：将多个 SQL 片段用指定的分隔符连接起来
2. sql.val 函数：安全地插入值，防止 SQL 注入
3. 动态映射：通过数组的 map 方法为每对值创建 SQL 片段

这种方法会生成正确的参数化查询：

SELECT "t"."a", "t"."b"
FROM (VALUES ($1, $2), ($3, $4)) AS t(a, b)

安全注意事项

1. 始终使用 sql.val 而不是直接插入值，防止 SQL 注入
2. 对于大量数据，考虑分批处理以避免性能问题
3. 类型系统会确保返回的数据结构与定义的类型一致

扩展应用

这种技术不仅适用于简单的值对，还可以扩展到更复杂的场景：

• 动态构建批量插入语句
• 创建临时数据集用于复杂查询
• 实现行级数据转换和处理

通过掌握 Kysely 的动态 SQL 构建能力，开发者可以创建更加灵活和强大的数据库查询。