修复响应体过大出现的乱码问题

RuoYi

2021-11-16 360ccc7adc703d3aeb3dacf54a8a1936168f7466

 ruoyi-gateway/src/main/java/com/ruoyi/gateway/service/impl/ValidateCodeServiceImpl.java

@@ -3,13 +3,11 @@
import java.awt.image.BufferedImage;

import java.io.IOException;

import java.util.concurrent.TimeUnit;



import javax.annotation.Resource;

import javax.imageio.ImageIO;



import org.springframework.beans.factory.annotation.Autowired;

import org.springframework.stereotype.Service;

import org.springframework.util.FastByteArrayOutputStream;



import com.google.code.kaptcha.Producer;

import com.ruoyi.common.core.constant.Constants;

import com.ruoyi.common.core.exception.CaptchaException;

@@ -18,6 +16,7 @@
import com.ruoyi.common.core.utils.sign.Base64;

import com.ruoyi.common.core.web.domain.AjaxResult;

import com.ruoyi.common.redis.service.RedisService;

import com.ruoyi.gateway.config.properties.CaptchaProperties;

import com.ruoyi.gateway.service.ValidateCodeService;



/**

@@ -28,11 +27,17 @@
@Service

public class ValidateCodeServiceImpl implements ValidateCodeService

{

    @Autowired

    private Producer producer;

    @Resource(name = "captchaProducer")

    private Producer captchaProducer;



    @Resource(name = "captchaProducerMath")

    private Producer captchaProducerMath;



    @Autowired

    private RedisService redisService;



    @Autowired

    private CaptchaProperties captchaProperties;



    /**

     * 生成验证码

@@ -40,16 +45,37 @@
    @Override

    public AjaxResult createCapcha() throws IOException, CaptchaException

    {

        // 生成验证码

        String capText = producer.createText();

        String capStr = capText.substring(0, capText.lastIndexOf("@"));

        String verifyCode = capText.substring(capText.lastIndexOf("@") + 1);

        BufferedImage image = producer.createImage(capStr);

        AjaxResult ajax = AjaxResult.success();

        boolean captchaOnOff = captchaProperties.getEnabled();

        ajax.put("captchaOnOff", captchaOnOff);

        if (!captchaOnOff)

        {

            return ajax;

        }



        // 保存验证码信息

        String uuid = IdUtils.simpleUUID();

        String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid;



        redisService.setCacheObject(verifyKey, verifyCode, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);

        String capStr = null, code = null;

        BufferedImage image = null;



        String captchaType = captchaProperties.getType();

        // 生成验证码

        if ("math".equals(captchaType))

        {

            String capText = captchaProducerMath.createText();

            capStr = capText.substring(0, capText.lastIndexOf("@"));

            code = capText.substring(capText.lastIndexOf("@") + 1);

            image = captchaProducerMath.createImage(capStr);

        }

        else if ("char".equals(captchaType))

        {

            capStr = code = captchaProducer.createText();

            image = captchaProducer.createImage(capStr);

        }



        redisService.setCacheObject(verifyKey, code, Constants.CAPTCHA_EXPIRATION, TimeUnit.MINUTES);

        // 转换流信息写出

        FastByteArrayOutputStream os = new FastByteArrayOutputStream();

        try

@@ -61,7 +87,6 @@
            return AjaxResult.error(e.getMessage());

        }



        AjaxResult ajax = AjaxResult.success();

        ajax.put("uuid", uuid);

        ajax.put("img", Base64.encode(os.toByteArray()));

        return ajax;