java根据ip限流_自定义ip限流注解

keys = ImmutableList.of(StringUtils.join(limit.prefix(), "_", key, "_", request.getRequestURI().replaceAll("/","_")));

String luaScript = buildLuaScript();

RedisScriptredisScript = new DefaultRedisScript<>(luaScript, Number.class);

Number count = redisTemplate.execute(redisScript, keys, limit.count(), limit.period());

if (null != count && count.intValue() <= limit.count()) {

logger.info("第{}次访问key为 {}，描述为 [{}] 的接口", count, keys, limit.name());

return joinPoint.proceed();

} else {

throw new BadRequestException("访问次数受限制");

}

}

/**

* 限流脚本

*/

private String buildLuaScript() {

return "local c" +

"\nc = redis.call('get',KEYS[1])" +

"\nif c and tonumber(c) > tonumber(ARGV[1]) then" +

"\nreturn c;" +

"\nend" +

"\nc = redis.call('incr',KEYS[1])" +

"\nif tonumber(c) == 1 then" +

"\nredis.call('expire',KEYS[1],ARGV[2])" +

"\nend" +

"\nreturn c;";

}

}