Spring Boot保存DynamoDB实体时触发ValidationException问题

问题背景 #

我正在开发基于Spring Boot的AWS DynamoDB CRUD应用，初始代码结构如下：

实体类 #

// getter, setter, args, no args annotation here
@DynamoDBTable(tableName = "TableName")
public class Entity {

    @DynamoDBHashKey(attributeName = "PartitionKey")
    private String key; 

    @DynamoDBRangeKey(attributeName = "SortKey")
    private String tempSortKey; 

    @DynamoDBAttribute(attributeName = "Value")
    private String tempValue;

    @DynamoDBAttribute(attributeName = "TerminationDate")
    private Date tempTerminationDate;
 
    // 其他属性
}

Repository #

public Entity saveEntity(Entity entity) {
    dynamoDBMapper.save(entity);
    return entity;
}

Controller #

// Controller Class
@PostMapping(value="URL")
public Entity saveItem(@RequestBody Entity entity) {
    return repository.saveEntity(entity);
}

错误信息 #

保存操作抛出异常：

"Unable to unmarshall exception response with the unmarshallers provided (Service: AmazonDynamoDBvZ; Status Code: 400; Error Code: ValidationException"

已知条件与需求 #

• DynamoDB表仅定义了PartitionKey（字符串类型），未设置Sort Key，且无权修改表结构
• 业务需求：key字段可能存在重复值，tempSortKey字段始终唯一，需要通过tempSortKey区分同key下的不同条目
• 已尝试两种可行但不符合预期的方案：
  • 新增自动生成的唯一键作为HashKey，原key转为普通属性
  • 将key同时标记为HashKey和普通属性

解决方案 #

核心问题分析 #

初始实体类中标记了@DynamoDBRangeKey，但实际表并未定义RangeKey，DynamoDB服务端校验时认为请求不符合表结构，因此抛出ValidationException。

1. 调整实体类 #

移除tempSortKey上的@DynamoDBRangeKey注解，将其改为普通属性：

// getter, setter, args, no args annotation here
@DynamoDBTable(tableName = "TableName")
public class Entity {

    @DynamoDBHashKey(attributeName = "PartitionKey")
    private String key; 

    @DynamoDBAttribute(attributeName = "SortKey") // 改为普通属性
    private String tempSortKey; 

    @DynamoDBAttribute(attributeName = "Value")
    private String tempValue;

    @DynamoDBAttribute(attributeName = "TerminationDate")
    private Date tempTerminationDate;
 
    // 其他属性
}

2. 添加保存时的唯一性校验 #

在Repository的save方法中，使用SaveExpression添加条件表达式，确保同一个key下的tempSortKey唯一：

public Entity saveEntity(Entity entity) {
    SaveExpression saveExpression = new SaveExpression();
    Map<String, ExpectedAttributeValue> expectedConditions = new HashMap<>();
    // 条件：当前key下不存在指定的tempSortKey时，才允许保存
    expectedConditions.put("SortKey", new ExpectedAttributeValue().withExists(false));
    saveExpression.setExpected(expectedConditions);
    
    try {
        dynamoDBMapper.save(entity, saveExpression);
        return entity;
    } catch (ConditionalCheckFailedException e) {
        // 处理重复场景，可自定义业务异常或返回提示
        throw new RuntimeException("当前key下的SortKey已存在", e);
    }
}

3. 长期优化建议（若有权限修改表结构） #

如果后续能修改表结构，建议将表主键改为复合主键（HashKey为key，RangeKey为tempSortKey），这样DynamoDB原生就能保证每个HashKey下的RangeKey唯一，无需手动编写条件表达式。

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