揭开 SpringBoot 定时任务优化之谜：如何利用 Redis 分布式锁让任务井然有序

后端

2023-11-03 21:11:47

分布式定时任务：用 Redis 分布式锁优化你的 SpringBoot 定时任务

在分布式或集群环境中运行 SpringBoot 定时任务时，你可能会遇到一个常见的挑战：任务重复触发 。这可能会导致系统性能问题，甚至数据不一致。为了解决这个问题，分布式锁 可以派上用场。

分布式锁：协调分布式任务的守护者

分布式锁是一种协调机制，它确保在同一时间段内只有一个服务器节点可以执行特定任务。这类似于你在现实生活中看到的停车锁。当你把车停在一个停车位时，你会放置一个锁来防止其他车辆占用这个停车位。

在分布式环境中，Redis 是实现分布式锁的一种流行选择。Redis 是一个内存内键值数据库，可以快速可靠地处理大量请求。它提供了一个名为 SETNX（如果不存在，则设置）的命令，可以用来创建分布式锁。

使用 Redis 实现分布式锁

步骤 1：获取锁

在执行定时任务之前，使用 SETNX 命令获取分布式锁。该命令会检查一个键是否存在。如果键不存在，它会将键值对存储在 Redis 中并返回 1，表示锁已获取。

if (redisTemplate.opsForValue().setIfAbsent(LOCK_KEY, "true")) {
    // 获取锁成功，执行任务
}

步骤 2：执行任务

如果获取锁成功，这意味着你拥有了执行定时任务的独占权。现在你可以安全地执行任务，而不用担心其他节点同时执行相同的任务。

步骤 3：释放锁

在完成定时任务后，释放分布式锁至关重要。这将允许其他节点获取锁并执行任务。使用 DEL 命令从 Redis 中删除键来释放锁。

redisTemplate.delete(LOCK_KEY);

代码示例

以下是使用 Redis 分布式锁优化 SpringBoot 定时任务的代码示例：

@Component
public class MyScheduledTask {

    @Autowired
    private RedisTemplate<String, String> redisTemplate;

    private static final String LOCK_KEY = "my_scheduled_task_lock";

    @Scheduled(cron = "0 0/1 * * * ?")
    public void run() {
        if (redisTemplate.opsForValue().setIfAbsent(LOCK_KEY, "true")) {
            try {
                // 执行定时任务
                System.out.println("定时任务正在执行...");
                Thread.sleep(10000);
            } catch (InterruptedException e) {
                e.printStackTrace();
            } finally {
                redisTemplate.delete(LOCK_KEY);
            }
        }
    }
}