数据校验是我们日常开发过程中常做的工作,其目的性是为了提升我们系统的健壮性,举个简单的例子,我们要做一个创建用户的api接口服务,用户昵称是必填项,那么接口就需要对用户昵称进行非空校验。如果没填昵称抛出异常或者返回一些错误码。
那么问题来了,一次调用可能有n多个参数需要校验,比如用户手机绑定的注册服务,入参如下所示:
public class MobileRegisterDTO extends BaseDTO {
private String mobile;
private String countryCode;
private String verificationCode;
private String password;
private String nickname;
private Boolean isThirdAccountRegistration;
}
对于这个场景,很多人参数校验的代码可能就写成了下面这个样子:
public void mobileRegister(MobileRegisterDTO mobileRegisterDTO) {
if (Objects.isNull(mobileRegisterDTO) || StringUtils.isNotEmpty(mobileRegisterDTO.getMobile())
|| StringUtils.isNotEmpty(mobileRegisterDTO.getNickname())
|| StringUtils.isNotEmpty(mobileRegisterDTO.getPassword())
|| StringUtils.isNotEmpty(mobileRegisterDTO.getVerificationCode())
|| Objects.isNull(mobileRegisterDTO.getIsThirdAccountRegistration())) {
throw new ApplicationException("参数不合法")
}
// TODO 其他逻辑
}
如果增加参数,还需要再加一个if条件判断,十分不优雅,如何让这段代码变得高大上呢?
两种优雅解决注解问题 1.java function本质上和if-else区别不大,使用function显得高级点
private final List2.使用 validate注解> conditionChecked = Lists.newArrayList( Objects::nonNull, c -> StringUtils.isNotEmpty(c.getMobile()), c -> StringUtils.isNotEmpty(c.getNickname()), c -> StringUtils.isNotEmpty(c.getPassword()), c -> StringUtils.isNotEmpty(c.getVerificationCode()), c -> Objects.isNull(c.getIsThirdAccountRegistration()) ); public void mobileRegister(MobileRegisterDTO mobileRegisterDTO) { boolean argumentValid = conditionChecked.stream().allMatch(condition -> condition.apply(mobileRegisterDTO)); if (!argumentValid) { //参数校验不通过 } }
结合javax的validate功能,封装通用参数校验服务,实现参数校验的功能。
在参数变量上添加校验的注解:
public class MobileRegisterDTO extends BaseDTO {
@NotBlank(message = "手机号不能为空")
@Size(min = 6, max = 11)
private String mobile;
private String countryCode;
@NotBlank(message = "验证码不能为空")
private String verificationCode;
@NotBlank
private String password;
@NotBlank
private String nickname;
@NotNull
private Boolean isThirdAccountRegistration;
}
封装通用的校验工具服务:
import org.apache.commons.lang3.StringUtils;
import javax.validation.ConstraintViolation;
import javax.validation.Validation;
import javax.validation.Validator;
import java.util.List;
import java.util.Set;
public class ValidationUtil {
public final static Validator validator = Validation.buildDefaultValidatorFactory().getValidator();
public static String validateOne(T t) {
Set> validateResult = validator.validate(t);
if (validateResult != null && validateResult.size() != 0) {
StringBuilder sb = new StringBuilder();
for (ConstraintViolation valid : validateResult) {
sb.append(valid.getPropertyPath().toString()).append(StringUtils.SPACE).append(valid.getMessage())
.append(",");
}
return sb.toString();
}
return StringUtils.EMPTY;
}
public static String validateMutil(List ts) {
StringBuilder sb = new StringBuilder();
for (int index = 0; index < ts.size(); ++index) {
String validateOneResult = validateOne(ts.get(index));
if (!StringUtils.isBlank(validateOneResult)) {
sb.append("index[" + index + "]:").append(validateOneResult).append(";");
}
}
return sb.toString();
}
}
使用:
public void mobileRegister(MobileRegisterDTO mobileRegisterDTO) {
String errorMsg = ValidationUtil.validateOne(mobileRegisterDTO);
if (StringUtils.isNotBlank(errorMsg)) {
//参数校验不通过
}
}
总结
参数校验虽然看起来简单,简单的if-else也能实现参数校验的效果,但是使用一些spring、java新特性,可以让我们的代码看起来更加简洁优雅。



