【示例】sentinel ip和参数授权规则拦截器(黑名单白名单)

zhangdaiscott
1 parent 956b1c21
Showing 1 changed file with 13 additions and 4 deletions
jeecg-boot/jeecg-boot-starter/jeecg-boot-starter-cloud/src/main/java/org/jeecg/starter/cloud/interceptor/DefaultRequestOriginParser.java
 package org.jeecg.starter.cloud.interceptor;
  
 import com.alibaba.csp.sentinel.adapter.spring.webmvc.callback.RequestOriginParser;
+import org.apache.commons.lang3.StringUtils;
+import org.jeecg.common.util.IpUtils;
 import org.springframework.stereotype.Component;
  
 import javax.servlet.http.HttpServletRequest;
  
 /**
- * sentinel 授权规则拦截器
+ * 【示例】sentinel ip和参数授权规则拦截器(黑名单白名单)
+ * 1. 有参数origin的时候走参数拦截规则
+ * 2. 当参数为空时走ip拦截模式
+ *
  * @author zyf
  */
 @Component
@@ -15,8 +20,12 @@ public class DefaultRequestOriginParser implements RequestOriginParser {
     public String parseOrigin(HttpServletRequest request) {
         //基于请求参数,origin对应授权规则中的流控应用名称,也可通过getHeader传参
         String origin = request.getParameter("origin");
-
-        //TODO 此处做个通过IP做白名单的例子
-        return origin;
+        if (StringUtils.isNotEmpty(origin)) {
+            return origin;
+        } else {
+            //当参数为空使用ip拦截模式
+            String ip = IpUtils.getIpAddr(request);
+            return ip;
+        }
     }
 }