限流

haoban-manage3-web/src/main/java/com/gic/haoban/manage/web/errCode/HaoBanErrCode.java

@@ -57,6 +57,11 @@ public enum HaoBanErrCode {
     ERR_11(11, "验证码发送失败"),
 
     /**
+     * 验证码发送失败
+     */
+    ERR_13(11, "客官，当前访问人数较多，请稍后再试"),
+
+    /**
      * 验证码校验失败
      */
     ERR_12(12, "验证码校验失败"),

haoban-manage3-web/src/main/java/com/gic/haoban/manage/web/interceptor/WebInterceptor.java

@@ -10,6 +10,7 @@ import com.gic.haoban.manage.web.auth.AuthRequestUtil;
 import com.gic.haoban.manage.web.errCode.HaoBanErrCode;
 import com.gic.haoban.manage.web.vo.LoginVO;
 
+import com.gic.redis.data.util.GicRateLimiter;
 import org.apache.commons.lang3.StringUtils;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
@@ -37,7 +38,7 @@ public class WebInterceptor extends HandlerInterceptorAdapter {
 
 //    @Autowired
 //    private ClerkRightManageService clerkRightManageService;
-
+    private static String MOUDLE="haoban-manage3-web";
 
     private void errorResult(HttpServletResponse httpServletResponse, HaoBanErrCode errCode) {
         HaobanResponse response = new HaobanResponse();
@@ -98,6 +99,17 @@ public class WebInterceptor extends HandlerInterceptorAdapter {
         if (loginUser.getStaffDTO() != null) {
             httpServletRequest.setAttribute("staffDTO", loginUser.getStaffDTO());
         }
+        String wxEnterpriseId = loginUser.getWxEnterpriseId();
+
+        GicRateLimiter instance = GicRateLimiter.getInstance(MOUDLE);
+
+        Long acquire = instance.acquire(requestURI, wxEnterpriseId);
+        logger.info("限流返回值：{}",acquire);
+        if (acquire.equals(0)) {
+            this.errorResult(httpServletResponse, HaoBanErrCode.ERR_13);
+            logger.info("限流!");
+            return false;
+        }
         return true;
     }