redis实现分布式锁,lua脚本实现上锁原子操作
-
基础操作
-
使用Lua脚本的好处
1、 一次性发送多个命令,减少网络开销。(是多个reids命令的集合,不用每次都去建立连接)
2、原子性 (redis会将这个lua脚本认为是一个整体去执行,不会被打断,所以保证原子性)
3、lua 文件复用 (命令非常多,可以放在一个文件中,这样其他的redis也可以调用,使其复用) -
基本用法
127.0.0.1:6379> eval "return 'hello world'" 0
"hello world"
127.0.0.1:6379> keys *
(empty list or set)
127.0.0.1:6379> eval "return redis.call('set',KEYS[1],ARGV[1])" 1 HZ 2222
OK
127.0.0.1:6379> keys *
1) "HZ"
127.0.0.1:6379> get hz
(nil)
127.0.0.1:6379> get HZ
"2222"
127.0.0.1:6379>
- demo
import lombok.extern.slf4j.Slf4j;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.StringRedisTemplate;
import org.springframework.data.redis.core.script.DefaultRedisScript;
import org.springframework.stereotype.Component;
import java.util.Collections;
import java.util.concurrent.TimeUnit;
@Component
@Slf4j
public class RedisUtil {
/**
* 获取锁的value,判断是否等于期待的value,满足的话,则删除该锁,并返回1;否则直接返回-1。
*/
private static final DefaultRedisScript<Long> UNLOCK_LUA_SCRIPT = new DefaultRedisScript<>(
"if redis.call('get', KEYS[1]) == ARGV[1] then return redis.call('del', KEYS[1]) else return -1 end", Long.class
);
@Autowired
private StringRedisTemplate redisTemplate;
/**
* 上锁
* @param key 键
* @param value 值
* @param expireTime 失效时间
* @param timeUnit 时间单位
* @return
*/
public boolean tryLockTime(String key, String value, long expireTime, TimeUnit timeUnit) {
Boolean flag = redisTemplate.opsForValue().setIfAbsent(key, value, expireTime, timeUnit);
if (flag == null || !flag) {
log.error("申请锁(" + key + "," + value + ")失败");
return false;
}
return true;
}
/**
* 解锁
*
* @param key 键
* @param value 值
*/
public void unLockTime(String key, String value) {
Long result = redisTemplate.execute(UNLOCK_LUA_SCRIPT, Collections.singletonList(key), value);
if (result == -1) {
log.error("释放锁(" + key + "," + value + ")失败,该锁不存在或锁已经过期");
}
}
}
- 上锁
Boolean flag = redisUtil.tryLockTime("key", "value", 180, TimeUnit.SECONDS);
- 解锁
redisUtil.unLockTime("key", "value");