java生成图片验证码

在日常工作中难免遇见各种登陆验证需求，这里提供一个简单的示例：
废话不多说直接上代码

import com.alibaba.fastjson.JSONObject;
import com.riching.cloud.common.annotations.AnonymousAccess;
import org.springframework.web.bind.annotation.GetMapping;
import org.springframework.web.bind.annotation.RestController;

import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.File;
import java.io.IOException;
import java.util.Random;

/**
 * 验证码生成器
 *
 * @author
 */
@RestController
public class VerificationCodeImgUtil {
  /**
   * @param width     图片宽
   * @param height    图片高
   * @param codeCount 字符个数
   * @param lineCount 干扰线条数
   */
  public static JSONObject createCode(int width, int height, int codeCount, int lineCount) {
    // 验证码
    String code = null;
    // 验证码图片Buffer
    BufferedImage buffImg = null;

    // 验证码范围,去掉0(数字)和O(拼音)容易混淆的(小写的1和L也可以去掉,大写不用了)
    char[] codeSequence = {'A', 'B', 'C', 'D', 'E', 'F', 'G', 'H', 'I', 'J', 'K', 'L', 'M', 'N', 'P', 'Q', 'R',
      'S', 'T', 'U', 'V', 'W', 'X', 'Y', 'Z', '1', '2', '3', '4', '5', '6', '7', '8', '9'};
    int x = 0, fontHeight = 0, codeY = 0;
    int red = 0, green = 0, blue = 0;

    x = width / (codeCount + 2);// 每个字符的宽度(左右各空出一个字符)
    fontHeight = height - 2;// 字体的高度
    codeY = height - 4;

    // 图像buffer
    buffImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
    Graphics2D g = buffImg.createGraphics();
    // 生成随机数
    Random random = new Random();
    // 将图像填充为白色
    g.setColor(Color.WHITE);
    g.fillRect(0, 0, width, height);
    // 创建字体,可以修改为其它的
    Font font = new Font("Fixedsys", Font.PLAIN, fontHeight);
    // Font font = new Font("Times New Roman", Font.ROMAN_BASELINE, fontHeight);
    g.setFont(font);

    for (int i = 0; i < lineCount; i++) {
      // 设置随机开始和结束坐标
      int xs = random.nextInt(width);// x坐标开始
      int ys = random.nextInt(height);// y坐标开始
      int xe = xs + random.nextInt(width / 8);// x坐标结束
      int ye = ys + random.nextInt(height / 8);// y坐标结束

      // 产生随机的颜色值，让输出的每个干扰线的颜色值都将不同。
      red = random.nextInt(255);
      green = random.nextInt(255);
      blue = random.nextInt(255);
      g.setColor(new Color(red, green, blue));
      g.drawLine(xs, ys, xe, ye);
    }

    // randomCode记录随机产生的验证码
    StringBuffer randomCode = new StringBuffer();
    // 随机产生codeCount个字符的验证码。
    for (int i = 0; i < codeCount; i++) {
      String strRand = String.valueOf(codeSequence[random.nextInt(codeSequence.length)]);
      // 产生随机的颜色值，让输出的每个字符的颜色值都将不同。
      red = random.nextInt(255);
      green = random.nextInt(255);
      blue = random.nextInt(255);
      g.setColor(new Color(red, green, blue));
      g.drawString(strRand, (i + 1) * x, codeY);
      // 将产生的四个随机数组合在一起。
      randomCode.append(strRand);
    }

    JSONObject jsonObject = new JSONObject();
    String randomStar = random.toString();
    jsonObject.put("randomStar",randomStar.replaceAll(",",""));
    try {
      // 生成文件名
      String files = "/code.png";
      // 生成文件路径//PathKit.getWebRootPath()获取项目绝对路径
      String filename = System.getProperty("user.dir")+files;
      File newimageFile = new File(filename);
      newimageFile.createNewFile();
      if(!newimageFile.exists()){
        newimageFile.createNewFile();
      }
      //清空缓冲
      g.dispose();
      //向文件中写入
      ImageIO.write(buffImg, "png", newimageFile);
      // 将验证码图片生成地址保存到jsonObject中。
      jsonObject.put("filePath",filename);
    } catch (IOException e) {
      e.printStackTrace();
    }
    code = randomCode.toString();
    // 将四位数字的验证码保存到jsonObject中。
    jsonObject.put("codeNumber",code);
    return jsonObject;
  }

  /**
   * 测试
   */
  @GetMapping("/testCode")
  @AnonymousAccess
  public void testCode() {
    JSONObject codeInfo = createCode(160, 40, 4, 150);
    System.out.println("图片验证码信息codeInfo======>" + JSONObject.toJSONString(codeInfo));
    String code = codeInfo.getString("codeNumber");
    String filePath = codeInfo.getString("filePath");
    System.out.println("验证码======>" + code);
    System.out.println("验证码图片生成地址filePath======>" + filePath);
  }
}