支持自定义注解实现接口限流

RuoYi

2021-08-17 b8f7b48b79e496c3d4e6a0661a2f0f6e436587fd

 src/main/java/com/ruoyi/framework/config/RedisConfig.java

@@ -6,6 +6,7 @@
import org.springframework.context.annotation.Configuration;

import org.springframework.data.redis.connection.RedisConnectionFactory;

import org.springframework.data.redis.core.RedisTemplate;

import org.springframework.data.redis.core.script.DefaultRedisScript;

import org.springframework.data.redis.serializer.StringRedisSerializer;

import com.fasterxml.jackson.annotation.JsonAutoDetect;

import com.fasterxml.jackson.annotation.JsonTypeInfo;

@@ -47,4 +48,32 @@
        template.afterPropertiesSet();

        return template;

    }



    @Bean

    public DefaultRedisScript<Long> limitScript()

    {

        DefaultRedisScript<Long> redisScript = new DefaultRedisScript<>();

        redisScript.setScriptText(limitScriptText());

        redisScript.setResultType(Long.class);

        return redisScript;

    }



    /**

     * 限流脚本

     */

    private String limitScriptText()

    {

        return "local key = KEYS[1]\n" +

                "local count = tonumber(ARGV[1])\n" +

                "local time = tonumber(ARGV[2])\n" +

                "local current = redis.call('get', key);\n" +

                "if current and tonumber(current) > count then\n" +

                "    return current;\n" +

                "end\n" +

                "current = redis.call('incr', key)\n" +

                "if tonumber(current) == 1 then\n" +

                "    redis.call('expire', key, time)\n" +

                "end\n" +

                "return current;";

    }

}