具体实现
import lombok.extern.slf4j.Slf4j;
import java.util.Base64;
import javax.imageio.ImageIO;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.util.Random;
import java.util.concurrent.ThreadLocalRandom;
/**
* 验证码工具类
*
*/
@Slf4j
public class LoginVerification {
public static final String ATU_NAME = "validateRand";
private static final ThreadLocal<String> SRAND_THREAD_LOCAL = new ThreadLocal(){
@Override
protected Object initialValue() {
return "";
}
};
private static final ThreadLocalRandom THREAD_LOCAL_RANDOM = ThreadLocalRandom.current();
public static void unloadThreadLocal(){
SRAND_THREAD_LOCAL.remove();
}
//设置随机的背景颜色
static Color getRandColor(int fc,int bc) {
Random random = THREAD_LOCAL_RANDOM;
if(fc>255){fc=255;}
if(bc>255) {bc=255;}
int r=fc+random.nextInt(bc-fc);
int g=fc+random.nextInt(bc-fc);
int b=fc+random.nextInt(bc-fc);
return new Color(r,g,b);
}
//具体获取base64串方法
public static String getPicBase64(){
BufferedImage pic = LoginVerification.getPic();
//io流
ByteArrayOutputStream baos = new ByteArrayOutputStream();
//写入流中
try {
ImageIO.write(pic, "JPEG", baos);
} catch (IOException e) {
log.error("出现异常:{}",e);
}
//转换成字节
byte[] bytes = baos.toByteArray();
Base64.Encoder encoder = Base64.getEncoder();
//转换成base64串
String png_base64 = encoder.encodeToString(bytes).trim();
log.info("生成的base64串 替换前:{}",png_base64);
//删除 \r\n
png_base64 = png_base64.replaceAll("\n", "").replaceAll("\r", "");
log.info("生成的base64串 替换后:{}",png_base64);
return png_base64;
}
public static BufferedImage getPic() {
int width=200, height=40;
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
Graphics g= image.getGraphics();
Random random = THREAD_LOCAL_RANDOM;
//设置验证码的文本
String sRand="";
//设置字体的大小
Font mFont = new Font("Times New Roman", Font.PLAIN,28);
g.setColor(getRandColor(200,250));
g.fillRect(0, 0, width, height);
g.setColor(new Color(102,102,102));
g.drawRect(0, 0, width-1, height-1);
g.setFont(mFont);
//设置颜色
g.setColor(getRandColor(160,200));
for (int i=0;i<155;i++)
{
int x = random.nextInt(width - 1);
int y = random.nextInt(height - 1);
int xl = random.nextInt(6) + 1;
int yl = random.nextInt(12) + 1;
g.drawLine(x,y,x + xl,y + yl);
}
for (int i = 0;i < 70;i++)
{
int x = random.nextInt(width - 1);
int y = random.nextInt(height - 1);
int xl = random.nextInt(12) + 1;
int yl = random.nextInt(6) + 1;
g.drawLine(x,y,x - xl,y - yl);
}
//设置验证码的类型
sRand = three(g);
log.info("验证码:{}",sRand);
SRAND_THREAD_LOCAL.set(sRand);
g.dispose();
return image;
}
/**
* 获取验证码
* @return
*/
public static String getRand(){
return SRAND_THREAD_LOCAL.get();
}
public static String getRandAndClear(){
String rand = SRAND_THREAD_LOCAL.get();
unloadThreadLocal();
return rand;
}
/**
* 英文字符加数字的验证码
*/
private static String one(Graphics g){
Random random = THREAD_LOCAL_RANDOM;
String sRand = "";
String[] letter = { "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" };
for (int i = 0; i < 4; i++) {
String tempRand = "";
if (random.nextBoolean()) {
tempRand = String.valueOf(random.nextInt(10));
} else {
tempRand = letter[random.nextInt(25)];
/* if (random.nextBoolean()) {// 随机将该字母变成小写
tempRand = tempRand.toLowerCase();
} */
}
sRand += tempRand;
g.setColor(new Color(20 + random.nextInt(10), 20 + random
.nextInt(110), 20 + random.nextInt(110)));
g.drawString(tempRand,20 * i + 20, 30);
}
return sRand;
}
//英文
private static String two(Graphics g){
Random random = THREAD_LOCAL_RANDOM;
String sRand = "";
for (int i=0;i<5;i++)
{
int itmp = random.nextInt(26) + 65;
char ctmp = (char)itmp;
sRand += String.valueOf(ctmp);
g.setColor(new Color(20+random.nextInt(110),20+random.nextInt(110),20+random.nextInt(110)));
g.drawString(String.valueOf(ctmp),20*i+10,30);
}
return sRand;
}
//4个数
private static String three(Graphics g){
Random random = THREAD_LOCAL_RANDOM;
String sRand = "";
for (int i = 0; i < 4; i++) {
String rand = String.valueOf(random.nextInt(10));
sRand += rand;
// 将认证码显示到图象中
g.setColor(new Color(20 + random.nextInt(110), 20 + random
.nextInt(110), 20 + random.nextInt(110)));
// 调用函数出来的颜色相同,可能是因为种子太接近,所以只能直接生成
g.drawString(rand,20*i+10,30);
}
return sRand;
}
//中文 肆+?=24
private static String four(Graphics g) {
Random random = THREAD_LOCAL_RANDOM;
String[] cn = {"壹", "贰", "叁", "肆", "伍", "陆", "柒", "捌", "玖", "拾"};
int x = random.nextInt(10) + 1;
int y = random.nextInt(30);
String sRand = String.valueOf(y);
g.setFont(new Font("楷体", Font.PLAIN, 25));// 设定字体
g.setColor(new Color(20 + random.nextInt(10), 20 + random.nextInt(110),
20 + random.nextInt(110)));
g.drawString(cn[x - 1], 1 * 1 + 6, 30);
g.drawString("+", 33, 30);
g.drawString("?", 50, 30);
g.drawString("=", 62, 30);
g.drawString(String.valueOf(x + y), 75, 30);
return sRand;
}
}