Java實現圖片驗證碼

代碼實現

public static final int WIDTH = 120;
/**
 * 圖片高度：50px
 */
public static final int HEIGHT = 50;
/**
 * 處理圖片驗證碼方法
 *
 * @param request
 * @param response
 */
@RequestMapping(value="/jcaptcha/captcha")
public void handleCaptchaRequest(HttpServletRequest request, HttpServletResponse response) {
        //生成6位隨機驗證碼
        String captcha = this.getRandomCode(3);
    try {
        //創建字節數組輸出流
        ByteArrayOutputStream jpegOutputStream = new ByteArrayOutputStream();
        //創建圖片緩存對象,BufferedImage.TYPE_INT_RGB ： 表示一個圖像，該圖像具有整數像素的 8 位 RGB 顏色
        BufferedImage bufferedImage = new BufferedImage(WIDTH, HEIGHT, BufferedImage.TYPE_INT_RGB);
        //獲取圖片的畫布
        Graphics graphics = bufferedImage.getGraphics();
        //設置畫布背景色
        graphics.setColor(Color.GREEN);
        //設置畫布填充區域
        graphics.fillRect(0, 0, WIDTH, HEIGHT);
        //邊框區域
        graphics.drawRect(1, 1, WIDTH - 2, HEIGHT - 2);
        //設置字體顏色
        graphics.setColor(Color.BLACK);
        //設置字體樣式
        graphics.setFont(new Font("微軟雅黑", Font.ITALIC, 32));
        //填充數據
        graphics.drawString(captcha, 10, 38);
        //將生成的驗證碼存放到session中
        request.getSession().setAttribute(Constants.CAPTCHA, captcha);

        ImageIO.write(bufferedImage, "jpeg", jpegOutputStream);
        byte[] captchaChallengeAsJpeg = jpegOutputStream.toByteArray();
        //將驗證碼輸出到頁面
        response.setHeader("Cache-Control", "no-store");
        response.setHeader("Pragma", "no-cache");
        response.setDateHeader("Expires", 0L);
        response.setContentType("image/jpeg");
        ServletOutputStream respOs = response.getOutputStream();
        respOs.write(captchaChallengeAsJpeg);
        respOs.flush();
        respOs.close();
    } catch (Exception e) {
        e.printStackTrace();
    }
}
/**
 * 生成0-9，a-z,A-Z的隨機字符串
 * @param count
 * @return
 */
private String getRandomCode(int count) {
    String[] array = {"A","B","C","D","E","F","G","H","I","J","K","L","M","N","O","P","Q","R","S","T","U","V","W","X","Y","Z","a","b","c","d","e","f","g","h","i","j","k","l","m","n","o","p","q","r","s","t","u","v","w","x","y","z","0","1","2","3","4","5","6","7","8","9"};

    StringBuilder result = new StringBuilder();

    for (int i = 0; i < count; i++) {
        //生成一個下標0-61之間一個整數
        int index = (int) Math.round(Math.random() * 61);
        //從array數組中獲取該下標字符放到result中
        result.append(array[index]);
    }
    return result.toString();
}