Java Spring Boot多实例下基于悲观锁的行级并发控制方案

由于你的应用部署在多实例、无共享缓存的环境下，数据库层面的悲观锁是实现跨JVM/实例行级互斥的可靠方案，完全能满足你同一时间仅一个实例处理特定id消息的需求。下面是具体实现方案和注意事项：

核心思路 #

利用数据库的排他行锁（PESSIMISTIC_WRITE），在查询待处理消息时锁定对应行，其他实例/线程尝试访问同一id的消息时会阻塞等待，直到当前事务释放锁（提交或回滚）。结合状态标记（initial/processing/completed），确保只有未处理的消息会被选中。

具体实现步骤 #

1. 定义JPA Repository并添加带悲观锁的查询 #

在Repository接口中，使用@Lock注解指定悲观写锁，同时查询条件限定status = 'initial'，确保只获取未处理的消息：

@Repository
public interface MessageRepository extends JpaRepository<Message, String> {

    // 悲观写锁：锁定查询到的行，其他事务需等待锁释放
    @Lock(LockModeType.PESSIMISTIC_WRITE)
    // 可选：设置锁超时时间（单位毫秒），避免无限阻塞
    @QueryHints({@QueryHint(name = "javax.persistence.lock.timeout", value = "5000")})
    @Query("SELECT m FROM Message m WHERE m.id = :id AND m.status = 'initial'")
    Optional<Message> findByIdAndInitialStatusForUpdate(@Param("id") String id);
}

2. 实现消息处理的事务逻辑 #

在Service层中，将查询、状态更新、消息处理、最终状态更新放在同一事务中，确保锁持有至整个处理流程完成：

@Service
@Transactional
public class MessageProcessingService {

    private final MessageRepository messageRepository;

    public MessageProcessingService(MessageRepository messageRepository) {
        this.messageRepository = messageRepository;
    }

    // 结合重试机制：捕获锁超时异常后自动重试
    @Retryable(value = LockTimeoutException.class, maxAttempts = 3, backoff = @Backoff(delay = 1000))
    public void processMessageById(String id) {
        messageRepository.findByIdAndInitialStatusForUpdate(id)
                .ifPresent(message -> {
                    // 标记为处理中
                    message.setStatus("processing");
                    messageRepository.save(message);

                    // 执行实际消息处理逻辑（注意：避免耗时操作，防止锁持有过久）
                    executeMessageProcessing(message.getMessage());

                    // 标记为处理完成
                    message.setStatus("completed");
                    messageRepository.save(message);
                });
    }

    private void executeMessageProcessing(String content) {
        // 替换为你的消息处理业务代码
        System.out.println("Processing message content: " + content);
    }
}

3. 配置重试机制（可选但推荐） #

如果使用Spring Retry，需在启动类添加@EnableRetry注解，确保@Retryable生效。当其他实例尝试获取锁超时（抛出LockTimeoutException）时，会自动等待指定时间后重试，符合你“等待后重试”的需求。

关键注意事项 #

• 数据库引擎支持：确保使用支持行级锁的数据库引擎（如MySQL InnoDB），MyISAM等不支持行级锁的引擎会退化为表锁，影响性能。
• 事务边界控制：整个处理流程必须在同一事务内，锁会随事务提交/回滚自动释放。避免在事务中执行耗时操作，防止锁持有时间过长导致并发阻塞。
• 异常回滚处理：若消息处理过程中抛出异常，事务会回滚，消息状态恢复为initial，其他实例可重新获取锁处理，避免消息丢失。
• 状态校验冗余：即使加了悲观锁，查询时仍需过滤status = 'initial'，防止因锁超时、事务回滚等场景导致的重复处理。

替代方案参考（可选） #

除了悲观锁，也可以使用乐观锁（通过版本号字段）或分布式锁（如Redis锁），但悲观锁无需额外依赖，且与数据库操作强绑定，在你的场景下更简单可靠。

内容的提问来源于stack exchange，提问作者Sumeet Kumar Yadav