A股上市公司传智教育(股票代码 003032)旗下技术交流社区北京昌平校区

    当多个进程不在同一个系统中,可以使用分布式锁控制多个进程对同一资源的访问。因为在分布式情况下(多JVM),线
A和线程B很可能不是在同一JVM中,这样线程锁就无法起到作用了,这时候就要用到分布式锁来解决。为了确保分布式锁可
用,我们至少要确保锁的实现同时满足以下四个条件:
    1、互斥性。在任意时刻,只有一个客户端能持有锁。
    2、防止死锁。即使有一个客户端在持有锁的期间崩溃而没有主动解锁,也能保证后续其他客户端能加锁。
    3、容错性。只要大部分的Redis节点正常运行,客户端就可以加锁和解锁。
    4、加锁和解锁必须是同一个客户端。

工具类:
package com.it;

import redis.clients.jedis.Jedis;
import java.util.Collections;

/*使用前需导入如下依赖:
<dependency>
    <groupId>redis.clients</groupId>
    <artifactId>jedis</artifactId>
</dependency>*/
public class RedisTool {

    private static final String LOCK_SUCCESS = "OK";
    // NX,意思是SET IF NOT EXIST,即当key不存在时,我们进行set操作;若key已经存在,则不做任何操作。
    private static final String SET_IF_NOT_EXIST = "NX";
    // PX,意思是要给key加一个过期的设置,具体时间由下一个参数决定。
    private static final String SET_WITH_EXPIRE_TIME = "PX";
    private static final Long RELEASE_SUCCESS = 1L;

    private RedisTool() {}

    /**
     * 尝试获取分布式锁
     * @param jedis Redis客户端
     * @param lockKey 锁:我们使用lockKey来当锁,因为key是唯一的。
     * @param requestId 请求标识:分布式锁要满足解铃还须系铃人,在解锁的时候就可以以此为依据。
     *                      requestId可以使用UUID.randomUUID().toString()方法生成。
     * @param expireTime 超期时间:与“SET_WITH_EXPIRE_TIME”相呼应,代表key的过期时间。
     * @return 是否获取成功
     */
    public static boolean tryGetDistributedLock(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) {
        /*Lua脚本说明:首先获取KEYS[1]对应的value,如果与ARGV[1]相等,则解锁。*/
        String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
        /*eval()方法是将Lua脚本交给Redis服务端执行,参数KEYS[1]赋值为lockKey,ARGV[1]赋值为requestId。*/
        Object result = jedis.eval(script, Collections.singletonList(lockKey), Collections.singletonList(requestId));
        if (RELEASE_SUCCESS.equals(result)) {
            return true;
        }
        return false;
    }
}


0 个回复

您需要登录后才可以回帖 登录 | 加入黑马