ShedLock 项目中优雅关闭与锁释放问题的分析与解决

背景介绍

在分布式系统中，定时任务的调度是一个常见需求。ShedLock 是一个轻量级的分布式锁库，用于确保定时任务在分布式环境中只在一个节点上执行。然而，在实际应用中，当系统需要优雅关闭时，ShedLock 的锁释放机制可能会遇到一些问题。

问题现象

在 Spring Boot 应用中配置了 ShedLock 后，当系统接收到 SIGTERM 信号时，会出现以下行为序列：

1. 系统开始关闭流程
2. 正在执行的定时任务被终止
3. 系统尝试释放持有的锁
4. 锁释放过程中抛出 LockException
5. 尽管抛出异常，锁实际上已被成功释放

问题分析

从异常堆栈中可以清楚地看到问题的根源：

1. 当系统开始关闭时，Spring 会中断所有正在执行的线程
2. ShedLock 尝试通过 Redis 删除锁节点
3. 由于线程已被中断，Redis 操作被取消
4. 抛出 RedisCommandInterruptedException
5. 最终包装为 LockException 向上抛出

虽然异常被抛出，但 Redis 操作可能在中断前已经完成，因此锁实际上被释放了。这种不一致的状态可能会导致日志中出现错误信息，给运维人员带来困扰。

解决方案

针对这个问题，ShedLock 提供了几种解决方案：

1. 使用 TrackingLockProviderWrapper

ShedLock 提供了一个 TrackingLockProviderWrapper 工具类，它可以跟踪当前持有的锁。在关闭时，可以显式释放这些锁：

@Bean
public LockProvider lockProvider(RedisConnectionFactory connectionFactory) {
    return new TrackingLockProviderWrapper(new RedisLockProvider(connectionFactory));
}

@PreDestroy
public void unlockAll() {
    TrackingLockProviderWrapper.retryOnInterruptedException(() -> 
        lockProvider.getActiveLocks().forEach(Lock::unlock)
    );
}

这种方法确保在关闭时所有活跃的锁都会被正确释放。

2. 配置合理的锁超时时间

确保 lockAtMostFor 参数设置合理，这样即使锁没有被显式释放，也会在超时后自动失效：

@SchedulerLock(name = "taskName", lockAtMostFor = "10s")
public void scheduledTask() {
    // 任务逻辑
}

3. 自定义锁提供者

可以继承现有的锁提供者，重写解锁逻辑，使其在中断情况下更加健壮：

public class RobustRedisLockProvider extends RedisLockProvider {
    @Override
    protected void doUnlock(String lockKey) {
        try {
            super.doUnlock(lockKey);
        } catch (LockException e) {
            if (Thread.currentThread().isInterrupted()) {
                // 如果是中断导致的异常，可以记录日志但不需要处理
                log.warn("锁释放被中断，但可以忽略");
            } else {
                throw e;
            }
        }
    }
}

最佳实践

1. 合理配置锁超时：总是设置 lockAtMostFor 参数，作为锁释放的最终保障
2. 使用跟踪包装器：在生产环境中使用 TrackingLockProviderWrapper
3. 优雅关闭处理：在 @PreDestroy 方法中显式释放所有锁
4. 日志监控：对锁释放异常进行监控，区分中断异常和其他类型的异常

总结

ShedLock 在优雅关闭场景下的锁释放问题主要是由于线程中断导致的。通过使用 TrackingLockProviderWrapper 和合理的配置，可以确保系统在关闭时锁能够被正确释放，避免出现不一致的状态。对于关键业务场景，建议结合多种解决方案，既保证系统的健壮性，又确保良好的可观测性。