一、验证码工具类使用到的对象

验证码对象

  1. import lombok.Data;
  3. /**
  4.  * @Author ：JADemo
  5.  * @Date ：2020/11/2
  6.  */
  7. @Data
  8. public class ImageCode {
  10.     private String code;
  12.     private Long expiration;
  14. }

自定义异常

  1. /**
  2.  * 自定义异常(知道原因的异常)
  3.  *
  4.  * @Auther : wodaoanran
  5.  * @Date : 2022/4/14
  6.  */
  7. public class CustomException extends RuntimeException{
  8.     public CustomException(String msg){
  9.         super(msg);
  10.     }
  11. }

验证码工具类

  1. import com.hzdy.credit.common.exception.CustomException;
  2. import com.hzdy.credit.entity.system.ImageCode;
  3. import com.hzdy.credit.util.maintain.Assert;
  4. import com.hzdy.credit.util.shiro.ShiroUtils;
  5. import org.apache.commons.lang3.StringUtils;
  6. import org.apache.shiro.session.Session;
  8. import javax.imageio.ImageIO;
  9. import javax.servlet.http.HttpServletRequest;
  10. import javax.servlet.http.HttpSession;
  11. import java.awt.*;
  12. import java.awt.geom.AffineTransform;
  13. import java.awt.image.BufferedImage;
  14. import java.io.IOException;
  15. import java.io.OutputStream;
  16. import java.security.SecureRandom;
  17. import java.util.Arrays;
  18. import java.util.HashMap;
  19. import java.util.Objects;
  20. import java.util.Random;
  22. /**
  23. * 验证码工具类
  24. * 
  25. * @author ruoyi
  26. */
  27. public class VerifyCodeUtils
  28. {
  29.     // 使用到Algerian字体，系统里没有的话需要安装字体，字体只显示大写，去掉了1,0,i,o几个容易混淆的字符
  30.     public static final String VERIFY_CODES = "123456789ABCDEFGHJKLMNPQRSTUVWXYZ";
  32.     private static final long  EFFECTIVITY_TIME = 1000 * 60;
  34.     private static final String IMAGE_CODE_KEY = ImageCode.class.getName();
  36.     private static final HashMap<String,ImageCode> MAP = new HashMap();
  38.     private static Random random = new SecureRandom();
  41.     /**
  42.     * 生成随机字符串并存入map
  43.     * @param verifySize
  44.     * @return
  45.     */
  46.     public static String generateAndSetMap(int verifySize){
  47.         String code = generateVerifyCode(verifySize);
  48.         ImageCode imageCode = new ImageCode();
  49.         imageCode.setCode(code);
  50.         imageCode.setExpiration(System.currentTimeMillis() + EFFECTIVITY_TIME);
  51.         //        ShiroUtils.getSession().setAttribute(IMAGE_CODE_KEY, imageCode);
  52.         MAP.put(IMAGE_CODE_KEY,imageCode);
  53.         return code;
  54.     }
  56.     public static void verifyMap(String code){
  57.         //转换为大写
  58.         code = code.toUpperCase();
  59.         //        ImageCode imageCode = (ImageCode) ShiroUtils.getSession().getAttribute(IMAGE_CODE_KEY);
  60.         ImageCode imageCode = MAP.get(IMAGE_CODE_KEY);
  61.         Assert.isEmpty(imageCode, "验证码为空请重新获取");
  62.         //        ShiroUtils.getSession().removeAttribute(IMAGE_CODE_KEY);
  63.         MAP.remove(IMAGE_CODE_KEY);
  64.         if(imageCode.getExpiration() < System.currentTimeMillis()){
  65.             throw new CustomException("验证码过期，请重新获取");
  66.         }
  67.         if(!imageCode.getCode().equals(code)){
  68.             throw new CustomException("错误的验证码");
  69.         }
  70.     }
  72.     /**
  73.     * 生成字符串验证码
  74.     * 
  75.     * @param verifySize 验证码长度
  76.     * @return
  77.     */
  78.     public static String generateVerifyCode(int verifySize)
  79.     {
  80.         return generateVerifyCode(verifySize, VERIFY_CODES);
  81.     }
  83.     /**
  84.     * 使用指定源生成验证码
  85.     * 
  86.     * @param verifySize 验证码长度
  87.     * @param sources 验证码字符源
  88.     * @return
  89.     */
  90.     public static String generateVerifyCode(int verifySize, String sources)
  91.     {
  92.         if (sources == null || sources.length() == 0)
  93.         {
  94.             sources = VERIFY_CODES;
  95.         }
  96.         int codesLen = sources.length();
  97.         Random rand = new Random(System.currentTimeMillis());
  98.         StringBuilder verifyCode = new StringBuilder(verifySize);
  99.         for (int i = 0; i < verifySize; i++)
  100.         {
  101.             verifyCode.append(sources.charAt(rand.nextInt(codesLen - 1)));
  102.         }
  103.         return verifyCode.toString();
  104.     }
  106.     /**
  107.     * 输出指定验证码图片流
  108.     * 
  109.     * @param w
  110.     * @param h
  111.     * @param os
  112.     * @param code
  113.     * @throws IOException
  114.     */
  115.     public static void outputImage(int w, int h, OutputStream os, String code) throws IOException
  116.     {
  117.         int verifySize = code.length();
  118.         BufferedImage image = new BufferedImage(w, h, BufferedImage.TYPE_INT_RGB);
  119.         Random rand = new Random();
  120.         Graphics2D g2 = image.createGraphics();
  121.         g2.setRenderingHint(RenderingHints.KEY_ANTIALIASING, RenderingHints.VALUE_ANTIALIAS_ON);
  122.         Color[] colors = new Color[5];
  123.         Color[] colorSpaces = new Color[] { Color.WHITE, Color.CYAN, Color.GRAY, Color.LIGHT_GRAY, Color.MAGENTA,
  124.                                            Color.ORANGE, Color.PINK, Color.YELLOW };
  125.         float[] fractions = new float[colors.length];
  126.         for (int i = 0; i < colors.length; i++)
  127.         {
  128.             colors[i] = colorSpaces[rand.nextInt(colorSpaces.length)];
  129.             fractions[i] = rand.nextFloat();
  130.         }
  131.         Arrays.sort(fractions);
  133.         g2.setColor(Color.GRAY);// 设置边框色
  134.         g2.fillRect(0, 0, w, h);
  136.         Color c = getRandColor(200, 250);
  137.         g2.setColor(c);// 设置背景色
  138.         g2.fillRect(0, 2, w, h - 4);
  140.         // 绘制干扰线
  141.         Random random = new Random();
  142.         g2.setColor(getRandColor(160, 200));// 设置线条的颜色
  143.         for (int i = 0; i < 20; i++)
  144.         {
  145.             int x = random.nextInt(w - 1);
  146.             int y = random.nextInt(h - 1);
  147.             int xl = random.nextInt(6) + 1;
  148.             int yl = random.nextInt(12) + 1;
  149.             g2.drawLine(x, y, x + xl + 40, y + yl + 20);
  150.         }
  152.         // 添加噪点
  153.         float yawpRate = 0.05f;// 噪声率
  154.         int area = (int) (yawpRate * w * h);
  155.         for (int i = 0; i < area; i++)
  156.         {
  157.             int x = random.nextInt(w);
  158.             int y = random.nextInt(h);
  159.             int rgb = getRandomIntColor();
  160.             image.setRGB(x, y, rgb);
  161.         }
  163.         shear(g2, w, h, c);// 使图片扭曲
  165.         g2.setColor(getRandColor(100, 160));
  166.         int fontSize = h - 4;
  167.         Font font = new Font("Algerian", Font.ITALIC, fontSize);
  168.         g2.setFont(font);
  169.         char[] chars = code.toCharArray();
  170.         for (int i = 0; i < verifySize; i++)
  171.         {
  172.             AffineTransform affine = new AffineTransform();
  173.             affine.setToRotation(Math.PI / 4 * rand.nextDouble() * (rand.nextBoolean() ? 1 : -1),
  174.                                  (w / verifySize) * i + fontSize / 2, h / 2);
  175.             g2.setTransform(affine);
  176.             g2.drawChars(chars, i, 1, ((w - 10) / verifySize) * i + 5, h / 2 + fontSize / 2 - 10);
  177.         }
  179.         g2.dispose();
  180.         ImageIO.write(image, "jpg", os);
  181.     }
  183.     private static Color getRandColor(int fc, int bc)
  184.     {
  185.         if (fc > 255)
  186.             fc = 255;
  187.         if (bc > 255)
  188.             bc = 255;
  189.         int r = fc + random.nextInt(bc - fc);
  190.         int g = fc + random.nextInt(bc - fc);
  191.         int b = fc + random.nextInt(bc - fc);
  192.         return new Color(r, g, b);
  193.     }
  195.     private static int getRandomIntColor()
  196.     {
  197.         int[] rgb = getRandomRgb();
  198.         int color = 0;
  199.         for (int c : rgb)
  200.         {
  201.             color = color << 8;
  202.             color = color | c;
  203.         }
  204.         return color;
  205.     }
  207.     private static int[] getRandomRgb()
  208.     {
  209.         int[] rgb = new int[3];
  210.         for (int i = 0; i < 3; i++)
  211.         {
  212.             rgb[i] = random.nextInt(255);
  213.         }
  214.         return rgb;
  215.     }
  217.     private static void shear(Graphics g, int w1, int h1, Color color)
  218.     {
  219.         shearX(g, w1, h1, color);
  220.         shearY(g, w1, h1, color);
  221.     }
  223.     private static void shearX(Graphics g, int w1, int h1, Color color)
  224.     {
  226.         int period = random.nextInt(2);
  228.         boolean borderGap = true;
  229.         int frames = 1;
  230.         int phase = random.nextInt(2);
  232.         for (int i = 0; i < h1; i++)
  233.         {
  234.             double d = (double) (period >> 1)
  235.                 * Math.sin((double) i / (double) period + (6.2831853071795862D * (double) phase) / (double) frames);
  236.             g.copyArea(0, i, w1, 1, (int) d, 0);
  237.             if (borderGap)
  238.             {
  239.                 g.setColor(color);
  240.                 g.drawLine((int) d, i, 0, i);
  241.                 g.drawLine((int) d + w1, i, w1, i);
  242.             }
  243.         }
  245.     }
  247.     private static void shearY(Graphics g, int w1, int h1, Color color)
  248.     {
  250.         int period = random.nextInt(40) + 10; // 50;
  252.         boolean borderGap = true;
  253.         int frames = 20;
  254.         int phase = 7;
  255.         for (int i = 0; i < w1; i++)
  256.         {
  257.             double d = (double) (period >> 1)
  258.                 * Math.sin((double) i / (double) period + (6.2831853071795862D * (double) phase) / (double) frames);
  259.             g.copyArea(i, 0, 1, h1, 0, (int) d);
  260.             if (borderGap)
  261.             {
  262.                 g.setColor(color);
  263.                 g.drawLine(i, (int) d, i, 0);
  264.                 g.drawLine(i, (int) d + h1, i, h1);
  265.             }
  267.         }
  268.     }
  271. }

二、验证码使用流程

  1. 使用generateAndSetMap()方法生成验证码
  2. 使用verifyMap()方法进行验证