基于Redis实现分布式锁

我们知道分布式锁的特性是排他、避免死锁、高可用。分布式锁的实现可以通过数据库的乐观锁(通过版本号)或者悲观锁(通过for update)、Redis的setnx()命令、Zookeeper(在某个持久节点添加临时有序节点，判断当前节点是否是序列中最小的节点，如果不是则监听比当前节点还要小的节点。如果是，获取锁成功。当被监听的节点释放了锁(也就是被删除)，会通知当前节点。然后当前节点再尝试获取锁，如此反复)

 

redis.png

本篇文章，主要讲如何用Redis的形式实现分布式锁。后续文章会讲解热点KEY读取，缓存穿透和缓存雪崩的场景和解决方案、缓存更新策略等等知识点，理论知识点较多。

Redis配置

spring:  redis:    port: 6379    database: 0    host: 127.0.0.1    password:    jedis:      pool:        max-active: 8        max-wait: -1ms        max-idle: 8        min-idle: 0    timeout: 5000ms　

Jedis工具类

public class JedisConnectionUtil {    private static JedisPool pool = null;    static {        JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();        jedisPoolConfig.setMaxTotal(100);        pool = new JedisPool(jedisPoolConfig, "127.0.0.1", 6379);    }    public static Jedis getJedis(){        return pool.getResource();    }}

　　

分布式锁分析与编码

下面进入正文。因为分布式系统之间是不同进程的，单机版的锁无法满足要求。所以我们可以借助中间件Redis的setnx()命令实现分布式锁。setnx()命令只会对不存在的key设值，返回1代表获取锁成功。对存在的key设值，会返回0代表获取锁失败。这里的value是System.currentTimeMillis() (获取锁的时间)+锁持有的时间。我这里设置锁持有的时间是200ms，实际业务执行的时间远比这200ms要多的多，持有锁的客户端应该检查锁是否过期，保证锁在释放之前不会过期。因为客户端故障的情况可能是很复杂的。比如现在有A，B俩个客户端。A客户端获取了锁，执行业务中做了骚操作导致阻塞了很久，时间应该远远超过200ms，当A客户端从阻塞状态下恢复继续执行业务代码时，A客户端持有的锁由于过期已经被其他客户端占有。这时候A客户端执行释放锁的操作，那么有可能释放掉其他客户端的锁。

我这里设置的客户端等待锁的时间是200ms。这里通过轮询的方式去让客户端获取锁。如果客户端在200ms之内没有锁的话，直接返回false。实际场景要设置合适的客户端等待锁的时间，避免消耗CPU资源。

获取锁

