You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
118 lines
3.6 KiB
118 lines
3.6 KiB
package com.yx.codeutil;
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import javax.imageio.ImageIO;
|
|
import java.awt.*;
|
|
import java.awt.image.BufferedImage;
|
|
import java.io.ByteArrayOutputStream;
|
|
import java.io.IOException;
|
|
import java.io.OutputStream;
|
|
import java.util.Random;
|
|
|
|
/**
|
|
* 验证码实现类
|
|
*/
|
|
public class SimpleCharVerifyCodeGenImpl implements IVerifyCodeGen {
|
|
|
|
private static final Logger logger = LoggerFactory.getLogger(SimpleCharVerifyCodeGenImpl.class);
|
|
|
|
private static final String[] FONT_TYPES = { "\u5b8b\u4f53", "\u65b0\u5b8b\u4f53", "\u9ed1\u4f53", "\u6977\u4f53", "\u96b6\u4e66" };
|
|
|
|
private static final int VALICATE_CODE_LENGTH = 4;
|
|
|
|
/**
|
|
* 设置背景颜色及大小，干扰线
|
|
*
|
|
* @param graphics
|
|
* @param width
|
|
* @param height
|
|
*/
|
|
private static void fillBackground(Graphics graphics, int width, int height) {
|
|
// 填充背景
|
|
graphics.setColor(Color.WHITE);
|
|
//设置矩形坐标x y 为0
|
|
graphics.fillRect(0, 0, width, height);
|
|
|
|
// 加入干扰线条
|
|
for (int i = 0; i < 8; i++) {
|
|
//设置随机颜色算法参数
|
|
graphics.setColor(RandomUtils.randomColor(40, 150));
|
|
Random random = new Random();
|
|
int x = random.nextInt(width);
|
|
int y = random.nextInt(height);
|
|
int x1 = random.nextInt(width);
|
|
int y1 = random.nextInt(height);
|
|
graphics.drawLine(x, y, x1, y1);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 生成随机字符
|
|
*
|
|
* @param width
|
|
* @param height
|
|
* @param os
|
|
* @return
|
|
* @throws IOException
|
|
*/
|
|
@Override
|
|
public String generate(int width, int height, OutputStream os) throws IOException {
|
|
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
|
|
Graphics graphics = image.getGraphics();
|
|
fillBackground(graphics, width, height);
|
|
String randomStr = RandomUtils.randomString(VALICATE_CODE_LENGTH);
|
|
createCharacter(graphics, randomStr);
|
|
graphics.dispose();
|
|
//设置JPEG格式
|
|
ImageIO.write(image, "JPEG", os);
|
|
return randomStr;
|
|
}
|
|
|
|
/**
|
|
* 验证码生成
|
|
*
|
|
* @param width
|
|
* @param height
|
|
* @return
|
|
*/
|
|
@Override
|
|
public VerifyCode generate(int width, int height) {
|
|
VerifyCode verifyCode = null;
|
|
try (
|
|
//将流的初始化放到这里就不需要手动关闭流
|
|
ByteArrayOutputStream baos = new ByteArrayOutputStream();
|
|
) {
|
|
String code = generate(width, height, baos);
|
|
verifyCode = new VerifyCode();
|
|
verifyCode.setCode(code);
|
|
verifyCode.setImgBytes(baos.toByteArray());
|
|
} catch (IOException e) {
|
|
logger.error(e.getMessage(), e);
|
|
verifyCode = null;
|
|
}
|
|
return verifyCode;
|
|
}
|
|
|
|
/**
|
|
* 设置字符颜色大小
|
|
*
|
|
* @param g
|
|
* @param randomStr
|
|
*/
|
|
private void createCharacter(Graphics g, String randomStr) {
|
|
char[] charArray = randomStr.toCharArray();
|
|
for (int i = 0; i < charArray.length; i++) {
|
|
//设置RGB颜色算法参数
|
|
g.setColor(new Color(50 + RandomUtils.nextInt(100),
|
|
50 + RandomUtils.nextInt(100), 50 + RandomUtils.nextInt(100)));
|
|
//设置字体大小，类型
|
|
g.setFont(new Font(FONT_TYPES[RandomUtils.nextInt(FONT_TYPES.length)], Font.BOLD, 26));
|
|
//设置x y 坐标
|
|
g.drawString(String.valueOf(charArray[i]), 15 * i + 5, 19 + RandomUtils.nextInt(8));
|
|
}
|
|
}
|
|
}
|
|
|
|
|