无需使用GetItem和Sort key，从DynamoDB表中获取单个值

方案1：用ScanEnhancedRequest加limit（适配现有表结构） #

Java增强版V2的ScanEnhancedRequest支持设置limit，完全不需要排序键，刚好匹配你当前的单分区键表结构，性能比拉取1MB数据再过滤好得多。

代码示例：

// 实体类定义
@DynamoDbTable(tableName = "available_account_numbers")
public class AvailableAccountNumber {
    private Long accountNumber;

    @DynamoDbPartitionKey
    public Long getAccountNumber() {
        return accountNumber;
    }

    public void setAccountNumber(Long accountNumber) {
        this.accountNumber = accountNumber;
    }
}

// 取号并删除逻辑
public Long getAndRemoveAccountNumber(DynamoDbEnhancedClient enhancedClient) {
    DynamoDbTable<AvailableAccountNumber> table = enhancedClient.table(
        "available_account_numbers", 
        TableSchema.fromBean(AvailableAccountNumber.class)
    );

    // 构建扫描请求，只返回1条记录
    ScanEnhancedRequest scanRequest = ScanEnhancedRequest.builder()
            .limit(1)
            .consistentRead(true) // 开启强一致读，避免读取已被删除的记录
            .build();

    Iterator<AvailableAccountNumber> iterator = table.scan(scanRequest).items().iterator();
    if (!iterator.hasNext()) {
        return null; // 无可用账号
    }

    AvailableAccountNumber target = iterator.next();
    table.deleteItem(target);
    return target.getAccountNumber();
}

方案2：调整表结构为单分区键+排序键（性能最优） #

如果你的数据量极大，扫描的性能满足不了需求，可以修改表结构：新增一个固定值的分区键（比如group_id，所有记录值设为ACCOUNT_POOL），把原来的account_number设为排序键。这样就能用QueryEnhancedRequest快速取号，查询性能远高于扫描。

调整后的表结构：

代码示例：

// 调整后的实体类
@DynamoDbTable(tableName = "available_account_numbers")
public class AvailableAccountNumber {
    private String groupId;
    private Long accountNumber;

    @DynamoDbPartitionKey
    public String getGroupId() {
        return groupId;
    }

    public void setGroupId(String groupId) {
        this.groupId = groupId;
    }

    @DynamoDbSortKey
    public Long getAccountNumber() {
        return accountNumber;
    }

    public void setAccountNumber(Long accountNumber) {
        this.accountNumber = accountNumber;
    }
}

// 查询取号逻辑
public Long getAndRemoveAccountNumber(DynamoDbEnhancedClient enhancedClient) {
    DynamoDbTable<AvailableAccountNumber> table = enhancedClient.table(
        "available_account_numbers", 
        TableSchema.fromBean(AvailableAccountNumber.class)
    );

    QueryEnhancedRequest queryRequest = QueryEnhancedRequest.builder()
            .queryConditional(QueryConditional.keyEqualTo(Key.builder()
                    .partitionValue("ACCOUNT_POOL")
                    .build()))
            .limit(1)
            .consistentRead(true)
            .build();

    Iterator<AvailableAccountNumber> iterator = table.query(queryRequest).items().iterator();
    if (!iterator.hasNext()) {
        return null;
    }

    AvailableAccountNumber target = iterator.next();
    table.deleteItem(target);
    return target.getAccountNumber();
}

预填充数据时，只需要给每条记录的groupId设为固定值即可，成本极低。

方案3：随机取号+事务删除（适合随机分配场景） #

如果需要随机分配账号而不是按顺序取，可以用随机生成+条件删除的方式：随机生成一个账号范围里的数字，尝试删除它，成功则返回，失败则重试（说明该账号已被取走）。

代码示例：

public Long getRandomAccountNumber(DynamoDbClient client) {
    // 假设预填充的账号范围是1到1000000
    long min = 1;
    long max = 1000000;

    while (true) {
        long randomNum = ThreadLocalRandom.current().nextLong(min, max + 1);
        String deleteSql = "DELETE FROM available_account_numbers WHERE account_number = ?";
        
        ExecuteStatementRequest request = ExecuteStatementRequest.builder()
                .statement(deleteSql)
                .parameters(List.of(AttributeValue.builder().n(String.valueOf(randomNum)).build()))
                .build();

        try {
            ExecuteStatementResponse response = client.executeStatement(request);
            // 删除成功则返回该账号
            if (response.attributes() != null && !response.attributes().isEmpty()) {
                return randomNum;
            }
        } catch (ResourceNotFoundException e) {
            // 账号已被取走，继续重试
            continue;
        }
    }
}

这个方案适合数据量充足的场景，当剩余账号较少时，重试次数会增加，需要根据实际情况调整。

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