疯狂的狮子li
2021-05-17 2ebcffb22ef17421b322e58a0f2b96388a4d83cd
ruoyi-admin/src/main/java/com/ruoyi/web/controller/common/CaptchaController.java
@@ -13,8 +13,8 @@
import com.ruoyi.common.constant.Constants;
import com.ruoyi.common.core.domain.AjaxResult;
import com.ruoyi.common.core.redis.RedisCache;
import com.ruoyi.framework.config.properties.CaptchaProperties;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;
@@ -42,18 +42,8 @@
   @Autowired
   private RedisCache redisCache;
   // 验证码类型
   @Value("${captcha.captchaType}")
   private String captchaType;
   // 验证码类别
   @Value("${captcha.captchaCategory}")
   private String captchaCategory;
   // 数字验证码位数
   @Value("${captcha.captchaNumberLength}")
   private int numberLength;
   // 字符验证码长度
   @Value("${captcha.captchaCharLength}")
   private int charLength;
   @Autowired
   private CaptchaProperties captchaProperties;
   /**
    * 生成验证码
@@ -67,17 +57,17 @@
      // 生成验证码
      CodeGenerator codeGenerator;
      AbstractCaptcha captcha;
      switch (captchaType) {
      switch (captchaProperties.getType()) {
         case "math":
            codeGenerator = new MathGenerator(numberLength);
            codeGenerator = new MathGenerator(captchaProperties.getNumberLength());
            break;
         case "char":
            codeGenerator = new RandomGenerator(charLength);
            codeGenerator = new RandomGenerator(captchaProperties.getCharLength());
            break;
         default:
            throw new IllegalArgumentException("验证码类型异常");
      }
      switch (captchaCategory) {
      switch (captchaProperties.getCategory()) {
         case "line":
            captcha = lineCaptcha;
            break;
@@ -92,9 +82,9 @@
      }
      captcha.setGenerator(codeGenerator);
      captcha.createCode();
      if ("math".equals(captchaType)) {
      if ("math".equals(captchaProperties.getType())) {
         code = getCodeResult(captcha.getCode());
      } else if ("char".equals(captchaType)) {
      } else if ("char".equals(captchaProperties.getType())) {
         code = captcha.getCode();
      }
      redisCache.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);
@@ -105,6 +95,7 @@
   }
   private String getCodeResult(String capStr) {
      int numberLength = captchaProperties.getNumberLength();
      int a = Convert.toInt(StrUtil.sub(capStr, 0, numberLength).trim());
      char operator = capStr.charAt(numberLength);
      int b = Convert.toInt(StrUtil.sub(capStr, numberLength + 1, numberLength + 1 + numberLength).trim());