fix(captcha): 修复行为验证码接口请求次数限制

continew-org · May 20, 2024 · 573e634 · 573e634
1 parent c1e9d31
commit 573e634
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/continew-admin-webapi/src/main/java/top/continew/admin/webapi/common/CaptchaController.java b/continew-admin-webapi/src/main/java/top/continew/admin/webapi/common/CaptchaController.java
@@ -23,6 +23,7 @@
 import cn.hutool.core.map.MapUtil;
 import cn.hutool.core.util.IdUtil;
 import cn.hutool.core.util.RandomUtil;
+import cn.hutool.core.util.StrUtil;
 import cn.hutool.extra.servlet.JakartaServletUtil;
 import com.anji.captcha.model.common.RepCodeEnum;
 import com.anji.captcha.model.common.ResponseModel;
@@ -86,13 +87,17 @@ public class CaptchaController {
  @GetMapping("/behavior")
  public R<Object> getBehaviorCaptcha(CaptchaVO captchaReq, HttpServletRequest request) {
  captchaReq.setBrowserInfo(JakartaServletUtil.getClientIP(request) + request.getHeader(HttpHeaders.USER_AGENT));
- return R.ok(behaviorCaptchaService.get(captchaReq).getRepData());
+ ResponseModel responseModel = behaviorCaptchaService.get(captchaReq);
+ CheckUtils.throwIf(() -> !StrUtil.equals(RepCodeEnum.SUCCESS.getCode(), responseModel
+ .getRepCode()), responseModel.getRepMsg());
+ return R.ok(responseModel.getRepData());
  }
 
  @Log(ignore = true)
  @Operation(summary = "校验行为验证码", description = "校验行为验证码")
  @PostMapping("/behavior")
- public R<Object> checkBehaviorCaptcha(@RequestBody CaptchaVO captchaReq) {
+ public R<Object> checkBehaviorCaptcha(@RequestBody CaptchaVO captchaReq, HttpServletRequest request) {
+ captchaReq.setBrowserInfo(JakartaServletUtil.getClientIP(request) + request.getHeader(HttpHeaders.USER_AGENT));
  return R.ok(behaviorCaptchaService.check(captchaReq));
  }