java實現隨機驗證碼圖片生成

本文實例為大傢分享瞭java生成隨機驗證碼圖片的具體代碼,供大傢參考,具體內容如下

1.controller

 /**
     * 獲取隨機數驗證碼圖片
     *
     * @param request  HttpServletRequest
     * @param response HttpServletResponse
     */
    @GetMapping(value = "/getRandomCode" ,produces = MediaType.APPLICATION_OCTET_STREAM_VALUE)
    public void getRandomCode(HttpServletRequest request, HttpServletResponse response) {
        baseService.getRandomCode(request, response);
}

2.service

void getRandomCode(HttpServletRequest request, HttpServletResponse response);

3.serviceImpl

@Autowired


 private RedisTemplate redisTemplate;

    @Autowired
    private CodeImageUtils codeImageUtils;
    /**
     *  獲取隨機數驗證碼圖片
     */
    @Override
    public void getRandomCode(HttpServletRequest request, HttpServletResponse response) {
        try {
            // 設置輸出類型為圖片
            response.setContentType("image/jpeg");

            // 生成隨機碼
            String verificationCode = codeImageUtils.getRandomCode();

            // 背景顏色與字符顏色
            Color color = codeImageUtils.getRandomColor();
            Color reverseColor = codeImageUtils.getReverseColor(color);

            //創建一張彩色的圖片
            BufferedImage bi = new BufferedImage(100, 30, BufferedImage.TYPE_INT_RGB);
            Graphics2D graphics = bi.createGraphics();
            //設置字體
            graphics.setFont(new Font(Font.SANS_SERIF, Font.BOLD, 16));
            graphics.setColor(color);
            //繪制背景
            graphics.fillRect(0, 0, 100, 30);
            graphics.setColor(reverseColor);
            //繪制字符
            graphics.drawString(verificationCode, 18, 20);

            //繪制最多50個隨機噪點//繪制最多50個隨機噪點
            Random random = new Random();
            int n = random.nextInt(50);
            for (int i = 0; i < n; i++) {
                graphics.drawRect(100, 30, 1, 1);
            }

            // 放到redis中 sessionId:code,過期時間60s
            redisTemplate.opsForValue().set(request.getSession().getId(),verificationCode,60, TimeUnit.SECONDS);

            //輸出到客戶端
            ServletOutputStream sos = response.getOutputStream();
            JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(sos);
            encoder.encode(bi);
            sos.flush();
        }catch (Exception ex){
            log.error("生成驗證碼失敗!",ex);
        }
    }

4.CodeImageUtils

@Component
public class CodeImageUtils {

    @Autowired
    RedisTemplate redisTemplate;

    /**
     * 生成4位隨機的驗證碼字符串
     *
     * @return
     */
    public  String getRandomCode() {
        // 通過random生成4位隨機數字碼
        Random random = new Random();
        StringBuilder rs = new StringBuilder();
        for (int i = 0; i < 4; i++) {
            rs.append(random.nextInt(10));
        }
        return rs.toString();
    }

    /**
     * 生成隨機的顏色
     *
     * @return
     */
    public  Color getRandomColor() {
        Random random = new Random();
        return new Color(random.nextInt(255), random.nextInt(255), random.nextInt(255));
    }

    /**
     * 獲取某顏色的反色
     * @param color
     * @return
     */
    public  Color getReverseColor(Color color) {
        return new Color(255 - color.getRed(), 255 - color.getGreen(), 255 - color.getBlue());
    }

    /**
     * 校驗
     * @param code
     * @return
     */
    public boolean valid(HttpServletRequest request,String code){
        // 獲取緩存的code
       Object cacheCode =  redisTemplate.opsForValue().get(request.getSession().getId());
       if(cacheCode ==null){
           return false;
       }
       return code.equals(cacheCode);
    }
}

5.請求該接口

6.ImageIO

JPEGCodec這個在高版本的jdk中過期瞭,替換為ImageIO

//輸出到客戶端
            ServletOutputStream sos = response.getOutputStream();
            /*JPEGImageEncoder encoder = JPEGCodec.createJPEGEncoder(sos);
            encoder.encode(bi);*/
            ImageIO.write(bi, "PNG", sos);
            sos.flush();

7.sessionId不一致

以上代碼在單節點部署情況下暫時沒發現什麼問題,但是在多節點部署時會出現sessionId不一致的問題,(別問我怎麼知道的…踩坑使我進步)。

解決方案為使用@EnableRedisHttpSession註解來開啟session共享。

新增依賴

<dependency>
            <groupId>org.springframework.session</groupId>
            <artifactId>spring-session-data-redis</artifactId>
            <version>2.1.12.RELEASE</version>
</dependency>

新增配置

@EnableRedisHttpSession(maxInactiveIntervalInSeconds = 86400)

以上就是本文的全部內容,希望對大傢的學習有所幫助,也希望大傢多多支持WalkonNet。

推薦閱讀: