@initial 7.0.0-preview.422.1 Add validator for request params

Author: TalexDreamSoul

src/main/java/com/talexframe/frame/core/modules/network/connection/RequestAnalyser.java

@@ -16,6 +16,7 @@
 import com.talexframe.frame.core.pojo.wrapper.WrappedResponse;
 import com.talexframe.frame.utils.ReqMethodUtil;
 import com.talexframe.frame.utils.UrlUtil;
+import com.talexframe.frame.utils.ValidatorUtil;
 import lombok.Getter;
 import lombok.Setter;
 import lombok.SneakyThrows;
@@ -148,6 +149,8 @@ private void process() {
                         TParam param = paramReceiver.tParam;
                         Parameter parameter = paramReceiver.getParameter();
 
+                        TPParamValidator validator = parameter.getAnnotation(TPParamValidator.class);
+
                         try {
 
                             String fieldName = param.field() != null ? param.field() : parameter.getName();
@@ -164,6 +167,39 @@ private void process() {
 
                                 Object obj = json.get(fieldName, parameter.getType());
 
+                                if ( validator != null ) {
+
+                                    int back = ValidatorUtil.validateData(validator, obj);
+
+                                    if( back != 100 ) {
+
+                                        wr.returnDataByFailed(ResultData.ResultEnum.INFORMATION_ERROR, "Data error");
+
+                                        switch ( back ) {
+
+                                            case ValidatorUtil.MISS_MAX:
+                                                log.info("[解析层] AccessDenied # MissingMax " + fieldName);
+                                            case ValidatorUtil.MISS_MIN:
+                                                log.info("[解析层] AccessDenied # MissingMin " + fieldName);
+                                            case ValidatorUtil.MISS_PATTERN:
+                                                log.info("[解析层] AccessDenied # MissingPattern " + fieldName);
+                                            case ValidatorUtil.MISS_MAX_LENGTH:
+                                                log.info("[解析层] AccessDenied # MissingMaxLength " + fieldName);
+                                            case ValidatorUtil.MISS_MIN_LENGTH:
+                                                log.info("[解析层] AccessDenied # MissingMinLength " + fieldName);
+                                            case ValidatorUtil.MISS_ASSERT:
+                                                log.info("[解析层] AccessDenied # MissingAssert " + fieldName);
+                                            case ValidatorUtil.MISS_DATA:
+                                                log.info("[解析层] AccessDenied # MissingData " + fieldName);
+
+                                        }
+
+                                        return;
+
+                                    }
+
+                                }
+
                                 params.add(obj);
 
                             }
@@ -178,7 +214,6 @@ private void process() {
 
                             }
 
-
                         }
 
                     }

src/main/java/com/talexframe/frame/core/pojo/annotations/TPParamValidator.java

@@ -0,0 +1,60 @@
+package com.talexframe.frame.core.pojo.annotations;
+
+import java.lang.annotation.ElementType;
+import java.lang.annotation.Retention;
+import java.lang.annotation.RetentionPolicy;
+import java.lang.annotation.Target;
+
+/**
+ * 加在字段上进行请求校验
+ * {@link com.talexframe.frame.core.pojo.annotations Package }
+ *
+ * @author TalexDreamSoul 22/04/17 上午 08:34 Project: TalexFrame
+ */
+@Retention( RetentionPolicy.RUNTIME )
+@Target( { ElementType.PARAMETER, ElementType.TYPE } )
+public @interface TPParamValidator {
+
+    /**
+     * 校验规则 # 必须满足当前正则表达式
+     */
+    String pattern() default "";
+
+    /**
+     * 校验规则 # 必须满足不为 null
+     */
+    boolean notNull() default true;
+
+    /**
+     * 自动类型断言 # true
+     * 必须元素为true可用
+     */
+    boolean assertTrue() default false;
+
+    /**
+     * 自动类型断言 # false
+     * 必须元素为false可用
+     */
+    boolean assertFalse() default false;
+
+    /**
+     * 校验规则 # 必须满足最小值
+     */
+    int min() default Integer.MIN_VALUE;
+
+    /**
+     * 校验规则 # 必须满足最大值
+     */
+    int max() default Integer.MAX_VALUE;
+
+    /**
+     * 校验规则 # 必须满足最小长度
+     */
+    int minLength() default Integer.MIN_VALUE;
+
+    /**
+     * 校验规则 # 必须满足最大长度
+     */
+    int maxLength() default Integer.MAX_VALUE;
+
+}

src/main/java/com/talexframe/frame/utils/ValidatorUtil.java

@@ -0,0 +1,90 @@
+package com.talexframe.frame.utils;
+
+import cn.hutool.core.util.StrUtil;
+import com.talexframe.frame.core.pojo.annotations.TPParamValidator;
+
+/**
+ * {@link com.talexframe.frame.utils Package }
+ *
+ * @author TalexDreamSoul 22/04/22 下午 09:20 Project: TalexFrame
+ */
+public class ValidatorUtil {
+
+    public static final int SUCCESS = 100;
+    public static final int MISS_DATA = 0;
+    public static final int MISS_PATTERN = -100;
+    public static final int MISS_ASSERT = -10;
+    public static final int MISS_MIN_LENGTH = -21;
+    public static final int MISS_MAX_LENGTH = -22;
+
+    public static final int MISS_MIN = -31;
+    public static final int MISS_MAX = -32;
+
+    public static int validateData(TPParamValidator validator, Object data) {
+
+        if( validator == null ) return SUCCESS;
+
+        if( data == null ) return validator.notNull() ? MISS_DATA : SUCCESS;
+
+        if( !StrUtil.isBlankIfStr(validator.pattern()) ) {
+
+            if( !data.toString().matches(validator.pattern()) ) return MISS_PATTERN;
+
+        }
+
+        if( validator.assertFalse() || validator.assertTrue() ) {
+
+            if( data instanceof Boolean ) {
+
+                boolean b = (boolean) data;
+
+                return ((b && validator.assertTrue()) || (!b && validator.assertFalse())) ? SUCCESS : MISS_ASSERT;
+
+            } else {
+
+                return MISS_ASSERT;
+
+            }
+
+        }
+
+        String str = (String) data;
+
+        if( validator.maxLength() > 0 && str.length() < validator.minLength() ) return MISS_MIN_LENGTH;
+        if( validator.maxLength() > 0 && str.length() > validator.maxLength() ) return MISS_MAX_LENGTH;
+
+        if( validator.min() != Integer.MIN_VALUE ) {
+
+            try {
+
+                int tmp = Integer.parseInt(str);
+                return tmp < validator.min() ? MISS_MIN : SUCCESS;
+
+            } catch (NumberFormatException e) {
+
+                return MISS_MIN;
+
+            }
+
+        }
+
+        if( validator.max() != Integer.MAX_VALUE ) {
+
+            try {
+
+                int tmp = Integer.parseInt(str);
+                return tmp > validator.max() ? MISS_MAX : SUCCESS;
+
+            } catch (NumberFormatException e) {
+
+                return MISS_MAX;
+
+            }
+
+        }
+
+        return SUCCESS;
+
+    }
+
+}