自定义注解验证

注解类


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

@Target({ ElementType.FIELD, ElementType.METHOD })
@Retention(RetentionPolicy.RUNTIME)
public @interface Attribute {
    // 字段属性说明
    String name() default "";

    // 字段是否强制校验必填,优于required,在任何校验方法内都会校验该字段是否为空
    boolean everyRequired() default false;

    // 字段是否必填
    boolean required() default false;

    // 字段固定长度限制
    int length() default 0;

    // 字段最大长度限制
    int maxLength() default 0;

    // 日期类型格式限制
    String dateFormat() default "";

    // 赋值限制列表,用于下拉列表
    String[] valueSetLimit() default {};
}

验证类


import java.lang.reflect.Field;
import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Modifier;
import java.lang.reflect.Proxy;
import java.text.ParseException;
import java.text.SimpleDateFormat;
import java.util.HashMap;
import java.util.Map;
import org.apache.commons.lang3.StringUtils;
import com.hikvision.annotation.Attribute;
import java.util.Arrays;

public class ParamVaildate {
    /***
     * 设置类属性的注解值,设置后该注解值将持续生效,若只是暂时需要改变,请在结束后再次调用更改回去
     * 
     * @param object
     *            对象类
     * @param atts
     *            属性数组
     * @param anns
     *            注解数组
     * @param values
     *            更改后的注解值数组
     */
    public static void setAttributeAnnotation(Object object, String[] atts, String[] anns, Object[] vals) {
        for (int i = 0; i < atts.length; i++) {
            setAttributeAnnotation(object, atts[i], anns[i], vals[i]);
        }
    }

    public static void setAttributeAnnotation(Object object, String att, String ann, Object val) {
        try {
            Field field = object.getClass().getDeclaredField(att);
            InvocationHandler h = Proxy.getInvocationHandler(field.getAnnotation(Attribute.class));
            Field hField = h.getClass().getDeclaredField("memberValues");
            hField.setAccessible(true);
            Map memberValues = (Map) hField.get(h);
            memberValues.put(ann, val);
        } catch (NoSuchFieldException | SecurityException e) {
            e.printStackTrace();
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
        } catch (IllegalAccessException e) {
            e.printStackTrace();
        }
    }

    /***
     * 验证类所有属性
     * 
     * @param object
     * @return
     */
    public static Map<String, Object> vaildateAll(Object object) {
        Map<String, Object> vailMap = new HashMap<String, Object>();
        if (object == null) {
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "验证的对象为空,请检查");
            return vailMap;
        }
        vailMap.put("errorCode", "0");

