简介
Redis分布式锁算法有两种，一种是单个Redis实例下的，一种是多个Redis实例的Redlock算法。
官方推荐Redlock算法，但是这个算法需要比较多的Redis实例而且是完全互相独立，不存在主从复制或者其他集群协调机制的，所以不太适合小项目。
单Redis实例
原理
某个线程调用Redis命令 SET key value NX PX 30000。
这个命令的意思是，仅在不存在key的时候才能被执行成功（NX选项），并且这个key有一个30秒的自动失效时间（NX选项）。key的失效时间是一个调用线程独占锁的时间。这个key的value最好是一个随机数，value在所有的调用线程中必须是唯一的。value是随机数主要是为了更安全的释放锁，释放锁的时候使用脚本告诉Redis，只有key存在并且存储的值和我指定的值一样才能告诉我删除成功。可以通过以下Lua脚本实现：
if redis.call("get",KEYS[1]) == ARGV[1] then return redis.call("del",KEYS[1]) else return 0 end。
使用这种方式释放锁可以避免删除别的调用线程获取成功的锁。举个例子：线程A取得资源锁，当线程A运行完毕其他操作后要释放锁时，原来的锁早已超时并且被Redis自动释放，并且在这期间资源锁又被线程B再次获取到。如果仅使用DEL命令将key删除，那么这种情况就会把线程B的锁给删除掉。
但是单实例有个缺点，如果Redis服务器宕机，那么就会导致无法获取锁，后果就是无法执行后续方法。假如采用主从复制，因为Redis主从是异步的，就是master不会等待slave接收了数据再响应客户端。考虑这种情景，线程A请求锁，master写入成功后发送给slave前master宕机，触发故障转移，slave升级为master，正好这时线程B又来获取锁并且成功，那么线程A和B都可以执行任务，所以用这种方式的话，宁愿不要有slave。不过这种属于小概率错误，在一个保证永不宕机的环境下这个方式没有任何问题。
示例
以下Demo类中的调用线程可以假设是分布在多个JVM进程中的线程，为了方便测试，共享数据也是设置到Redis中，客户端是Jedis。
import redis.clients.jedis.Jedis;
/**
* 单实例Redis分布式锁工具类
*/
public class RedisLockUtils {
/**
* 加锁
*
* @param jedis
* @param key
* @param value
* @param seconds
* @return
*/
public static Boolean lock(Jedis jedis, String key, String value, int seconds) {
String result = jedis.set(key, value, "NX", "EX", seconds);
return (result != null) && ("OK".equals(result));
}
/**
* 释放锁
*
* @param jedis
* @param key
* @param value
* @return
*/
public static Boolean unlock(Jedis jedis, String key, String value) {
if (value.equals(jedis.get(key))) {
return jedis.del(key) == 1;
}
return false;
}
/**
* 通过执行Lua脚本释放锁
*
* @param jedis
* @param key
* @param value
* @return
*/
public static Boolean unlockByLua(Jedis jedis, 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";
long result = (long) jedis.eval(script, 1, key, value);
return result == 1;
}
}
import com.ice.util.RedisLockUtils;
import redis.clients.jedis.Jedis;
import redis.clients.jedis.JedisPool;
import redis.clients.jedis.JedisPoolConfig;
import java.util.UUID;
import java.util.concurrent.CountDownLatch;
public class Demo {
public static void main(String[] args) throws Exception {
JedisPoolConfig jedisPoolConfig = new JedisPoolConfig();
jedisPoolConfig.setMinIdle(10);
jedisPoolConfig.setMaxIdle(50);
jedisPoolConfig.setMaxTotal(150);
JedisPool jedisPool = new JedisPool(jedisPoolConfig, "192.168.164.128", 6379);
CountDownLatch countDownLatch = new CountDownLatch(100);
for (int i = 0; i < 100; i++) {
new Thread(() -> {
String uuid = UUID.randomUUID().toString();
String key = "lock";
Jedis jedis = jedisPool.getResource();
try {
boolean result = false;
do {
result = RedisLockUtils.lock(jedis, key, uuid, 30);
} while (result == false);
int value = Integer.parseInt(jedis.get("value"));
jedis.set("value", ++value + "");
} finally {
// RedisLockUtils.unlock(jedis, key, uuid);
RedisLockUtils.unlockByLua(jedis, key, uuid);
jedis.close();
countDownLatch.countDown();
}
}).start();
}
countDownLatch.await();
try (Jedis jedis = new Jedis("192.168.164.128", 6379)) {
int value = Integer.parseInt(jedis.get("value"));
System.out.println("value=" + value);
}
}
}
Redlock算法
参考官网。