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        是              無