        Field[] fields = object.getClass().getDeclaredFields();
        try {
            for (Field field : fields) {
                // 排除static属性
                if (Modifier.isStatic(field.getModifiers())) {
                    continue;
                }
                Attribute att = field.getAnnotation(Attribute.class);
                // 排除没有注解的属性
                if (att == null) {
                    continue;
                }
                String value = (String) object.getClass()
                        .getMethod("get" + field.getName().substring(0, 1).toUpperCase() + field.getName().substring(1))
                        .invoke(object);
                // 强制校验必填
                if (att.everyRequired() && StringUtils.isEmpty(value)) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 必填校验
                if (att.required() && StringUtils.isEmpty(value)) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 固定长度校验
                if (value != null && att.length() > 0 && value.length() != att.length()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")固定长度限制为" + att.length() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 最大长度校验
                if (value != null && att.maxLength() > 0 && value.length() > att.maxLength()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")最大长度限制为" + att.maxLength() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 日期类型校验
                if (value != null && att.dateFormat().length() > 0) {
                    SimpleDateFormat sdf = new SimpleDateFormat(att.dateFormat());
                    try {
                        sdf.parse(value);
                    } catch (ParseException e) {
                        e.printStackTrace();
                        vailMap.put("errorCode", "1");
                        vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                                + ")日期类型限制为" + att.dateFormat() + "格式,请检查参数");
                        break;
                    }
                }
                // 赋值限制列表校验
                if (value != null && att.valueSetLimit().length > 0) {
                    boolean limitFlag = false;
                    for (int index = 0; index < att.valueSetLimit().length; index++) {
                        if (value.equals(att.valueSetLimit()[index])) {
                            limitFlag = true;
                            continue;
                        }
                    }
                    if (!limitFlag) {
                        vailMap.put("errorCode", "1");
                        vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                                + ")赋值限制列表为" + Arrays.toString(att.valueSetLimit()) + ",当前设值为'" + value + "'不符合规范,请检查参数");
                        break;
                    }
                }
            }
        } catch (NoSuchMethodException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "未在对应的实体中找到该方法:" + e.getMessage());
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (IllegalAccessException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (InvocationTargetException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (SecurityException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        }
        return vailMap;
    }

    /***
     * 验证类所有非空属性
     * 
     * @param object
     * @return
     */
    public static Map<String, Object> vaildateIsNotNulllAtt(Object object) {
        Map<String, Object> vailMap = new HashMap<String, Object>();
        if (object == null) {
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "验证的对象为空,请检查");
            return vailMap;
        }
        vailMap.put("errorCode", "0");

        Field[] fields = object.getClass().getDeclaredFields();
        try {
            for (Field field : fields) {
                // 排除static属性
                if (Modifier.isStatic(field.getModifiers())) {
                    continue;
                }
                Attribute att = field.getAnnotation(Attribute.class);
                // 排除没有注解的属性
                if (att == null) {
                    continue;
                }
                String value = (String) object.getClass()
                        .getMethod("get" + field.getName().substring(0, 1).toUpperCase() + field.getName().substring(1))
                        .invoke(object);
                // 强制校验必填
                if (att.everyRequired() && StringUtils.isEmpty(value)) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 非空且必填,不为空字符校验
                if (value != null && att.required() && "".equals(value.trim())) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 固定长度校验
                if (value != null && att.length() > 0 && value.length() != att.length()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")固定长度限制为" + att.length() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 最大长度校验
                if (value != null && att.maxLength() > 0 && value.length() > att.maxLength()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")最大长度限制为" + att.maxLength() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 日期类型校验
                if (value != null && att.dateFormat().length() > 0) {
                    SimpleDateFormat sdf = new SimpleDateFormat(att.dateFormat());
                    try {
                        sdf.parse(value);
                    } catch (ParseException e) {
                        e.printStackTrace();
                        vailMap.put("errorCode", "1");
                        vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                                + ")日期类型限制为" + att.dateFormat() + "格式,请检查参数");
                        break;
                    }
                }
            }
        } catch (NoSuchMethodException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "未在对应的实体中找到该方法:" + e.getMessage());
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (IllegalAccessException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (InvocationTargetException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (SecurityException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        }
        return vailMap;
    }

    /***
     * 根据传入的类名称和属性数组验证
     * 
     * @param object
     * @param atts
     * @return
     */
    public static Map<String, Object> vaildateAtts(Object object, String[] atts) {
        Map<String, Object> vailMap = new HashMap<String, Object>();
        if (object == null) {
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "验证的对象为空,请检查");
            return vailMap;
        }
        vailMap.put("errorCode", "0");

        try {
            for (int i = 0; i < atts.length; i++) {
                Field field = object.getClass().getDeclaredField(atts[i]);
                Attribute att = field.getAnnotation(Attribute.class);
                // 排除没有注解的属性
                if (att == null) {
                    continue;
                }
                String value = (String) object.getClass()
                        .getMethod("get" + field.getName().substring(0, 1).toUpperCase() + field.getName().substring(1))
                        .invoke(object);
                // 强制校验必填
                if (att.everyRequired() && StringUtils.isEmpty(value)) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 必填校验
                if (att.required() && StringUtils.isEmpty(value)) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage",
                            field.getAnnotation(Attribute.class).name() + "(" + field.getName() + ")不允许为空,请检查参数");
                    break;
                }
                // 固定长度校验
                if (value != null && att.length() > 0 && value.length() != att.length()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")固定长度限制为" + att.length() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 最大长度校验
                if (value != null && att.maxLength() > 0 && value.length() > att.maxLength()) {
                    vailMap.put("errorCode", "1");
                    vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                            + ")最大长度限制为" + att.maxLength() + ",实际长度为" + value.length() + ",请检查参数");
                    break;
                }
                // 日期类型校验
                if (value != null && att.dateFormat().length() > 0) {
                    SimpleDateFormat sdf = new SimpleDateFormat(att.dateFormat());
                    try {
                        sdf.parse(value);
                    } catch (ParseException e) {
                        e.printStackTrace();
                        vailMap.put("errorCode", "1");
                        vailMap.put("errorMessage", field.getAnnotation(Attribute.class).name() + "(" + field.getName()
                                + ")日期类型限制为" + att.dateFormat() + "格式,请检查参数");
                        break;
                    }
                }
            }
        } catch (NoSuchFieldException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "未在对应的实体中找到该属性:" + e.getMessage());
        } catch (NoSuchMethodException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", "未在对应的实体中找到该方法:" + e.getMessage());
        } catch (IllegalArgumentException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (IllegalAccessException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (SecurityException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        } catch (InvocationTargetException e) {
            e.printStackTrace();
            vailMap.put("errorCode", "1");
            vailMap.put("errorMessage", e.getMessage());
        }
        return vailMap;
    }
}

model类

public class PartnerModel{
    @Attribute(name = "名称", everyRequired = true, dateFormat = "YYYY-MM-DD")
    private String name;
    setget..;
}
  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值