分布式锁。

原创已于 2023-03-16 15:54:33 修改 · 72 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#java

于 2021-03-30 18:42:12 首次发布

分布式锁一般有三种实现方式

数据库乐观锁。
基于Redis的分布式锁。
基于ZooKeeper的分布式锁。

一、Redis实现分布式锁

通过setnx，不存在key则插入成功，存在key则不插入
且加上过期时间，防止释放锁失败一直死锁
插入和设置过期时间是原子性操作
释放锁，只有当前持有改锁的客户端能够释放锁。通过Lua脚本执行，如果redis value相同才删除key。如果不是原子性操作，A客户端的value=redisValue，刚好过期，B客户端过来访问获取到锁，此时A客户端执行删除key释放锁。可能误释放B的锁

	/**
     * 获取分布式锁, 且锁加上过期时间
     * 单个redis服务器, 已有key存在, 则函数不会调用成功。只有一个客户端能持有key, 通过setNx + 过期时间实现锁
     * @param key
     * @param value
     * @param expireTime 秒
     * @return
     */
    public static boolean getDistributedLock(RedisService redisService, String key, String value, int expireTime) {
        boolean res = redisService.setNxExpire(key, value, expireTime);
        logger.info(String.format("key：%s, 获取锁结果：%s", key, res));

        return res;
    }

	public boolean setNxExpire(String key, String value, int seconds) {
        Jedis jedis = null;
        try {
            jedis = getJedis(jedisPool);
            String result = jedis.set(key, value, "NX", "EX", seconds);
            if ("OK".equals(result)) {
                return true;
            } else {
                return false;
            }
        } catch (Exception e) {
            returnBrokenResource(jedis);
        } finally {
            returnResource(jedis);
        }
        return false;
    }

	/**
     * 释放分布式锁, 只有value相同时才能释放
     * @param key
     * @param value
     * @return
     */
    public static boolean releaseDistributedLock(RedisService redisService, String key, String value) {
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        Object result = redisService.eval(script, key, value);
        if (null == result) {
            return false;
        }
        if (Long.valueOf(1).equals(result)) {
            return true;
        } else {
            return false;
        }
    }

	@Override
    public Object eval(String script, String key, String value) {
        Jedis jedis = null;
        try {
            jedis = getJedis(jedisPool);
            return jedis.eval(script, Collections.singletonList(key), Collections.singletonList(value));
        } catch (Exception e) {
            returnBrokenResource(jedis);
        } finally {
            returnResource(jedis);
        }
        return null;
    }