DynamoDB单表多模型设计下，单次获取指定多项目的最优方案咨询

针对你在DynamoDB单表设计中跨实体批量获取数据的需求，以下是几种可行方案的对比和推荐：

一、推荐方案：使用低级DynamoDbClient的BatchGetItem API #

由于DynamoDbEnhancedClient对单表多实体映射的局限性，直接使用底层客户端是最灵活且高效的方式，精准获取目标条目，不会返回多余结果，且利用主键索引无全表扫描风险。

实现示例 #

// 构造批量获取请求
BatchGetItemRequest request = BatchGetItemRequest.builder()
    .requestItems(Map.of(
        "APP_TABLE", KeysAndAttributes.builder()
            .keys(List.of(
                // 指定Employee的主键
                Map.of(
                    "pk", AttributeValue.builder().s("EMP_DTL").build(),
                    "sk", AttributeValue.builder().s("E1").build()
                ),
                // 指定Company的主键
                Map.of(
                    "pk", AttributeValue.builder().s("CMPNY_DTL").build(),
                    "sk", AttributeValue.builder().s("C1").build()
                )
            ))
            .build()
    ))
    .build();

// 发送请求并处理结果
BatchGetItemResponse response = dynamoDbClient.batchGetItem(request);
List<Map<String, AttributeValue>> items = response.responses().get("APP_TABLE");

for (Map<String, AttributeValue> item : items) {
    String pkValue = item.get("pk").s();
    if ("EMP_DTL".equals(pkValue)) {
        // 手动将AttributeValue映射为Employee实体
        Employee employee = mapToEmployee(item);
    } else if ("CMPNY_DTL".equals(pkValue)) {
        // 手动将AttributeValue映射为Company实体
        Company company = mapToCompany(item);
    }
}

二、备选方案：适配DynamoDbEnhancedClient的多实体映射 #

若想保留DynamoDbEnhancedClient的便利性，可通过实体类继承+类型识别的方式实现：

步骤1：定义父类统一主键字段 #

@DynamoDbBean
public abstract class BaseEntity {
    private String pk;
    private String sk;

    @DynamoDbPartitionKey
    public String getPk() { return pk; }
    public void setPk(String pk) { this.pk = pk; }

    @DynamoDbSortKey
    public String getSk() { return sk; }
    public void setSk(String sk) { this.sk = sk; }
}

@DynamoDbBean
public class Employee extends BaseEntity {
    private String empName;
    // 员工专属属性的getter/setter
}

@DynamoDbBean
public class Company extends BaseEntity {
    private String companyName;
    // 公司专属属性的getter/setter
}

步骤2：批量获取并转换类型 #

DynamoDbTable<BaseEntity> table = enhancedClient.table("APP_TABLE", TableSchema.fromBean(BaseEntity.class));

// 构造主键列表
List<Key> keys = List.of(
    Key.builder().partitionValue("EMP_DTL").sortValue("E1").build(),
    Key.builder().partitionValue("CMPNY_DTL").sortValue("C1").build()
);

// 执行批量请求
BatchGetItemEnhancedRequest request = BatchGetItemEnhancedRequest.builder()
    .readBatches(
        BatchGetItemEnhancedRequest.Builder.ReadBatch.builder(BaseEntity.class)
            .mappedTableResource(table)
            .keys(keys)
            .build()
    )
    .build();

BatchGetItemEnhancedResponse<BaseEntity> response = enhancedClient.batchGetItem(request);

// 处理结果并转换为对应实体
for (BaseEntity entity : response.resultsForTable(table)) {
    if (entity instanceof Employee) {
        // 处理Employee逻辑
    } else if (entity instanceof Company) {
        // 处理Company逻辑
    }
}

三、关于PartiQL方案的说明 #

你提到的PartiQL语句SELECT * FROM APP_TABLE WHERE pk IN ('EMP_DTL','CMPNY_DTL') AND sk IN ('E1','C1')：

• 不会触发全表扫描：DynamoDB会利用主键索引定位符合条件的条目
• 逻辑局限性：该语句会匹配所有pk和sk的笛卡尔积组合（如EMP_DTL+C1、CMPNY_DTL+E1），若表中存在这类无效条目，会被错误返回。仅当能确保这些组合不存在时，才适合使用。

内容的提问来源于stack exchange，提问作者Kgan