fix:校验参数
This commit is contained in:
@@ -4,11 +4,14 @@ import lombok.AllArgsConstructor;
|
|||||||
import lombok.Data;
|
import lombok.Data;
|
||||||
import lombok.NoArgsConstructor;
|
import lombok.NoArgsConstructor;
|
||||||
|
|
||||||
|
import javax.validation.constraints.NotBlank;
|
||||||
|
|
||||||
@Data
|
@Data
|
||||||
@AllArgsConstructor
|
@AllArgsConstructor
|
||||||
@NoArgsConstructor
|
@NoArgsConstructor
|
||||||
public class CaptchaImageRequest {
|
public class CaptchaImageRequest {
|
||||||
|
|
||||||
|
@NotBlank(message = "用户名不能为空")
|
||||||
private String username;
|
private String username;
|
||||||
|
|
||||||
private Integer number;
|
private Integer number;
|
||||||
|
|||||||
@@ -15,6 +15,7 @@ import org.springframework.web.bind.annotation.RestController;
|
|||||||
|
|
||||||
import javax.validation.Valid;
|
import javax.validation.Valid;
|
||||||
import javax.validation.constraints.NotNull;
|
import javax.validation.constraints.NotNull;
|
||||||
|
import java.util.Objects;
|
||||||
|
|
||||||
@Slf4j
|
@Slf4j
|
||||||
@RestController
|
@RestController
|
||||||
@@ -25,9 +26,11 @@ public class ValidationCodeController {
|
|||||||
|
|
||||||
@PostMapping("/public/captchaImage")
|
@PostMapping("/public/captchaImage")
|
||||||
public Result<CaptchaDTO> getCaptchaImage(@RequestBody @NotNull(message = "请求参数不能为空") @Valid CaptchaImageRequest request) {
|
public Result<CaptchaDTO> getCaptchaImage(@RequestBody @NotNull(message = "请求参数不能为空") @Valid CaptchaImageRequest request) {
|
||||||
|
if (Objects.isNull(request.getNumber())) {
|
||||||
|
request.setNumber(4);
|
||||||
|
}
|
||||||
try {
|
try {
|
||||||
String randomCode = RandomCodeUtils.getNumberRandomCode(1);
|
String randomCode = RandomCodeUtils.getNumberRandomCode(1);
|
||||||
// String randomCode = "1";
|
|
||||||
CaptchaDTO captchaImage;
|
CaptchaDTO captchaImage;
|
||||||
if (StringUtils.isBlank(randomCode)) {
|
if (StringUtils.isBlank(randomCode)) {
|
||||||
captchaImage = validCodeGenerateFactory.get(CaptchaTypeEnum.NORMAL_CAPTCHA_IMAGE.getType()).getCaptchaImage(request);
|
captchaImage = validCodeGenerateFactory.get(CaptchaTypeEnum.NORMAL_CAPTCHA_IMAGE.getType()).getCaptchaImage(request);
|
||||||
@@ -49,6 +52,10 @@ public class ValidationCodeController {
|
|||||||
|
|
||||||
@PostMapping("/public/getSmsCode")
|
@PostMapping("/public/getSmsCode")
|
||||||
public Result<Void> getSmsCode(@RequestBody @NotNull(message = "请求参数不能为空") @Valid CaptchaImageRequest request) {
|
public Result<Void> getSmsCode(@RequestBody @NotNull(message = "请求参数不能为空") @Valid CaptchaImageRequest request) {
|
||||||
|
if (Objects.isNull(request.getNumber())) {
|
||||||
|
request.setNumber(4);
|
||||||
|
}
|
||||||
|
|
||||||
try {
|
try {
|
||||||
validCodeGenerateFactory.get(CaptchaTypeEnum.NUMBER_CAPTCHA_CODE.getType()).getCaptchaImage(request);
|
validCodeGenerateFactory.get(CaptchaTypeEnum.NUMBER_CAPTCHA_CODE.getType()).getCaptchaImage(request);
|
||||||
return Result.success();
|
return Result.success();
|
||||||
|
|||||||
Reference in New Issue
Block a user