业务场景
文章添加评论功能,根据用户IP地址,限制同一IP用户三十秒内只能评论一次
代码
@PostMapping
@ApiOperation("保存")
public Result<Void> save(@RequestBody CommentEntity entity){
String key = getKey();
if (Boolean.TRUE.equals(redisTemplate.hasKey(key))) {
throw new GlobalException(10000, "三十秒内只能评论一次");
}
redisTemplate.opsForValue().set(key, Constants.REDIS_DEFAULT_VALUE, Cons
commentService.saveComment(entity);
return new Result<>();
这段代码首先获取请求端IP,首先以用户IP构建key,从Redis中查询是否存在已评论标志位,如果存在,代码当前用户三十秒内已经评论过。
如果不存在,则添加标志位,并设置过期时间为30秒。
功能正常。但是这种方式存在并发问题:
业务代码中,判断Redis中是否存在标志位,以及设置标志位,是两步操作,若此时同一IP有大量请求同时进入,前面的请求没来得及设置标志位,导致之后的请求在判断标志位时,都为false(即标志位不存在),造成重复添加。
使用Jmeter进行10000线程并发场景模拟:
可以看到有些请求被成功拦截;
但是可以看到评论区依然存在多个同一时刻、同一用户的评论,即不止一个请求执行插入方法,当前写法存在并发问题。
使用setIfAbsent改进
上述方法的问题在于检查Redis值是否存在和设值的操作不是原子性,从而导致并发问题。
而Redis的set nx ex命令是一个原子操作,只有当键不存在时,它才设置键值,并设定过期时间,有效避免的并发冲突。
因此,代码改进为:
@PostMapping
@ApiOperation("保存")
public Result<Void> save(@RequestBody CommentEntity entity){
String key = getKey();
Boolean isCommentAllowed = redisTemplate.opsForValue()
.setIfAbsent(key, Constants.REDIS_DEFAULT_VALUE, Constants.Redis.COMMENT_TIME_INTERVAL_SECONDS.getValue(), TimeUnit.SECONDS);
if (Boolean.FALSE.equals(isCommentAllowed)) {
throw new GlobalException(10000, "三十秒内只能评论一次");
}
commentService.saveComment(entity);
return new Result<>();
}
setIfAbsent对应Redis的set nx命令。
使用Jmeter进行并发测试:
10000并发下,最终只有一条评论成功入库。