常见的分布式锁--Redis如何实现分布式锁？

IT_WEH_coder

于 2024-08-21 22:53:45 发布

阅读量440

点赞数 3

分类专栏：开发工具文章标签：分布式 redis 数据库

本文链接：https://blog.csdn.net/IT_WEH_coder/article/details/141332518

版权

开发工具专栏收录该内容

18 篇文章 0 订阅

订阅专栏

分布式锁是实现分布式系统中常见的一种技术，可以保证在多个节点同时操作共享资源时的数据一致性和并发控制。Redis作为一种高性能的缓存数据库，也可以用来实现分布式锁。

实现原理

在Redis中可以使用SET命令来实现分布式锁。具体实现如下：

客户端尝试获取锁：客户端向Redis发送SET命令，将某个唯一标识作为键值对应的值存储到Redis中，并设置过期时间，通常使用NX（当键不存在时才设置）和PX（过期时间）选项。如果SET命令成功执行，则表示客户端成功获取到了锁。
锁的过期时间：在设置锁的过期时间时，应该设置一个合理的超时时间来避免死锁。在分布式环境中，如果某个客户端获得锁后发生了故障或者网络异常，导致它没有释放锁，那么其他客户端将无法获取到锁。为了解决这个问题，可以为锁设置一个合理的过期时间，确保即使锁没有被释放，也能在一定时间后自动过期。
释放锁：客户端在完成操作后需要主动释放锁，以便其他客户端可以获取到锁。客户端可以通过DEL命令来删除锁对应的键值对，以释放锁。

示例项目代码

下面是一个使用Spring Boot整合Redis实现分布式锁的示例项目代码：

引入相关依赖：在pom.xml中添加Spring Boot和Redis相关的依赖。

<dependencies>
    <dependency>
        <groupId>org.springframework.boot</groupId>
        <artifactId>spring-boot-starter-data-redis</artifactId>
    </dependency>
</dependencies>

配置Redis连接：在application.properties中配置Redis连接信息。

spring.redis.host=127.0.0.1
spring.redis.port=6379

编写分布式锁工具类：创建一个RedisLockUtil工具类，用于获取和释放分布式锁。

import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.data.redis.core.script.RedisScript;
import org.springframework.stereotype.Component;

import java.util.Collections;
import java.util.concurrent.TimeUnit;

@Component
public class RedisLockUtil {

    private final StringRedisTemplate redisTemplate;

    public RedisLockUtil(StringRedisTemplate redisTemplate) {
        this.redisTemplate = redisTemplate;
    }

    public boolean tryLock(String lockKey, String requestId, long expireTime) {
        RedisScript<String> script = new DefaultRedisScript<>(
                "if redis.call('setnx', KEYS[1], ARGV[1]) == 1 then return redis.call('expire', KEYS[1], ARGV[2]) else return 0 end",
                String.class);

        String result = redisTemplate.execute(script, Collections.singletonList(lockKey), requestId, String.valueOf(expireTime));

        return "OK".equals(result);
    }

    public void releaseLock(String lockKey, String requestId) {
        RedisScript<Long> script = new DefaultRedisScript<>(
                "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end",
                Long.class);

        redisTemplate.execute(script, Collections.singletonList(lockKey), requestId);
    }
}

使用分布式锁：在需要加锁的代码块中使用RedisLockUtil工具类来获取和释放分布式锁。

import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import java.util.UUID;

@RestController
public class LockController {

    private final RedisLockUtil redisLockUtil;

    public LockController(RedisLockUtil redisLockUtil) {
        this.redisLockUtil = redisLockUtil;
    }

    @GetMapping("/lock")
    public String lock() {
        // 生成唯一的请求ID
        String requestId = UUID.randomUUID().toString();
        // 加锁
        if (redisLockUtil.tryLock("lock_key", requestId, 30000L)) {
            try {
                // 执行需要加锁的操作
                // ...
                return "locked";
            } finally {
                // 释放锁
                redisLockUtil.releaseLock("lock_key", requestId);
            }
        } else {
            // 获取锁失败
            return "failed";
        }
    }
}

在上述示例中，我们通过RedisLockUtil工具类来获取和释放分布式锁。在获取锁时，我们使用了Redis的SETNX命令来实现原子性的加锁操作，并通过过期时间来避免锁被长时间占用。在释放锁时，我们使用了Redis的DEL命令来删除锁对应的键值对。

总结

通过Spring Boot和Redis的结合，我们可以很方便地实现分布式锁。在实际应用中，可以根据具体的业务需求和性能要求来优化分布式锁的实现方式。可以考虑使用RedLock算法来增加分布式锁的安全性，或者使用分布式锁的模板类来简化分布式锁的使用。以上只是一个简单的示例，实际使用中需要根据具体的业务需求进行调整和优化。