SpringMVC Controller List参数的校验工具方法

短命女 2022-09-11 05:17 184阅读 0赞

请求类：

@Data
    public class ColumnUpdateDTO {
    
        @NotNull(message = "字段id不能为空")
        @ApiModelProperty(value = "字段id", required = true)
        private Long id;
    
        @ApiModelProperty(value = "字段中文名")
        private String columnName;
    
        @ApiModelProperty(value = "字段描述")
        private String comment;
    }

Controller：

@PostMapping("/updateColumnInfos")
        public ResponseEntity<ResultResponse> updateColumnInfos(
                @RequestBody List<ColumnUpdateDTO> columnUpdateDTOList) {
            ResultResponse resultResponse = BindingResultUtils.checkListError(columnUpdateDTOList);
            if (resultResponse.getCode() == HttpCodeEnum.FAIL.getCode()) {
                return ResponseEntity.ok(resultResponse);
            }
            // 业务代码
            return ResponseEntity.ok(new ResultResponse().success());
        }

工具方法：

public static ResultResponse checkListError(List list) throws IllegalAccessException {
            // 校验参数
            if (CollectionUtils.isEmpty(list)) {
                ResultResponse resultResponse = new ResultResponse().fail(HttpCodeEnum.FAIL.getCode(), "数组不能为空");
                return resultResponse;
            }
            List<String> errorMessage = new ArrayList<>();
    
            for (int i = 0; i < list.size(); i++) {
                Object object = list.get(i);
                Field[] fields = object.getClass().getDeclaredFields();
                for (Field field : fields) {
                    field.setAccessible(true);
    
                    NotNull notNull = field.getAnnotation(NotNull.class);
                    if (Objects.nonNull(notNull) && Objects.isNull(field.get(object))) {
                        errorMessage.add("index[" + i + "]" + notNull.message());
                    }
    
                    NotBlank notBlank = field.getAnnotation(NotBlank.class);
                    if (Objects.nonNull(notBlank) && Objects.isNull(field.get(object))) {
                        errorMessage.add("index[" + i + "]" + notBlank.message());
    
                    }
    
                    NotEmpty notEmpty = field.getAnnotation(NotEmpty.class);
                    if (Objects.nonNull(notEmpty) && Objects.isNull(field.get(object))) {
                        errorMessage.add("index[" + i + "]" + notEmpty.message());
                    }
                }
            }
    
            if (errorMessage.size() > 0) {
                ResultResponse resultResponse = new ResultResponse().fail(HttpCodeEnum.FAIL.getCode(), errorMessage.stream().map(String::valueOf).collect(Collectors.joining(";")));
                return resultResponse;
            } else {
                return new ResultResponse().success();
            }
        }