使用Redis单实例实现分布式锁

自古美人都是妖i 提交于 2020-03-08 06:33:59

为什么使用redis分布式锁
在同一个jvm进程中时,可以使用JUC提供的一些锁来解决多个线程竞争同一个共享资源时候的线程安全问题,但是当多个不同机器上的不同jvm进程共同竞争同一个共享资源时候,juc包的锁就无能无力了,这时候就需要分布式锁了。常见的有使用zk的最小版本,redis的set函数,数据库锁来实现。
使用分布式锁

  1. package com.jiaduo.DistributedLock;
  2.  
  3. import java.util.Collections;
  4.  
  5. import redis.clients.jedis.Jedis;
  6. import redis.clients.jedis.JedisPool;
  7.  
  8. public class DistributedLock {
  9.  
  10.     private static final String LOCK_SUCCESS = "OK";
  11.     private static final String SET_IF_NOT_EXIST = "NX";
  12.     private static final String SET_WITH_EXPIRE_TIME = "PX";
  13.     private static final Long RELEASE_SUCCESS = 1L;
  14.  
  15.     private static void validParam(JedisPool jedisPool, String lockKey, String requestId, int expireTime) {
  16.         if (null == jedisPool) {
  17.             throw new IllegalArgumentException("jedisPool obj is null");
  18.         }
  19.  
  20.         if (null == lockKey || "".equals(lockKey)) {
  21.             throw new IllegalArgumentException("lock key  is blank");
  22.         }
  23.  
  24.         if (null == requestId || "".equals(requestId)) {
  25.             throw new IllegalArgumentException("requestId is blank");
  26.         }
  27.  
  28.         if (expireTime < 0) {
  29.             throw new IllegalArgumentException("expireTime is not allowed less zero");
  30.         }
  31.     }
  32.  
  33.     /**
  34.      * 
  35.      * @param jedis
  36.      * @param lockKey
  37.      * @param requestId
  38.      * @param expireTime
  39.      * @return
  40.      */
  41.     public static boolean tryLock(JedisPool jedisPool, String lockKey, String requestId, int expireTime) {
  42.  
  43.         validParam(jedisPool, lockKey, requestId, expireTime);
  44.  
  45.         Jedis jedis = null;
  46.         try {
  47.  
  48.             jedis = jedisPool.getResource();
  49.             String result = jedis.set(lockKey, requestId, SET_IF_NOT_EXIST, SET_WITH_EXPIRE_TIME, expireTime);
  50.  
  51.             if (LOCK_SUCCESS.equals(result)) {
  52.                 return true;
  53.             }
  54.         } catch (Exception e) {
  55.             throw e;
  56.         } finally {
  57.             if (null != jedis) {
  58.                 jedis.close();
  59.             }
  60.         }
  61.  
  62.         return false;
  63.     }
  64.  
  65.     /**
  66.      * 
  67.      * @param jedis
  68.      * @param lockKey
  69.      * @param requestId
  70.      * @param expireTime
  71.      */
  72.     public static void lock(JedisPool jedisPool, String lockKey, String requestId, int expireTime) {
  73.  
  74.         validParam(jedisPool, lockKey, requestId, expireTime);
  75.  
  76.         while (true) {
  77.             if (tryLock(jedisPool, lockKey, requestId, expireTime)) {
  78.                 return;
  79.             }
  80.         }
  81.     }
  82.  
  83.     /**
  84.      * 
  85.      * @param jedis
  86.      * @param lockKey
  87.      * @param requestId
  88.      * @return
  89.      */
  90.     public static boolean unLock(JedisPool jedisPool, String lockKey, String requestId) {
  91.  
  92.         validParam(jedisPool, lockKey, requestId, 1);
  93.  
  94.         String script = "if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end";
  95.  
  96.         Jedis jedis = null;
  97.         try {
  98.  
  99.             jedis = jedisPool.getResource();
  100.             Object result = jedis.eval(script, Collections.singletonList(lockKey),
  101.                     Collections.singletonList(requestId));
  102.  
  103.             if (RELEASE_SUCCESS.equals(result)) {
  104.                 return true;
  105.             }
  106.  
  107.         } catch (Exception e) {
  108.             throw e;
  109.         } finally {
  110.             if (null != jedis) {
  111.                 jedis.close();
  112.             }
  113.         }
  114.  
  115.         return false;
  116.  
  117.     }
  118.  
  119. }

复制代码

首先Redis的 public String set(final String key, final String value, final String nxxx, final String expx,
final int time)方法参数说明:

其中前面两个是key,value值;
nxxx为模式,这里我们设置为NX,意思是说如果key不存在则插入该key对应的value并返回OK,否者什么都不做返回null;
参数expx这里我们设置为PX,意思是设置key的过期时间为time 毫秒
通过tryLock方法尝试获取锁,内部是具体调用Redis的set方法,多个线程同时调用tryLock时候会同时调用set方法,但是set方法本身是保证原子性的,对应同一个key来说,多个线程调用set方法时候只有一个线程返回OK,其它线程因为key已经存在会返回null,所以返回OK的线程就相当与获取到了锁,其它返回null的线程则相当于获取锁失败。

另外这里我们要保证value(requestId)值唯一是为了保证只有获取到锁的线程才能释放锁,这个下面释放锁时候会讲解。

通过lock 方法让使用tryLock获取锁失败的线程本地自旋转重试获取锁,这类似JUC里面的CAS。

Redis有一个叫做eval的函数,支持Lua脚本执行,并且能够保证脚本执行的原子性,也就是在执行脚本期间,其它执行redis命令的线程都会被阻塞。这里解锁时候使用下面脚本:

  1. if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return 0 end

复制代码

其中keys[1]为unLock方法传递的key,argv[1]为unLock方法传递的requestId;脚本redis.call(‘get’, KEYS[1])的作用是获取key对应的value值,这里会返回通过Lock方法传递的requetId, 然后看当前传递的RequestId是否等于key对应的值,等于则说明当前要释放锁的线程就是获取锁的线程,则继续执行redis.call(‘del’, KEYS[1])脚本,删除key对应的值。

总结
使用redis单实例结合redis的set方法和eval函数实现了一个简单的分布式锁,但是这个实现还是明显有问题的。虽然使用set方法设置了超时时间,以避免线程获取到锁后redis挂了后锁没有被释放的情况,但是超时时间设置为多少合适那?如果设置太小,可能会存在线程获取锁后执行业务逻辑时间大于锁超时时间,那么就会存在逻辑还没执行完,锁已经因为超时自动释放了,而其他线程可能获取到锁,那么之前获取锁的线程的业务逻辑的执行就没有保证原子性。

另外还有一个问题是Lock方法里面是自旋调用tryLock进行重试,这就会导致像JUC中的AtomicLong一样,在高并发下多个线程竞争同一个资源时候造成大量线程占用cpu进行重试操作。这时候其实可以随机生成一个等待时间,等时间到后在进行重试,以减少潜在的同时对一个资源进行竞争的并发量。
转载于http://ifeve.com/redis-distributedlock/ 

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!