DynamoDB-Toolbox 中处理复合主键条件检查的最佳实践

在 DynamoDB 数据建模中，复合主键(PK/SK)的设计非常常见，但在 DynamoDB-Toolbox 中如何正确处理这些键的条件检查却是一个值得探讨的话题。本文将深入分析这一问题，并提供两种实用的解决方案。

问题背景

当使用 DynamoDB-Toolbox 的 computeKey 功能动态生成主键时，开发者可能会遇到一个典型问题：无法直接在条件表达式中引用这些计算生成的键(PK/SK)。这是因为这些键存在于 DynamoDB 中，但并未在实体模式(schema)中明确定义。

解决方案一：使用链接(links)替代computeKey

更优雅的解决方案是使用 DynamoDB-Toolbox 的链接功能，将计算逻辑纳入模式定义中：

schema: schema({
  subId: string()
    .key()
    .default(() => ulid()),
  bucketId: string().key(),
  amount: number()
}).and(prevSchema => ({
  pk: string()
    .key()
    .hidden()
    .link<typeof prevSchema>(({ bucketId }) => bucketId)
    .transform(prefix("BUCKET")),
  sk: string()
    .key()
    .hidden()
    .link<typeof prevSchema>(({ subId }) => subId)
    .transform(prefix("SUB")),
}))

这种方法的关键优势在于：

1. 保持了计算键的逻辑
2. 将PK/SK正式纳入模式，使其可用于条件检查
3. 使用hidden()方法保持模式整洁
4. 通过transform应用前缀转换

解决方案二：直接映射原始字段

对于更简单的情况，可以直接将业务字段映射为PK/SK：

schema: schema({
  bucketId: string()
    .key()
    .savedAs('pk')
    .transform(prefix('BUCKET')),
  subId: string()
    .key()
    .default(ulid)
    .savedAs('sk')
    .transform(prefix('SUB'))
})

这种方法的优点在于：

1. 更直接的字段映射关系
2. 条件检查时可以直接使用业务字段名
3. 自动处理前缀转换

事务处理场景

在需要事务处理的复杂场景中，开发者可能需要结合使用computeKey和显式事务操作。例如：

1. 检查父实体(Bucket)是否存在
2. 创建子实体(Sub)并确保唯一性

这种场景下，虽然computeKey无法直接用于条件检查，但可以通过事务API实现原子性操作。

总结

DynamoDB-Toolbox 提供了多种灵活的方式来处理复合主键的条件检查。根据具体场景选择合适的方法：

• 简单映射场景：使用savedAs直接映射字段
• 需要复杂计算逻辑：使用links保持模式完整性
• 事务场景：结合computeKey和显式事务API

理解这些模式的区别和适用场景，将帮助开发者构建更健壮的DynamoDB数据访问层。