import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.ByteArrayInputStream;
import java.io.ByteArrayOutputStream;
import java.util.Map;
import java.util.Random;
import javax.imageio.ImageIO;
import javax.imageio.stream.ImageOutputStream;
import org.apache.struts2.interceptor.SessionAware;
public class ValidCodeAction extends BaseAction implements SessionAware {
private int width;//配置在xml中;生成验证码的宽度
private int height;//配置在xml中;生成验证码的高度
private int codeLength;//验证码中数字个数,配置在xml中;值:4
private String fontStyle;//配置在xml中;值:Times New Roman
private String randomValue;//配置在xml中;值:ABCDEFGHJKLMNPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz123456789
private Map<String, Object> session;
private ByteArrayInputStream inputStream;
public void setWidth(int width) {
this.width = width;
}
public void setHeight(int height) {
this.height = height;
}
public void setCodeLength(int codeLength) {
this.codeLength = codeLength;
}
public void setFontStyle(String fontStyle) {
this.fontStyle = fontStyle;
}
public void setRandomValue(String randomValue) {
this.randomValue = randomValue;
}
public void setSession(Map<String, Object> session) {
this.session = session;
}
public void setInputStream(ByteArrayInputStream inputStream) {
this.inputStream = inputStream;
}
public ByteArrayInputStream getInputStream() {
return inputStream;
}
public String execute() throws Exception {
// 在内存中创建图象
BufferedImage image = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
// 获取图形上下文
Graphics g = image.getGraphics();
// 生成随机类
Random random = new Random();
// 设定背景色
g.setColor(getRandColor(200, 250));
g.fillRect(0, 0, width, height);
// 设定字体
g.setFont(new Font(fontStyle, Font.PLAIN, 30));
// 随机产生155条干扰线,使图象中的认证码不易被其它程序探测到
g.setColor(getRandColor(160, 200));
drawLine(g, random);
// 取随机产生的认证码
String rand = randomString(g, random);
// 将认证码存入session
session.put("code", rand);
// 图象生效
g.dispose();
ByteArrayOutputStream output = new ByteArrayOutputStream();
ImageOutputStream imageOut = ImageIO.createImageOutputStream(output);
ImageIO.write(image, "JPEG", imageOut);
imageOut.close();
setInputStream(new ByteArrayInputStream(output.toByteArray()));
return SUCCESS;
}
private void drawLine(Graphics g, Random random) {
for (int i = 0; i < 155; i++) {
int x = random.nextInt(width);
int y = random.nextInt(height);
int xl = random.nextInt(12);
int yl = random.nextInt(12);
g.drawLine(x, y, x + xl, y + yl);
}
}
private String randomString(Graphics g, Random random) {
StringBuffer sRand = new StringBuffer(0);
// 取随机产生的认证码
String rand = "";
int randomValueLength = randomValue.length();
for (int i = 0; i < codeLength; i++) {
rand = String.valueOf(randomValue.charAt(random.nextInt(randomValueLength - 1)));
sRand.append(rand);
// 将认证码显示到图象中
g.setColor(new Color(20 + random.nextInt(110), 20 + random.nextInt(110), 20 + random.nextInt(110)));
g.drawString(rand, 25 * i, 25);
}
return sRand.toString();
}
/*
* 给定范围获得随机颜色
*/
private Color getRandColor(int fc, int bc) {
Random random = new 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);
}
}