一、SETNX
SETNX 是 Redis 中的一个命令,用于设置指定键的值,但只有在键不存在时才会设置成功,如果键已经存在,则不进行任何操作。SETNX 的作用是在键不存在时设置值,实现了一种简单的分布式锁机制。
SETNX key value
key:要设置的键。
value:要设置的值。
如果键 key 不存在,则将键 key 的值设置为 value,并返回 1(设置成功);如果键 key 已经存在,则不进行任何操作,并返回 0(设置失败)。
SETNX 命令通常用于实现分布式锁的获取操作。在多个客户端尝试获取同一资源的锁时,只有一个客户端能够成功获取锁,其他客户端获取锁的操作将会失败。这种方式可以确保在并发环境下,只有一个客户端能够执行临界区代码,从而实现了资源的互斥访问。
二、Lua
Lua(发音为"loo-ah")是一种轻量级的、可扩展的脚本语言。在Redis中,Lua脚本可以用来执行一系列的命令,通常用于实现复杂的操作、事务、原子操作等。Redis提供了 EVAL 和 EVALSHA 命令来执行Lua脚本。
以下是Lua脚本的基本语法和示例:
-- 定义Lua函数
local function myFunction(arg1, arg2)
return arg1 + arg2
end
-- 调用Lua函数
local result = myFunction(10, 20)
print(result) -- 输出 30
在Redis中,可以通过 EVAL 命令执行Lua脚本,如下所示:
EVAL "return 'Hello, World!'" 0
这个例子中,“return ‘Hello, World!’” 是Lua脚本,0 是表示脚本不需要参数。执行结果会返回字符串 ‘Hello, World!’。
三、实现
@GetMapping(value = "/lock")
public String reduceSku() throws Exception {
String value = UUID.randomUUID().toString();
// value值任意即可,设置自动过期时间为10S
boolean lock = redisTemplate.opsForValue().setIfAbsent("key", value, 10, TimeUnit.SECONDS);
if (lock) {
// 当前key没有锁,加锁成功 执行数据库查询
log.info("===>>>加锁成功");
// 模拟耗时8S
TimeUnit.SECONDS.sleep(8);
// 所以要保证验证value值和验证成功必须满足原子性,通过redis+lua实现
runLuaScript("key", value);
log.info("解锁成功");
} else {
// 当前key已经存在锁,加锁失败
log.info("===>>>加锁失败,等待重试...");
}
return "操作成功";
}
public boolean runLuaScript(String lockKey, String value) {
List<String> keyList = Collections.singletonList(lockKey);
// 执行脚本 删除锁
DefaultRedisScript<String> redisScript = new DefaultRedisScript<>();
redisScript.setScriptText("if redis.call('get',KEYS[1]) == ARGV[1] then\n" +
" redis.call('del', KEYS[1])\n" +
" return '1';\n" +
"else\n" +
" return '0'\n" +
"end\n");
redisScript.setResultType(String.class);
Integer execute = (Integer) redisTemplate.execute(redisScript, keyList, value);
return execute.equals(1);
}
这里我们为求方便直接在Controller这层定义了具体的业务方法。
JMeter安装使用
这里我使用了JMeter模拟五个线程在一秒内的并发测试
执行结果:
到此为止我们就简单地实现了分布式锁了!