10.接口限流

来源:互联网 发布:公安网络监控电路板 编辑:程序博客网 时间:2024/06/06 13:14

0.参考链接

1.使用redis计数来控制单位时间内对某接口的访问量,防止刷验证码接口之类的http://m.blog.csdn.net/tianyaleixiaowu/article/details/745491452.征服Redis+Jedis+Springhttp://snowolf.iteye.com/blog/1824868/

1.业务需求

限制接口在单位时间内被访问的次数

2.技术点

1.Redis2.Spring整合redis(SpringData Jedis)3.AOP(环绕通知)4.自定义注解

3.Redis的安装配置

暂略

4.Spring整合redis

注:配置文件中各部分含义待补充0.涉及jar   commons-pool2-2.4.2.jar  jedis-2.9.0.jar  spring-data-commons-1.13.6.RELEASE.jar  spring-data-redis-1.6.2.RELEASE.jar  注意:各个jar包版本不匹配可能引起异常1.Spring xml配置        <context:property-placeholder location="classpath:redis.properties" />    <bean id="poolConfig" class="redis.clients.jedis.JedisPoolConfig"        p:maxIdle="${redis.maxIdle}" p:maxTotal="${redis.maxTotal}"        p:maxWaitMillis="${redis.maxWaitMillis}" p:testOnBorrow="${redis.testOnBorrow}" />    <bean id="jedisConnFactory"        class="org.springframework.data.redis.connection.jedis.JedisConnectionFactory"        p:hostName="${redis.host}" p:port="${redis.port}" p:password="${redis.pass}"        p:pool-config-ref="poolConfig" p:use-pool="false" />    <bean id="redisTemplate" class="org.springframework.data.redis.core.RedisTemplate"        p:connection-factory-ref="jedisConnFactory" />2.redis.properties    redis.host=127.0.0.1    redis.port=6379    redis.pass=123456    redis.maxIdle=300    redis.maxTotal=600    redis.maxWaitMillis=1000    redis.testOnBorrow=true

5.AOP(环绕通知)中结合自定义注解使用Redis

1.如何建立一个aop切面类?    详见 Java->4.配置->5.AOP2.创建自定义注解RequestLimit    import java.lang.annotation.Documented;    import java.lang.annotation.ElementType;    import java.lang.annotation.Retention;    import java.lang.annotation.RetentionPolicy;    import java.lang.annotation.Target;    import org.springframework.core.Ordered;    import org.springframework.core.annotation.Order;    @Retention(RetentionPolicy.RUNTIME)    @Target(ElementType.METHOD)    @Documented    @Order(Ordered.HIGHEST_PRECEDENCE)    public @interface RequestLimit {        /**         * 规定时间内的限制请求次数 (默认为5)         *          */        int count() default 5;        /**         * 单位时间(默认为60)         */        int time() default 60;    }3.在要限流的接口上加上注解@RequestLimit        //限制30秒内最多访问3次        @RequestLimit(count=3,time=30)    @RequestMapping(value = "/QueryInclusiveFinanceData",                     method = RequestMethod.POST, produces = "application/json;charset=UTF-8")    @ResponseBody    public String queryInclFina(@RequestParam("userName")  userName) throws Exception {            //处理业务        return result;    }4.环绕通知给接口限流        @Resource    private RedisTemplate<Serializable, Serializable> redisTemplate;    @Pointcut("execution(* com.phfinancial.controller.InclusiveFinanceApiController.queryInclFina(..)) ")    public void aspectPointcut() { }    @Around(value = "aspectPointcut() && @annotation(requestLimit)")    public Object aroundInclFinaController(ProceedingJoinPoint jp,RequestLimit requestLimit) throws Throwable {        Object[] args = jp.getArgs();        String userName = (String) args[0];        RedisSerializer<String> strSer = redisTemplate.getStringSerializer();        byte[] userNameSer = strSer.serialize(userName);        Long count = redisTemplate.opsForValue().increment(userNameSer, 1); //设置序列化的用户名为key,先自增1        if (count==1) { //如果为1,则设定过期时间(时间值来源于注解)            redisTemplate.expire(userNameSer, requestLimit.time(), TimeUnit.SECONDS);        }        if (count<=requestLimit.count()) { //如果没到阈值则可调用接口              //ProceedingJoinPoint.proceed()表示执行切点方法,返回值为切点方法的返回值              //一般使环绕通知的返回类型和切点方法返回类型保持一致              //ProceedingJoinPoint.proceed()之前类似前置通知,之后类似后置通知            return jp.proceed();         }                //达到阈值时候返回错误码        return ResultJson.errCode("5001", "限流控制");    }

6.达到效果

(设定如上)序号    访问时间    是否成功调取接口    错误码 1       00:00:00        是              无2       00:00:11        是              无3       00:00:18        是              无4       00:00:21        否              5001    5       00:00:27        否              50016       00:00:32        是              无
原创粉丝点击