如何在Spring Boot Aerospike中实现多集合更新的事务原子性？

Aerospike原生仅保证单键（单记录）操作的原子性，跨表（本质是跨不同键的记录）的原子事务需要通过以下几种方式实现，按推荐优先级排序：

1. 优先重构数据模型（最符合Aerospike设计理念） #

如果业务逻辑允许，将需要原子更新的TableA和TableB中的关联数据合并到同一个记录中，利用Aerospike原生的单记录原子性来保证操作一致性。

比如：

• 把原TableA的user_id和TableB的order_id关联的数据，合并到一个以user_order:{user_id}为键的记录中，将两个表的字段作为该记录的不同bins（字段）。
• 这样单次put操作就能原子更新所有相关字段，无需额外事务逻辑，性能最优。

2. 客户端侧实现两阶段提交（2PC） #

如果无法重构数据模型，可在客户端实现两阶段提交逻辑，确保跨表操作要么全部成功，要么全部回滚。核心步骤如下：

阶段1：预提交（锁定/标记记录） #

对TableA和TableB的目标记录添加事务标记，确保这些记录不会被其他操作修改：

// 生成唯一事务ID
String txnId = UUID.randomUUID().toString();
// 预提交TableA：添加事务ID和状态标记
Key keyA = new Key("your_namespace", "tableA", "target_key_a");
client.put(null, keyA, 
    new Bin("txn_id", txnId), 
    new Bin("txn_status", "PENDING")
);
// 预提交TableB：同样添加事务标记
Key keyB = new Key("your_namespace", "tableB", "target_key_b");
client.put(null, keyB, 
    new Bin("txn_id", txnId), 
    new Bin("txn_status", "PENDING")
);

阶段2：执行实际更新 #

如果预提交全部成功，执行业务逻辑的更新操作，并将事务状态改为COMMITTED：

// 更新TableA的业务字段
client.put(null, keyA, 
    new Bin("field_to_update_a", new_value_a), 
    new Bin("txn_status", "COMMITTED")
);
// 更新TableB的业务字段
client.put(null, keyB, 
    new Bin("field_to_update_b", new_value_b), 
    new Bin("txn_status", "COMMITTED")
);

阶段3：异常回滚 #

如果任何一步更新失败，回滚所有预提交的记录，恢复状态或标记为ROLLED_BACK：

// 回滚TableA
client.put(null, keyA, 
    new Bin("txn_status", "ROLLED_BACK"),
    // 可选：恢复原字段值，需提前在预提交时备份
    new Bin("field_to_update_a", original_value_a)
);
// 回滚TableB
client.put(null, keyB, 
    new Bin("txn_status", "ROLLED_BACK"),
    new Bin("field_to_update_b", original_value_b)
);

关键注意事项 #

• 需添加超时补偿机制：后台定时扫描所有txn_status为PENDING且超过阈值的记录，自动执行回滚，避免因客户端崩溃导致的数据不一致。
• 利用Aerospike的generation（记录版本号）字段，在预提交和更新时校验版本，防止并发修改冲突。

3. TCC事务模式（Try-Confirm-Cancel） #

对于更复杂的场景，可实现TCC模式：

• Try：检查资源可用性，锁定目标记录（类似预提交）。
• Confirm：确认执行业务更新，释放锁定。
• Cancel：如果Confirm失败，回滚所有Try阶段的操作。

这种模式需要引入事务协调器（如本地内存或外部存储）来跟踪事务状态，适合分布式环境下的跨节点跨表事务。

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