redis分布式锁的两种实现方式(redis和spring-data-redis)

一.redis实现

1.引入maven依赖

        <dependency>
            <groupId>redis.clients</groupId>
            <artifactId>jedis</artifactId>
            <version>2.9.0</version>
        </dependency>

2.分布式锁工具类

public class RedisTool {

    private static final String LOCK_SUCCESS = "OK";
    private static final String SET_IF_NOT_EXIST = "NX";
    private static final String SET_WITH_EXPIRE_TIME = "PX";
    private static final Long RELEASE_SUCCESS = 1L;

    /**
     * 获取分布式锁
     *
     * @param jedis      Redis客户端
     * @param lockKey    锁
     * @param requestId  请求标识
     * @param expireTime 超期时间
     * @return 是否获取成功
     */
    public static boolean getDistributedLock(Jedis jedis, String lockKey, String requestId, int expireTime) {


        String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
        if (LOCK_SUCCESS.equals(result)) {
            return true;
        }
        return false;

    }

    /**
     * 释放分布式锁
     *
     * @param jedis     Redis客户端
     * @param lockKey   锁
     * @param requestId 请求标识
     * @return 是否释放成功
     */
    public static boolean releaseDistributedLock(Jedis jedis, String lockKey, String requestId) {

        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        // redis单机模式
        Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
        // redis集群模式
//        Object result = JedisCluster.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));

        if (RELEASE_SUCCESS.equals(result)) {
            return true;
        }
        return false;
    }
}

二.spring-data-redis方式实现

1.工具类

@Service
public class RedisUtil {

	@Resource
	private StringRedisTemplate stringRedisTemplate;

	// 获取成功结果
	private static final String LOCK_SUCCESS = "OK";
	// 不存在,则 SET
	private static final String SET_IF_NOT_EXIST = "NX";
	// 过期时间毫秒值
	private static final String SET_WITH_EXPIRE_TIME = "PX";
	private static final Long RELEASE_SUCCESS = 1L;

	/**
	 * 获取分布式锁
	 *
	 * @param lockKey    锁
	 * @param requestId  请求标识(锁拥有者标识)
	 * @param expireTime 超期时间
	 * @return 是否获取成功
	 */
	public boolean getDistributedLock(String lockKey, String requestId, int expireTime) {
		// 通过RedisTemplate的execute方法,获取到jedis里操作命令的对象
		Boolean b = stringRedisTemplate.execute(new RedisCallback<Boolean>() {
			@Override
			public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
				JedisCommands commands = (JedisCommands) connection.getNativeConnection();
				String result = commands.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
				if (LOCK_SUCCESS.equals(result)) {
					return true;
				}
				return false;
			}
		});

		return b;
	}

	/**
	 * 释放分布式锁
	 *
	 * @param lockKey   锁
	 * @param requestId 请求标识
	 * @return 是否释放成功
	 */
	public boolean releaseDistributedLock(String lockKey, String requestId) {
		// Lua 脚本来达到释放锁的原子操作
		String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";

		Boolean b = stringRedisTemplate.execute(new RedisCallback<Boolean>() {
			@Override
			public Boolean doInRedis(RedisConnection connection) throws DataAccessException {
				Object nativeConnection = connection.getNativeConnection();
				// 集群模式和单机模式接口不同,按模式执行
				// 单机模式
				if(nativeConnection instanceof Jedis){
					Object result = ((Jedis) nativeConnection).eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
					if (RELEASE_SUCCESS.equals(result)) {
						return true;
					}
					return false;
				}
				// 集群模式
				if(nativeConnection instanceof JedisCluster){
					Object result = ((JedisCluster) nativeConnection).eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
					if (RELEASE_SUCCESS.equals(result)) {
						return true;
					}
					return false;
				}
				return false;
			}
		});

		return b;
	}
}

2.可能抛类型转换异常,去除lettuce依赖

        <dependency>
            <groupId>org.springframework.boot</groupId>
            <artifactId>spring-boot-starter-data-redis</artifactId>
            <exclusions>
                <!-- 不依赖Redis的异步客户端lettuce -->
                <exclusion>
                    <groupId>io.lettuce</groupId>
                    <artifactId>lettuce-core</artifactId>
                </exclusion>
            </exclusions>
        </dependency>
        <dependency>
            <groupId>redis.clients</groupId>
            <artifactId>jedis</artifactId>
            <version>2.9.0</version>
        </dependency>

参考:https://www.cnblogs.com/williamjie/p/9395659.html

https://blog.csdn.net/Dome_/article/details/87621610

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值