Spring + validate數據校驗

1. 在controller類上加入 @Validated註解

2. 在參數前加入@Valid

@RestController("user")
@Validated
public class UserController {
    @PostMapping("register")
    public ResultBean register(@Valid @RequestBody RegisterBody registerBody) {
        return null;
    }
}
import io.swagger.annotations.ApiModel;
import io.swagger.annotations.ApiModelProperty;
import lombok.Data;
import lombok.NoArgsConstructor;

import javax.validation.constraints.NotNull;
import javax.validation.constraints.Size;


@NoArgsConstructor
@Data
@ApiModel("註冊用戶")
public class RegisterBody {
    @NotNull
    @Size(min = 8, max = 11)
    @ApiModelProperty("手機號")
    private String mobileNo;

    @NotNull
    private String code;
    private String imei;
    private String staff_name;
    private String company_code;
    private String service_provider_code;
}

3. 加入全局消息處理

import com.emcs.utils.ResultBean;
import org.springframework.validation.FieldError;
import org.springframework.validation.ObjectError;
import org.springframework.web.bind.MethodArgumentNotValidException;
import org.springframework.web.bind.annotation.ExceptionHandler;
import org.springframework.web.bind.annotation.RestControllerAdvice;

import java.util.List;
import java.util.stream.Collectors;

@RestControllerAdvice
public class GlobalExceptionHandler {

    private ObjectError e;

    @ExceptionHandler(MethodArgumentNotValidException.class)
    public ResultBean<?> validationErrorHandler(MethodArgumentNotValidException ex) {
        // 處理校驗異常
        List<String> errorInformation = ex.getBindingResult().getAllErrors()
                .stream()
                .map(e -> {
                    if (e instanceof FieldError) {
                        FieldError fieldError = (FieldError) e;
                        return fieldError.getField() + e.getDefaultMessage();
                    }
                    return e.getDefaultMessage();
                })
                .collect(Collectors.toList());
        //ResultBean爲一個封裝,可以直接返回String
        return new ResultBean<>( errorInformation.toString());
    }
}

4.效果:

{
  "status": 400,
  "message": "[mobileNo個數必須在8和11之間, code不能爲null]",
  "result": ""
}

5.參考:https://www.jianshu.com/p/5ab860e4f25a

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章