1、代码
import java.util.Set;
import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.Validator;
import javax.validation.ValidatorFactory;
import org.apache.commons.collections.CollectionUtils;
import org.apache.commons.lang3.StringUtils;
import com.lifesense.base.beans.param.RequestHeader;
import com.lifesense.base.exception.LifesenseBaseException;
import com.lifesense.base.exception.LifesenseResultCode;
/**
*
* hibernate validator 工具类
*
* @author lizhifeng
*
*/
public class HibernateValidateUtil {
private static Validator validator;
private static ThreadLocal<RequestHeader> requestHeaderThreadLocal = new ThreadLocal<>();
static {
if (validator == null) {
ValidatorFactory factory = Validation.buildDefaultValidatorFactory();
validator = factory.getValidator();
}
}
/**
* 用于验证用户传递的userId是否合法
* @return
*/
public static RequestHeader getCurrentRequestHeader(){
return requestHeaderThreadLocal.get();
}
public static void validate(Object obj) {
Set<ConstraintViolation<Object>> constraintViolations = validator.validate(obj);
if (CollectionUtils.isNotEmpty(constraintViolations)) {
for (ConstraintViolation<Object> c : constraintViolations) {
String message = c.getMessage();
if (StringUtils.isEmpty(message)) {
message = "请求参数不正确";
}
throw new LifesenseBaseException(LifesenseResultCode.PARAMS_FORMAT_ERROR.code, message);
}
}
}
public static void validate(RequestHeader requestHeader, Object obj) {
requestHeaderThreadLocal.set(requestHeader);
Set<ConstraintViolation<Object>> constraintViolations = validator.validate(obj);
if (CollectionUtils.isNotEmpty(constraintViolations)) {
for (ConstraintViolation<Object> c : constraintViolations) {
String message = c.getMessage();
if (StringUtils.isEmpty(message)) {
message = "请求参数不正确";
}
throw new LifesenseBaseException(LifesenseResultCode.PARAMS_FORMAT_ERROR.code, message);
}
}
}
}
2、自定义验证注解
import static java.lang.annotation.ElementType.ANNOTATION_TYPE;
import static java.lang.annotation.ElementType.CONSTRUCTOR;
import static java.lang.annotation.ElementType.FIELD;
import static java.lang.annotation.ElementType.METHOD;
import static java.lang.annotation.ElementType.PARAMETER;
import static java.lang.annotation.RetentionPolicy.RUNTIME;
import java.lang.annotation.Documented;
import java.lang.annotation.Retention;
import java.lang.annotation.Target;
import javax.validation.Constraint;
import javax.validation.Payload;
/**
*
* hibernate 自定义校验,支持long、double、int
*
* @author lizhifeng
*
*/
@Target({ METHOD, FIELD, ANNOTATION_TYPE, CONSTRUCTOR, PARAMETER })
@Retention(RUNTIME)
@Documented
@Constraint(validatedBy = RangeValidator.class) // 具体的实现
public @interface Range {
Class<?>[] groups() default {};
Class<? extends Payload>[] payload() default {};
String message() default "";
String min();
String max();
boolean nullAndZeroAllow() default false;
}
import javax.validation.ConstraintValidator;
import javax.validation.ConstraintValidatorContext;
import org.apache.commons.lang.math.NumberUtils;
/**
* 自定义校验逻辑实现
*
* @author lizhifeng
*
*/
public class RangeValidator implements ConstraintValidator<Range, Object> {
private String min;
private String max;
private boolean nullAndZeroAllow;
@Override
public void initialize(Range range) {
this.min = range.min();
this.max = range.max();
this.nullAndZeroAllow = range.nullAndZeroAllow();
}
@Override
public boolean isValid(Object value, ConstraintValidatorContext context) {
if (value == null) {
if (nullAndZeroAllow) {
return true;
} else {
return false;
}
}
if (value instanceof Long) {
long val = NumberUtils.toLong(String.valueOf(value), 0L);
if (nullAndZeroAllow && val == 0L) {
return true;
}
if (val < NumberUtils.toLong(min)) {
return false;
}
if (val > NumberUtils.toLong(max)) {
return false;
}
} else if (value instanceof Double) {
double val = NumberUtils.toDouble(String.valueOf(value), 0d);
if (nullAndZeroAllow && val == 0d) {
return true;
}
if (val < NumberUtils.toDouble(min)) {
return false;
}
if (val > NumberUtils.toDouble(max)) {
return false;
}
} else if (value instanceof Integer) {
int val = NumberUtils.toInt(String.valueOf(value), 0);
if (nullAndZeroAllow && val == 0) {
return true;
}
if (val < NumberUtils.toInt(min)) {
return false;
}
if (val > NumberUtils.toInt(max)) {
return false;
}
}
return true;
}
}