| | |
| | | package com.ruoyi.framework.config; |
| | | |
| | | import cn.hutool.core.util.ObjectUtil; |
| | | import cn.hutool.core.util.StrUtil; |
| | | import com.ruoyi.common.utils.StringUtils; |
| | | import com.ruoyi.framework.config.properties.RedissonProperties; |
| | | import lombok.extern.slf4j.Slf4j; |
| | |
| | | import org.springframework.cache.annotation.EnableCaching; |
| | | import org.springframework.context.annotation.Bean; |
| | | import org.springframework.context.annotation.Configuration; |
| | | import org.springframework.data.redis.core.script.DefaultRedisScript; |
| | | |
| | | import java.io.IOException; |
| | | import java.util.HashMap; |
| | |
| | | config.put(group.getGroupId(), cacheConfig); |
| | | } |
| | | return new RedissonSpringCacheManager(redissonClient, config, JsonJacksonCodec.INSTANCE); |
| | | } |
| | | |
| | | @Bean |
| | | public DefaultRedisScript<Long> limitScript() { |
| | | DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>(); |
| | | redisScript.setScriptText(limitScriptText()); |
| | | redisScript.setResultType(Long.class); |
| | | return redisScript; |
| | | } |
| | | |
| | | /** |
| | | * 限流脚本 |
| | | */ |
| | | private String limitScriptText() { |
| | | return StrUtil.builder() |
| | | .append("local key = KEYS[1]\n") |
| | | .append("local count = tonumber(ARGV[1])\n") |
| | | .append("local time = tonumber(ARGV[2])\n") |
| | | .append("local current = redis.call('get', key);\n") |
| | | .append("if current and tonumber(current) > count then\n") |
| | | .append(" return current;\n") |
| | | .append("end\n") |
| | | .append("current = redis.call('incr', key)\n") |
| | | .append("if tonumber(current) == 1 then\n") |
| | | .append(" redis.call('expire', key, time)\n") |
| | | .append("end\n") |
| | | .append("return current;") |
| | | .toString(); |
| | | } |
| | | |
| | | } |