新增缓存列表菜单功能

RuoYi

2022-06-25 196f6d48d8768243af3eb14c78315200603781c0

 src/main/java/com/ruoyi/project/common/CaptchaController.java

@@ -1,18 +1,24 @@
package com.ruoyi.project.common;



import java.io.ByteArrayOutputStream;

import java.awt.image.BufferedImage;

import java.io.IOException;

import java.util.concurrent.TimeUnit;

import javax.annotation.Resource;

import javax.imageio.ImageIO;

import javax.servlet.http.HttpServletResponse;

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

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

import org.springframework.util.FastByteArrayOutputStream;

import org.springframework.web.bind.annotation.GetMapping;

import org.springframework.web.bind.annotation.RestController;

import com.google.code.kaptcha.Producer;

import com.ruoyi.common.constant.CacheConstants;

import com.ruoyi.common.constant.Constants;

import com.ruoyi.common.utils.IdUtils;

import com.ruoyi.common.utils.VerifyCodeUtils;

import com.ruoyi.common.utils.sign.Base64;

import com.ruoyi.common.utils.uuid.IdUtils;

import com.ruoyi.framework.redis.RedisCache;

import com.ruoyi.framework.web.domain.AjaxResult;

import com.ruoyi.project.system.service.ISysConfigService;



/**

 * 验证码操作处理

@@ -22,8 +28,21 @@
@RestController

public class CaptchaController

{

    @Resource(name = "captchaProducer")

    private Producer captchaProducer;



    @Resource(name = "captchaProducerMath")

    private Producer captchaProducerMath;



    @Autowired

    private RedisCache redisCache;

    
    // 验证码类型

    @Value("${ruoyi.captchaType}")

    private String captchaType;

    
    @Autowired

    private ISysConfigService configService;



    /**

     * 生成验证码

@@ -31,32 +50,49 @@
    @GetMapping("/captchaImage")

    public AjaxResult getCode(HttpServletResponse response) throws IOException

    {

        // 生成随机字串

        String verifyCode = VerifyCodeUtils.generateVerifyCode(4);

        // 唯一标识

        String uuid = IdUtils.simpleUUID();

        String verifyKey = Constants.CAPTCHA_CODE_KEY + uuid;



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

        // 生成图片

        int w = 111, h = 36;

        ByteArrayOutputStream stream = new ByteArrayOutputStream();

        VerifyCodeUtils.outputImage(w, h, stream, verifyCode);

        try

        AjaxResult ajax = AjaxResult.success();

        boolean captchaOnOff = configService.selectCaptchaOnOff();

        ajax.put("captchaOnOff", captchaOnOff);

        if (!captchaOnOff)

        {

            AjaxResult ajax = AjaxResult.success();

            ajax.put("uuid", uuid);

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

            return ajax;

        }

        catch (Exception e)



        // 保存验证码信息

        String uuid = IdUtils.simpleUUID();

        String verifyKey = CacheConstants.CAPTCHA_CODE_KEY + uuid;



        String capStr = null, code = null;

        BufferedImage image = null;



        // 生成验证码

        if ("math".equals(captchaType))

        {

            e.printStackTrace();

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

        }



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

        // 转换流信息写出

        FastByteArrayOutputStream os = new FastByteArrayOutputStream();

        try

        {

            ImageIO.write(image, "jpg", os);

        }

        catch (IOException e)

        {

            return AjaxResult.error(e.getMessage());

        }

        finally

        {

            stream.close();

        }



        ajax.put("uuid", uuid);

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

        return ajax;

    }

}