准备环境
导入jar包(我用gradle来管理的jar包;maven也是可以的)
implementation group: 'org.springframework.boot', name: 'spring-boot-starter-validation', version: '2.3.10.RELEASE'
创建分组
package com.validater;
import javax.validation.groups.Default;
/**
* @author : DC
* @description : 分组校验
* @data : 2021-12-22 14:37
*/
public interface ValidationGroup {
/**
* 添加分组校验
*/
interface InsertV extends Default {
}
/**
* 编辑分组校验
*/
interface UpdateV extends Default {
}
/**
* 删除分组校验
*/
interface Delete extends Default {
}
/**
* 查询分组校验
*/
interface SelectV extends Default {
}
}
然后配上校验并返回错误消息的类(校验类)
package com.validater;
import org.springframework.util.CollectionUtils;
import javax.validation.ConstraintViolation;
import javax.validation.Valid;
import javax.validation.Validation;
import javax.validation.ValidatorFactory;
import java.util.Set;
/**
* @author DC
* @description spring参数检验
* @date 2021-12-15 15:52
*/
public class SpringValidation {
/**
* 分组验证参数
*
* @param validateData 参数
* @param validateGroup 分组
* @param <T>
*/
public static <T> String groupVerificationParameters(T validateData, Class<?>... validateGroup) {
ValidatorFactory validatorFactory = Validation.buildDefaultValidatorFactory();
Set<ConstraintViolation<T>> validate = validatorFactory.getValidator().validate(validateData, validateGroup);
if (!CollectionUtils.isEmpty(validate)) {
String messages = validate.stream()
.map(ConstraintViolation::getMessage)
.reduce((m1, m2) -> m1 + ";" + m2)
.orElse("参数输入有误!");
return messages;
}
return null;
}
}
创建实体类
@Data
public class Contract implements Serializable {
@NotNull(message = "添加者不能为空",groups = {InsertV.class})
private String addAccount;
@NotNull(message = "时间不能为空")
private Date updTime;
@NotNull(message = "更新者不能为空",groups = {UpdateV .class})
private String updAccount;
}
在controller层我们不做任何校验
@RestController
public class ContractController {
@Autowired
ContractService contractService;
@PostMapping("/contract")
public ResultObjectC saveContract(@RequestBody Contract contract){
return contractService.insertSelective(contract);
}
}
在service层中使用校验类
@Override
public ResultObjectC insertSelective(Contract record) {
//数据校验返回消息
String validate = SpringValidation.groupVerificationParameters(record, InsertV.class);
if (validate != null) {
return new ResultObjectC(BaseEnum.ResultStat.FAIL, validate);
}
}
此时可以看出
@NotNull(message = “添加者不能为空”,groups = {InsertV.class})
private String addAccount;进行了数据非空校验,因为我们在调用校验类时分组参数为InsertV.class
而updAccount属性上添加了@NotNull(message = “更新者不能为空”,groups = {UpdateV .class})的没有进行数据校验
@NotNull(message = “时间不能为空”)
private Date updTime;也进行了数据校验,因为没有分组的字段会进入默认分组校验