fix:校验参数

This commit is contained in:
xiang
2025-09-06 09:16:15 +08:00
parent 7e011672a1
commit 813eec5e7f
2 changed files with 11 additions and 1 deletions

View File

@@ -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;

View File

@@ -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();