java如何根据注解参数类型数值来实现数据校验

原创

mob649e815f494b 2024-07-02 04:48:31 ©著作权

©著作权归作者所有：来自51CTO博客作者mob649e815f494b的原创作品，请联系作者获取转载授权，否则将追究法律责任

使用Java根据注解参数类型数值实现数据校验

在Java中，我们可以使用注解来对方法或者字段进行标记，然后通过解析注解的参数来实现数据校验。在本文中，我们将介绍如何根据注解参数类型数值来实现数据校验。

创建自定义注解

首先，我们需要创建一个自定义注解，用于标记需要进行数据校验的字段。我们可以定义注解的参数类型为枚举类型，来表示需要进行的校验规则。下面是一个简单的示例：

import java.lang.annotation.ElementType;
import java.lang.annotation.Retention;
import java.lang.annotation.RetentionPolicy;
import java.lang.annotation.Target;

@Retention(RetentionPolicy.RUNTIME)
@Target(ElementType.FIELD)
public @interface Validate {
    Rule rule() default Rule.NOT_NULL;
}

enum Rule {
    NOT_NULL,
    MIN,
    MAX
}

在上面的代码中，我们定义了一个Validate注解，并且定义了一个Rule枚举类型，包含了NOT_NULL, MIN, MAX三种校验规则。

创建数据校验工具类

接下来，我们需要创建一个数据校验的工具类，用于根据注解参数类型数值来判断是否符合校验规则。下面是一个简单的示例：

import java.lang.reflect.Field;

public class Validator {
    public static void validate(Object obj) throws IllegalAccessException {
        Class<?> clazz = obj.getClass();
        Field[] fields = clazz.getDeclaredFields();

        for(Field field : fields) {
            field.setAccessible(true);
            
            if(field.isAnnotationPresent(Validate.class)) {
                Validate validate = field.getAnnotation(Validate.class);
                Rule rule = validate.rule();

                switch(rule) {
                    case NOT_NULL:
                        if(field.get(obj) == null) {
                            throw new IllegalArgumentException(field.getName() + " cannot be null");
                        }
                        break;
                    case MIN:
                        int min = 10; // 举例最小值为10
                        if((int) field.get(obj) < min) {
                            throw new IllegalArgumentException(field.getName() + " should be greater than " + min);
                        }
                        break;
                    case MAX:
                        int max = 100; // 举例最大值为100
                        if((int) field.get(obj) > max) {
                            throw new IllegalArgumentException(field.getName() + " should be less than " + max);
                        }
                        break;
                }
            }
        }
    }
}

在上面的代码中，我们实现了一个validate方法，用于对传入的对象进行数据校验。我们首先获取对象的所有字段，然后判断字段是否被Validate注解标记，如果是，则根据注解参数类型数值进行相应的校验。

使用数据校验工具类

最后，我们来看一个使用数据校验工具类的例子：

public class User {
    @Validate(rule = Rule.NOT_NULL)
    private String name;

    @Validate(rule = Rule.MIN)
    private int age;

    public User(String name, int age) {
        this.name = name;
        this.age = age;
    }

    public static void main(String[] args) {
        User user = new User("Alice", 5);

        try {
            Validator.validate(user);
            System.out.println("Validation passed!");
        } catch (Exception e) {
            System.out.println("Validation failed: " + e.getMessage());
        }
    }
}

在上面的代码中，我们定义了一个User类，包含了name和age两个字段，分别使用了Validate注解，并指定了不为空和最小值校验规则。然后我们创建一个User对象，并调用Validator.validate方法进行数据校验。