10.接口限流

0.参考链接

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

1.业务需求

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

2.技术点

1.Redis
2.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.AOP
2.创建自定义注解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        否              5001
6       00:00:32        是              无