自定义注解实现@NotNull注解

先了解一下基本的自定义注解的时候，常用的注解含义

@Documented // 代表这个注解，可以被javadoc 处理的（加不加都行）
@Retention(RetentionPolicy.RUNTIME)
// RetentionPolicy 生命周期，一共有三个
// SOURCE 只留在源码，编译好后的代码是不有该注解的，该注解一旦编译就不存在了 ；
// class 只停留在字节码，运行时不会有；
// RUNTIME 运行时需要反射来获取该注解
@Target({ElementType.TYPE,ElementType.METHOD, ElementType.FIELD, ElementType.PARAMETER, ElementType.CONSTRUCTOR, ElementType.LOCAL_VARIABLE, ElementType.PACKAGE, ElementType.ANNOTATION_TYPE})
// 该注解打在哪个地方，方法，参数，类。。。。。。

 

---

1、编写 @NotNull 注解

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

/**
 * 自定义非空注释.
 *
 * @author ch
 * @version 1.0.0
 * @since 1.0.0
 *
 * Created at 2019-06-16 14:43
 */
@Target(ElementType.FIELD)
@Retention(RetentionPolicy.RUNTIME)
public @interface NotNull {

  /**
   * 获取参数名
   */
  String fileName();
}

 2、编写实体类

/**
 * 测试
 *
 * @author ch
 * @version 1.0.0
 * @since 1.0.0
 *
 * Created at 2019-06-17 14:02
 */
public class Person {
  
  @NotNull(fileName = "编号")
  private String id;
  @NotNull(fileName = "姓名")
  private String name;
  @NotNull(fileName = "年龄")
  private Integer age;

  public String getId() {
    return id;
  }

  public void setId(String id) {
    this.id = id;
  }

  public String getName() {
    return name;
  }

  public void setName(String name) {
    this.name = name;
  }

  public Integer getAge() {
    return age;
  }

  public void setAge(Integer age) {
    this.age = age;
  }

  @Override
  public String toString() {
    return "Person{" +
        "id='" + id + '\'' +
        ", name='" + name + '\'' +
        ", age=" + age +
        '}';
  }
}

 3、实现@NotNull

 /**
   * 检查非空.
   *
   * @param object 需要检查非空的对象.
   */
  public static void validObj(Object object)
      throws NoSuchMethodException, InvocationTargetException, IllegalAccessException {

    // 获取所有属性.
    Field[] declaredFields = object.getClass().getDeclaredFields();

    // 遍历所有属性.
    for (Field field : declaredFields) {

      // 进行必填项判断.
      if (field.isAnnotationPresent(NotNull.class)) {
        field.setAccessible(true);
        // get 方法拼串头
        String prefix = "get";
        String name = field.getName();
        Class<?> type = field.getType();
        // 判断布尔类型.
        if (type.isAssignableFrom(Boolean.class)) {
          prefix = "is";
        }

        Object fo = field.get(object);
        Object[] arr = null;
        // 如果是集合，循环调用 validObj
        if (fo instanceof Collection) {
          arr = ((Collection) fo).toArray();
        }

        // 如果是数组，循环调用 validObj
        if (field.getType().isArray()) {
          arr = (Object[]) fo;
        }
        // 递归 引用本方法 深度检测.
        if (null != arr) {
          for (Object o : arr) {
            validObj(o);
          }
        }

        Method method = object.getClass()
            .getMethod(prefix + name.substring(0, 1).toUpperCase() + name.substring(1));

        //调用getter方法获取属性值
        Object invoke = method.invoke(object);
        if (invoke == null) {
          NotNull notNull = field.getAnnotation(NotNull.class);
          throw new IllegalArgumentException(notNull.fileName() + " 不能为空");
        }

        // 如果是 list 则应该判断 list 长度.
        if (invoke != null && invoke.getClass().isAssignableFrom(ArrayList.class)
            && ((ArrayList<Object>) invoke).isEmpty()) {
          NotNull notNull = field.getAnnotation(NotNull.class);
          throw new IllegalArgumentException(notNull.fileName() + " 不能为空");
        }
      }

    }
  }