1. 程式人生 > >[email protected]註解對巢狀型別的校驗

[email protected]註解對巢狀型別的校驗

@Valid註解可以實現資料的驗證,你可以定義實體,在實體的屬性上新增校驗規則,而在API接收資料時新增@valid關鍵字,這時你的實體將會開啟一個校驗的功能,具體的程式碼如下,是最基本的應用:

實體:

public class DepartmentDto {

  @ApiModelProperty("id")
  private String id;

  @ApiModelProperty("上級Id")
  private String parentId;

  @ApiModelProperty("編號")
  @NotBlank(message = "部門編號不能為空。")
  
private String code; @ApiModelProperty("名稱") @NotBlank(message = "部門名稱不能為空。") private String name;
 @ApiModelProperty("員工集合")
@Builder.Default
private List<Employee> employees = new ArrayList<>();
}

Restful介面:

  @PostMapping()
  public Response<ClientAccount> initialAccount(
      @ApiParam(
"客戶編號") @PathVariable String code, @ApiParam("賬期") @PathVariable YearMonth accountPeriod, @ApiParam("請求體") @Valid @RequestBody Request<DepartmentDto> request) { ClientAccount result = clientAccountService.initialAccount( code, accountPeriod, request.getOperator(), request.getBody());{}

上面程式碼中,我們為請求體Request<DepartmentDto>添加了校驗,在測試時,如果你的DepartmnetDto.name為空字元時,當出現400的異常,麗時異常訊息是『部門名稱不能為空』,這對於我們來說是沒有問題的,也是符合我們要求的,下面看另一個場景。

需要驗證的實體是另一個實休的屬性

這種方式我們也需要會看到,一個大物件,如被封裝的其它小物件組成,比如部門下面有員工,這時如果需要驗證員工的有效性,需要如何實現呢?如果我們不修改原始碼,執行結果是否定的,它並不會校驗員工這個物件,而只針對第一層物件的屬性

我們將實體的員工屬性新增上@Valid即可實現對這個屬性的校驗

public class DepartmentDto {

  @ApiModelProperty("id")
  private String id;

  @ApiModelProperty("上級Id")
  private String parentId;

  @ApiModelProperty("編號")
  @NotBlank(message = "部門編號不能為空。")
  private String code;

  @ApiModelProperty("名稱")
  @NotBlank(message = "部門名稱不能為空。")
  private String name;

 @Valid
 @ApiModelProperty("員工集合")
 @Builder.Default
 private List<Employee> employees = new ArrayList<>();
}

下面看一下驗證結果,我們的400錯誤就可以在單元測試下面正常輸出了!

 @Test
  public void initialAccount_employee_name_empty() {
    List<Employee> employees = new ArrayList<>();
    employees.add(Employee.builder()
        .name("")
        .email("[email protected]")
        .idNumber("110111198203182012")
        .build());
    List<DepartmentDto> departments = new ArrayList<>();
    departments.add(DepartmentDto.builder()
        .name("部門")
        .description("技術部")
        .salaryType(SalaryType.ResearchAndDevelopmentCosts)
        .employees(employees)
        .build());
    ClientAccountDto clientAccountDto = ClientAccountDto.builder()
        .name("客戶")
        .departments(departments)
        .build();
    Request<ClientAccountDto> request = buildRequest(clientAccountDto);
    api.post()
        .uri("/v1/12345/2018-03")
        .body(BodyInserters.fromObject(request))
        .exchange()
        .expectStatus().isEqualTo(400)
        .expectBody()
        .jsonPath("$.errors[0].message").isEqualTo("姓名不能為空");
  }

結果如下,測試通過

如果是測試它是IsOk的話,由於使用者名稱為空,所以會出現錯誤提示

 api.post()
        .uri("/v1/12345/2018-03")
        .body(BodyInserters.fromObject(request))
        .exchange()
        .expectStatus().isOk();

可以看一下結果的提示資訊

感謝各位閱讀!

今天主要介紹 @Valid在專案中的使用!