/** * 获取锁 * @param lockName 锁的名字 * @param acquireTimeout 或得所的超时时间 * @param lockTimeout 所本身的超时时间 * @return */public String acquireLock(String lockName, long acquireTimeout, long lockTimeout){    // 锁标识    String identifler = UUID.randomUUID().toString();    String lockKey = "lock" + lockName;    int lockExpire = (int) (lockTimeout / 1000);    Jedis jedis = null;    try {        jedis = JedisConnectionUtil.getJedis();        long end = System.currentTimeMillis() + acquireTimeout;        // 获取锁的限定时间        while (System.currentTimeMillis() < end) {            if (jedis.setnx(lockKey, identifler) == 1) {                // 设置成功                // 设置超时时间                jedis.expire(lockKey, lockExpire);                // 或得锁成功                return identifler;            }            if (jedis.ttl(lockKey) == -1) {                // 如果一直没有或得锁设置超时时间                jedis.expire(lockKey, lockExpire);            }            try {                Thread.sleep(100);            } catch (InterruptedException e) {                e.printStackTrace();            }        }    }finally {        jedis.close();    }    return null;}

　　

所以，我们要加上锁过期，然后获取锁的策略。通过realKey获取当前的currentValue。currentValue也就是获取锁的时间 + 锁持有的时间。 如果currentValue不等于null 且 currentValue 小于当前时间，说明锁已经过期。这时候如果突然来了C，D两个客户端获取锁的请求，不就让C，D两个客户端都获取锁了吗。如果防止这种现象发生，我们采用getSet()命令来解决。getSet(key,value)的命令会返回key对应的value，然后再把key原来的值更新为value。也就是说getSet()返回的是已过期的时间戳。如果这个已过期的时间戳等于currentValue，说明获取锁成功。

 

释放锁

/** * 释放锁 * @param lockName 锁的名字 * @param identifler 锁标识 * @return */public boolean releaseLock(String lockName, String identifler){    System.out.println(lockName + "释放锁" + identifler);    String lockKey = "lock:" + lockName;    Jedis jedis = null;    boolean isrelease = false;    try {        jedis = JedisConnectionUtil.getJedis();        while (true){            jedis.watch(lockKey);            // 判断是否是同一把锁            if (identifler.equals(jedis.get(lockKey))){                Transaction transaction = jedis.multi();                transaction.del(lockKey);                if (transaction.exec().isEmpty()){                    continue;                }                isrelease = true;            }            // TODO 异常            jedis.unwatch();            break;        }    }finally {        jedis.close();    }    return isrelease;}

编写测试类用多线程模拟高并发

public class UnitTest extends Thread {    @Override    public void run() {        while (true){            DistributedLock distributedLock = new DistributedLock();            String rs = distributedLock.acquireLock("updateOrder", 2000, 5000);            if (rs != null){                System.out.println(Thread.currentThread().getName() + "-> 或得锁" + rs);                try {                    Thread.sleep(1000);                    distributedLock.releaseLock("updateOrder", rs);                } catch (InterruptedException e) {                    e.printStackTrace();                }                break;            }        }    }    public static void main(String[] args) {        UnitTest unitTest = new UnitTest();        for (int i = 0; i < 10; i++) {            new Thread(unitTest, "tName:" + i).start();        }    }}

代码运行结果

tName:0-> 或得锁a86a2867-a5f9-4b81-8be6-105d50b8a1abupdateOrder释放锁a86a2867-a5f9-4b81-8be6-105d50b8a1abtName:1-> 或得锁bcdeb19c-a47c-4a12-86df-8e5d90b2ecb0updateOrder释放锁bcdeb19c-a47c-4a12-86df-8e5d90b2ecb0tName:6-> 或得锁147d9fc8-5e15-4e86-8da6-e029a3e2d92fupdateOrder释放锁147d9fc8-5e15-4e86-8da6-e029a3e2d92ftName:2-> 或得锁93deb41d-2439-45e7-beb1-2e4ce672e8caupdateOrder释放锁93deb41d-2439-45e7-beb1-2e4ce672e8catName:4-> 或得锁094f921d-fe9b-46ba-873b-aee2ce974f16updateOrder释放锁094f921d-fe9b-46ba-873b-aee2ce974f16tName:5-> 或得锁216e0799-6d22-4ae4-bb83-9efe1e5f80c9updateOrder释放锁216e0799-6d22-4ae4-bb83-9efe1e5f80c9tName:9-> 或得锁678e2099-651c-4e23-a648-9fb588ecb42bupdateOrder释放锁678e2099-651c-4e23-a648-9fb588ecb42btName:7-> 或得锁f35cdbad-6fde-4f1e-a4c1-321805a39374updateOrder释放锁f35cdbad-6fde-4f1e-a4c1-321805a39374tName:3-> 或得锁0913a4a0-805a-48e2-ac5a-c0762b8c4072updateOrder释放锁0913a4a0-805a-48e2-ac5a-c0762b8c4072tName:8-> 或得锁a3964a2a-6b9c-4ca3-9ea5-53de854c23ceupdateOrder释放锁a3964a2a-6b9c-4ca3-9ea5-53de854c23ce

 

 

 

本文文字部分参考文章:https://www.jianshu.com/p/83224c0f3bb9