生成图形验证码

验证码控制器

 

package cn.mycs.authenticate.server.controller;

import cn.mycs.core.support.StrKit;
import cn.mycs.core.util.KaptchaUtil;
import cn.mycs.server.persistence.redis.RedisService;
import com.google.code.kaptcha.Constants;
import com.google.code.kaptcha.Producer;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.imageio.ImageIO;
import javax.servlet.ServletOutputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.awt.image.BufferedImage;
import java.io.IOException;

/**
  *<p>验证码控制器</p>
  *
  *<pre>
  * @author gitamacai
  * @date 2018/8/23 9:13
  *</pre>
  */
@Controller
public class KaptchaController {
    @Autowired
    private Producer producer;
    @Autowired
    private RedisService redisService;

    /**
     * 生成验证码
     */
    @RequestMapping("/validateCode")
    public void validateCode(HttpServletRequest request, HttpServletResponse response) {
        HttpSession session = request.getSession();

        response.setDateHeader("Expires", 0);
        response.addHeader("Cache-Control", "post-check=0, pre-check=0");
        response.setHeader("Pragma", "no-cache");

        response.setContentType("image/jpeg");
        response.setHeader("Cache-Control", "no-store, no-cache, must-revalidate");

        String capText = producer.createText();
        if (StrKit.isNotEmpty(capText)) {
            //保存到session 
            //session.setAttribute(Constants.KAPTCHA_SESSION_KEY, capText);
            //保存到redis ,有效期为10分钟
            redisService.setByDbIndex(KaptchaUtil.getRedisKey(), capText,KaptchaUtil.KAPTCHA_TTL, KaptchaUtil.DBINDEX);
        }

        BufferedImage bi = producer.createImage(capText);
        try (ServletOutputStream out = response.getOutputStream()) {
            ImageIO.write(bi, "jpg", out);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

}

 

备注：

pom.xml 配置：

<dependency>
    <groupId>com.github.penggle</groupId>
    <artifactId>kaptcha</artifactId>
</dependency>