Java 编写Validator验证器

1.编写annotaion 注解，定义提示信息，是否需要校验，正则规则等

import java.lang.annotation.*;

/**
 * 自定义校验注解
 * @author Andy
 */
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
@Documented
public @interface ValidateField {

    String message() default "Field validation failed";

    boolean required() default true;

    int maxLength() default -1;

    int minLength() default -1;

    String regex() default "";
}

2. 编写校验执行器，基于annotaion校验对象属性；

import cn.hutool.core.exceptions.ExceptionUtil;
import cn.hutool.core.util.StrUtil;
import lombok.experimental.UtilityClass;
import org.springframework.util.ObjectUtils;

import java.lang.reflect.Field;
import java.util.Arrays;
import java.util.Optional;

/**
 * 校验执行器
 *
 * @author Andy
 * @date 2023/3/7
 */
@UtilityClass
public class Validator {

    public void validate(Object obj) {
        Field[] fields = obj.getClass().getDeclaredFields();
        Optional<Field[]> optional = Optional.ofNullable(fields);
        searchValidate(obj, fields, optional);
    }

    private void searchValidate(Object obj, Field[] fields, Optional<Field[]> optional) {
        optional.ifPresent(fd -> Arrays.stream(fields).forEach(field -> {
            loop(obj, field);
        }));
    }

    private void loop(Object obj, Field field) {
        if (field.isAnnotationPresent(ValidateField.class)) {
            ValidateField validateField = getValidateField(field);
            Object value = null;
            try {
                value = field.get(obj);
            } catch (IllegalAccessException e) {
                throw new RuntimeException(e.getMessage());
            }
            validateRequired(validateField, value);
        }
    }

    private ValidateField getValidateField(Field field) {
        ValidateField validateField = field.getAnnotation(ValidateField.class);
        field.setAccessible(true);
        return validateField;
    }

    private void validateRequired(ValidateField validateField, Object value) {

        if (ObjectUtils.isEmpty(value)) {
            throw ExceptionUtil.wrapRuntime(validateField.message());
        }

        if (validateField.required() && StrUtil.isBlank(value.toString())) {
            throw ExceptionUtil.wrapRuntime(validateField.message());
        }
        if (validateField.maxLength() > 0 && value.toString().length() > validateField.maxLength()) {
            throw ExceptionUtil.wrapRuntime(validateField.message());
        }
        if (validateField.minLength() > 0 && value.toString().length() < validateField.minLength()) {
            throw ExceptionUtil.wrapRuntime(validateField.message());
        }
        if (StrUtil.isNotBlank(validateField.regex()) && !value.toString().matches(validateField.regex())) {
            throw ExceptionUtil.wrapRuntime(validateField.message());
        }

    }

}

3. 编写校验类

import io.swagger.v3.oas.annotations.media.Schema;
import lombok.Data;

import java.io.Serializable;

/**
 * 登录参数
 *
 * @author Andy
 * @date 2023/2/22
 */
@Data
@Schema(description = "登录")
public class LoginDTO implements Serializable {

    /**
     * 用户名
     */
    @ValidateField(message = "用户名为空")
    @Schema(description ="用户名")
    private String username;

    /**
     * 密码
     */
    @ValidateField(message = "用户密码为空")
    @Schema(description ="密码")
    private String password;
}

4. 测试

/**
 * 类说明
 *
 * @author Andy
 * @date 2023/3/7
 */
public class Demo {

    public static void main(String[] args) {

        LoginDTO loginDTO = new LoginDTO();
        CbValidator.validate(loginDTO);
    }
}