Java自定义注解源码+原理解释(使用Java自定义注解校验bean传入参数合法性)

Java自定义注解源码+原理解释(使用Java自定义注解校验bean传入参数合法性)

前言:由于前段时间忙于写接口,在接口中需要做很多的参数校验,本着简洁、高效的原则,便写了这个小工具供自己使用(内容为缩减版,具体业务可自行扩展)

思路:使用Java反射机制,读取实体类属性头部注解,通过get方法获取参数值进行校验,如果为空则进行异常抛出

CheckNull.java 类

package com.seesun2012.common.annotation;

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


/**
 * 自定义注解:校验非空字段
 *
 * @author [email protected]
 *
 */
@Documented
@Inherited
// 接口、类、枚举、注解
@Target(ElementType.FIELD)              
//只是在运行时通过反射机制来获取注解,然后自己写相应逻辑(所谓注解解析器)
@Retention(RetentionPolicy.RUNTIME)     
public @interface CheckNull {
    String message();
}

Userregister.java 类

package com.seesun2012.common.entity;

import java.io.Serializable;

import com.seesun2012.common.annotation.CheckNull;

/**
 * 用户注册实体类
 *
 * @author [email protected]
 *
 */
public class Userregister implements Serializable{

    private static final long serialVersionUID = 1L;
    
    //自定义注解
    @CheckNull(message="用户名不能为空")
    private String userAccount;
    //自定义注解
    @CheckNull(message="密码不能为空")
    private String passWord;

    public String getUserAccount() { return userAccount; }
    public void setUserAccount(String userAccount) { this.userAccount = userAccount; }
    public String getPassWord() { return passWord; }
    public void setPassWord(String passWord) { this.passWord = passWord; }

    @Override
    public String toString() {
        StringBuilder sb = new StringBuilder();
        sb.append(getClass().getSimpleName());
        sb.append(" [");
        sb.append("Hash = ").append(hashCode());
        sb.append(", userAccount=").append(userAccount);
        sb.append(", passWord=").append(passWord);
        sb.append(", serialVersionUID=").append(serialVersionUID);
        sb.append("]");
        return sb.toString();
    }
}

CommonUtils.java 类

package com.seesun2012.common.utils;

import java.beans.BeanInfo;
import java.beans.Introspector;
import java.beans.PropertyDescriptor;
import java.lang.reflect.Field;
import java.lang.reflect.Method;
import java.util.List;

import com.seesun2012.common.annotation.CheckNull;
import com.seesun2012.common.exception.CustBusinessException;

/**
 * 公共工具类
 *
 * @author [email protected]
 *
 */
public class CommonUtils{

    /**
     * 通过反射来获取javaBean上的注解信息,判断属性值信息,然后通过注解元数据来返回
     */
    public static <T> boolean doValidator(T clas){
        Class<?> clazz = clas.getClass();
        Field[] fields = clazz.getDeclaredFields();
        for (Field field : fields) {
            CheckNull checkNull = field.getDeclaredAnnotation(CheckNull.class);
            if (null!=checkNull) {
                Object value = getValue(clas, field.getName());
                if (!notNull(value)) {
                    throwExcpetion(checkNull.message());
                }
            }
        }
        return true;
    }

    /**
     * 获取当前fieldName对应的值
     *
     * @param clas      对应的bean对象
     * @param fieldName bean中对应的属性名称
     * @return
     */
    public static <T> Object getValue(T clas,String fieldName){
        Object value = null;
        try {
            BeanInfo beanInfo = Introspector.getBeanInfo(clas.getClass());
            PropertyDescriptor[] props = beanInfo.getPropertyDescriptors();
            for (PropertyDescriptor property : props) {
                if (fieldName.equals(property.getName())) {
                    Method method = property.getReadMethod();
                    value = method.invoke(clas, new Object[]{});
                }
            }
        } catch (Exception e) {
            e.printStackTrace();
        }
        return value;
    }

    /**
     * 非空校验
     *
     * @param value
     * @return
     */
    public static boolean notNull(Object value){
        if(null==value){
            return false;
        }
        if(value instanceof String && isEmpty((String)value)){
            return false;
        }
        if(value instanceof List && isEmpty((List<?>)value)){
            return false;
        }
        return null!=value;
    }

    public static boolean isEmpty(String str){
        return null==str || str.isEmpty();
    }
    public static boolean isEmpty(List<?> list){
        return null==list || list.isEmpty();
    }

    private static void throwExcpetion(String msg) {
        if(null!=msg){
            throw new CustBusinessException(msg);
        }
    }
    
}

Result.java 类

package com.seesun2012.common.entity;

import java.io.Serializable;
import java.util.HashMap;

public class Result extends HashMap<String, Object> implements Serializable {

    private static final long serialVersionUID = 1L;

    public static final Result SUCCEED = new Result(0, "操作成功");

    public Result(int status, String massage) {
        super();
        this.put("status", status).put("message", massage);
    }

    public Result put(String key, Object value) {
        super.put(key, value);
        return this;
    }

    public static Result build(int i, String message) {
        return new Result(i, message);
    }

}

CustBusinessException.java 类

package com.seesun2012.common.exception;

/**
 * 自定义异常类
 *
 * @author [email protected]
 *
 */
public class CustBusinessException extends RuntimeException{

    private static final long serialVersionUID = 1L;

    public CustBusinessException(){

    }

    public CustBusinessException(String str){
        super(str);
    }

    public CustBusinessException(Throwable throwable){
        super(throwable);
    }

    public CustBusinessException(String str, Throwable throwable){
        super(str, throwable);
    }

}

TestUtils.java 类

package com.seesun2012.test.utils;

import com.seesun2012.common.entity.Result;
import com.seesun2012.common.entity.Userregister;
import com.seesun2012.common.utils.CommonUtils;

public class TestUtils{

    public static void main(String[] args) {
        Userregister sss = new Userregister();
        sss.setUserAccount("asdflkjasokdfj");
        System.out.println(insertUser(sss));
    }

    public static Result insertUser(Userregister param){
        Result result = new Result(1, "新增失败");
        try {
            CommonUtils.doValidator(param);
            result = Result.build(0, "新增成功");
        } catch (Exception e) {
            result =  Result.build(1, e.getMessage());
        }
        return result;
    }

}

结语:该用户暂未添加

猜你喜欢

转载自www.cnblogs.com/seesun2012/p/9214632.html