BeanValidator校验

BeanValidator校验

import com.google.common.base.Preconditions;
import com.google.common.collect.Lists;
import com.google.common.collect.Maps;
import org.apache.commons.collections4.MapUtils;

import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.ValidatorFactory;
import java.util.*;

public class BeanValidator {

    private static ValidatorFactory validatorFactory = Validation.buildDefaultValidatorFactory();

    public BeanValidator() {
    }

    public static  Map validate(T t, Class... groups) {
        Validator validator = validatorFactory.getValidator();
        Set validateResult = validator.validate(t, groups);
        if (validateResult.isEmpty()) {
            return Collections.emptyMap();
        } else {
            linkedHashMap errors = Maps.newlinkedHashMap();
            Iterator i$ = validateResult.iterator();

            while (i$.hasNext()) {
                ConstraintViolation constraintViolation = (ConstraintViolation) i$.next();
                errors.put(constraintViolation.getPropertyPath().toString(), constraintViolation.getMessage());
            }

            return errors;
        }
    }

    public static Map validateForObjects(Object first, Object... others) {
        return others != null && others.length != 0 ? validateForList(Lists.asList(first, others)) : validate(first, new Class[0]);
    }

    public static Map validateForList(Collection collection) {
        Preconditions.checkNotNull(collection);
        Iterator i$ = collection.iterator();

        Map errors;
        do {
            if (!i$.hasNext()) {
                return Collections.emptyMap();
            }
            Object object = i$.next();
            errors = validate(object, new Class[0]);
        } while (errors.isEmpty());

        return errors;
    }

    public static void checkPara(Object para) throws IllegalArgumentException {
        Map errors = BeanValidator.validateForObjects(para);
        if (MapUtils.isNotEmpty(errors)) {
            throw new IllegalArgumentException(errors.toString());
        }
    }
}

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

public class ListIntegerValueConstraintValidator implements ConstraintValidator {

    private Set set = new HashSet<>();

    // 初始化方法
    @Override
    public void initialize(ListIntegerValue constraintAnnotation) {
        // 取出注解中的值，放入Set
        int[] vals = constraintAnnotation.vals();
        for (int val : vals) {
            set.add(val);
        }

    }

    
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

public class ListIntegerValueConstraintValidator implements ConstraintValidator {

    private Set set = new HashSet<>();

    // 初始化方法
    @Override
    public void initialize(ListIntegerValue constraintAnnotation) {
        // 取出注解中的值，放入Set
        int[] vals = constraintAnnotation.vals();
        for (int val : vals) {
            set.add(val);
        }

    }

    
    @Override
    public boolean isValid(Integer value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@documented
// 指定自定义校验器，可以指定多个类型的，使用逗号隔开
@Constraint(validatedBy = { ListIntegerValueConstraintValidator.class })
@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
@Retention(RUNTIME)
public @interface ListIntegerValue {

    String message() default "";

    Class[] groups() default { };

    Class[] payload() default { };

    int[] vals() default { };
}

import javax.validation.Constraint;
import javax.validation.Payload;
import java.lang.annotation.documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;

import static java.lang.annotation.ElementType.*;
import static java.lang.annotation.RetentionPolicy.RUNTIME;

@documented
// 指定自定义校验器，可以指定多个类型的，使用逗号隔开
@Constraint(validatedBy = { ListStringValueConstraintValidator.class })
@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER, TYPE_USE })
@Retention(RUNTIME)
public @interface ListStringValue {

    String message() default "";

    Class[] groups() default { };

    Class[] payload() default { };

    String[] vals() default { };
}

import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import java.util.HashSet;
import java.util.Set;

public class ListStringValueConstraintValidator implements ConstraintValidator {

    private Set set = new HashSet<>();

    // 初始化方法
    @Override
    public void initialize(ListStringValue constraintAnnotation) {
        // 取出注解中的值，放入Set
        String[] vals = constraintAnnotation.vals();
        for (String val : vals) {
            set.add(val);
        }
        Object[] s = new Object[1];
    }

    
    @Override
    public boolean isValid(String value, ConstraintValidatorContext context) {
        return set.contains(value);
    }
}