VerifyCodeUtil 对验证码进行处理
package com.imooc.myo2o.util;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.util.Random;
public class VerifyCodeUtil {
public static String drawRandomText(int width, int height, BufferedImage verifyImg) {
Graphics2D graphics = (Graphics2D) verifyImg.getGraphics();
graphics.setColor(Color.WHITE);
graphics.fillRect(0, 0, width, height);
graphics.setFont(new Font("微软雅黑", Font.BOLD, 40));
String baseNumLetter = "123456789abcdefghijklmnopqrstuvwxyzABCDEFGHJKLMNPQRSTUVWXYZ";
StringBuffer sBuffer = new StringBuffer();
int x = 10;
String ch = "";
Random random = new Random();
for (int i = 0; i < 4; i++) {
graphics.setColor(getRandomColor());
int degree = random.nextInt() % 30;
int dot = random.nextInt(baseNumLetter.length());
ch = baseNumLetter.charAt(dot) + "";
sBuffer.append(ch);
graphics.rotate(degree * Math.PI / 180, x, 45);
graphics.drawString(ch, x, 45);
graphics.rotate(-degree * Math.PI / 180, x, 45);
x += 48;
}
for (int i = 0; i < 6; i++) {
graphics.setColor(getRandomColor());
graphics.drawLine(random.nextInt(width), random.nextInt(height),
random.nextInt(width), random.nextInt(height));
}
for (int i = 0; i < 30; i++) {
int x1 = random.nextInt(width);
int y1 = random.nextInt(height);
graphics.setColor(getRandomColor());
graphics.fillRect(x1, y1, 2, 2);
}
return sBuffer.toString();
}
private static Color getRandomColor() {
Random ran = new Random();
Color color = new Color(ran.nextInt(256),
ran.nextInt(256), ran.nextInt(256));
return color;
}
}
在Controller层中的类调用生成图片方法时需要编写以下方法:
@RequestMapping(value = "/getVerifyCode", method = RequestMethod.GET)
public void getVerificationCode(HttpServletResponse response, HttpServletRequest request) {
try {
int width = 200;
int height = 69;
BufferedImage verifyImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_RGB);
String randomText = VerifyCodeUtil.drawRandomText(width, height, verifyImg);
request.getSession().setAttribute("verifyCode", randomText);
response.setContentType("image/png");
OutputStream os = response.getOutputStream();
ImageIO.write(verifyImg, "png", os);
os.flush();
os.close();
} catch (IOException e) {
e.printStackTrace();
}
}
CodeUtil 判断用户输入的验证码是否正确
package com.imooc.myo2o.util;
import javax.servlet.http.HttpServletRequest;
public class CodeUtil {
public static boolean checkVerifyCode(HttpServletRequest request){
String verifyCodeExpected = (String) request.getSession().getAttribute("verifyCode");
System.out.println(verifyCodeExpected);
String verifyCodeActual = HttpServletRequestUtil.getString(request,"verifyCodeActual");
System.out.println(verifyCodeActual);
if(verifyCodeActual == null || !verifyCodeActual.equalsIgnoreCase(verifyCodeExpected)){
return false;
}
return true;
}
}