在Spring中的控制器生成验证码

package com.ch.ebusiness.controller.before;

import org.springframework.stereotype.Controller;
import org.springframework.web.bind.annotation.RequestMapping;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.OutputStream;
import java.util.Random;

@Controller
public class ValidateCodeController {
    private static final int WIDTH = 50;
    private static final int HEIGHT = 20;
    private static final int LENGTH = 4;
    private char code[] = {'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'i', 'j',
            'k', 'm', 'n', 'p', 'q', 'r', 's', 't', 'u', 'v', 'w', 'x', 'y',
            'z', 'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'J', 'K', 'L', 'M',
            'N', 'P', 'Q', 'R', 'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '2',
            '3', '4', '5', '6', '7', '8', '9'};

    @RequestMapping("/validateCode")
    public void validateCode(HttpServletRequest request,
                             HttpServletResponse response) throws ServletException, IOException {
        // TODO Auto-generated method stub
        // 设置响应报头信息
        response.setHeader("Pragma", "No-cache");
        response.setHeader("Cache-Control", "no-cache");
        response.setDateHeader("Expires", 0);
        // 设置响应的MIME类型
        response.setContentType("image/jpeg");


        BufferedImage image = new BufferedImage(WIDTH, HEIGHT,
                BufferedImage.TYPE_INT_RGB);

        Font mFont = new Font("Arial", Font.TRUETYPE_FONT, 18);
        Graphics g = image.getGraphics();
        Random rd = new Random();

        // 设置背景颜色
        g.setColor(new Color(rd.nextInt(55) + 200, rd.nextInt(55) + 200, rd
                .nextInt(55) + 200));
        g.fillRect(0, 0, WIDTH, HEIGHT);

        // 设置字体
        g.setFont(mFont);

        // 画边框
        g.setColor(Color.black);
        g.drawRect(0, 0, WIDTH - 1, HEIGHT - 1);

        // 随机产生的验证码
        String result = "";
        for (int i = 0; i < LENGTH; ++i) {
            result += code[rd.nextInt(code.length)];
        }

        HttpSession se = request.getSession();
        se.setAttribute("rand", result);

        // 画验证码
        for (int i = 0; i < result.length(); i++) {
            g.setColor(new Color(rd.nextInt(200), rd.nextInt(200), rd
                    .nextInt(200)));
            g.drawString(result.charAt(i) + "", 12 * i + 1, 16);
        }

        // 随机产生2个干扰线
        for (int i = 0; i < 2; i++) {
            g.setColor(new Color(rd.nextInt(200), rd.nextInt(200), rd
                    .nextInt(200)));
            int x1 = rd.nextInt(WIDTH);
            int x2 = rd.nextInt(WIDTH);
            int y1 = rd.nextInt(HEIGHT);
            int y2 = rd.nextInt(HEIGHT);
            g.drawLine(x1, y1, x2, y2);
        }

        // 释放图形资源
        g.dispose();
        try {
            OutputStream os = response.getOutputStream();

            // 输出图像到页面
            ImageIO.write(image, "JPEG", os);
        } catch (IOException e) {
            e.printStackTrace();
        }
    }
}

代码学习

这段代码是一个Spring MVC控制器类，用于生成验证码图片并返回给前端页面。该控制器类包含了一个validateCode方法，它使用了@RequestMapping注解来映射HTTP请求路径为"/validateCode"。

下面是代码中主要部分的功能解释：

1. validateCode方法：这是一个处理HTTP请求的方法，用于生成验证码图片并将其返回给前端页面。

2. 设置响应报头信息：在处理验证码请求之前，设置了响应报头信息，包括禁用缓存和设置响应的MIME类型为"image/jpeg"，以确保验证码图片不会被缓存，并且浏览器能正确识别它是一个图片。

3. 创建验证码图片：使用BufferedImage类创建了一个宽度为WIDTH、高度为HEIGHT的image对象，然后设置了背景颜色和字体。

4. 生成随机验证码：通过随机选择code数组中的字符，生成了一个长度为LENGTH的随机验证码字符串result。

5. 存储验证码到Session：将生成的验证码字符串存储到HttpSession中，以便后续验证用户输入的验证码。

6. 绘制验证码图片：使用Graphics类将生成的验证码字符串绘制到图片上，并加入干扰线增加验证码的难度，防止机器识别。

7. 输出图像：最后，将生成的验证码图片以JPEG格式输出到响应的输出流中，返回给前端页面。

这个验证码控制器对于需要用户输入验证码的场景非常有用，例如在用户注册、登录或进行重要操作时，可以使用验证码来增加安全性和防止自动化攻击。