servlet验证码

package com.util;

import java.awt.Color;
import java.awt.Font;
import java.awt.Graphics;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.io.PrintWriter;
import java.util.Random;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

public class Code extends HttpServlet {
	
	public void doGet(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		
		Random r=new Random();
		//设置页面不要缓存
		response.addHeader("pragma", "no-cache");
		response.addHeader("cache-control", "no-cache");
		response.addHeader("expires", "-1");
		
		//设置页面的响应类型
		response.setContentType("image/jpeg");
		BufferedImage bi=new BufferedImage(80, 30, BufferedImage.TYPE_3BYTE_BGR);
		
		Graphics g=bi.getGraphics();
		
		g.setColor(new Color(r.nextInt(255),r.nextInt(255),r.nextInt(255)));
		
		g.fillRect(0,0,80,30);
		String str="";
		//随机四个字符
		for(int i=0;i<4;i++){
			str=str+RandomC();
		}
		request.getSession().setAttribute("code", str);
		g.setFont(new Font("黑体",0,25));
		g.setColor(new Color(r.nextInt(255),r.nextInt(255),r.nextInt(255)));
		g.drawString(str, 20, 20);	
		
		ImageIO.write(bi, "JPEG",response.getOutputStream());
		
	}
	
	
	public char RandomC(){
		Random r=new Random();
		
		char c=' ';
		
		int n=r.nextInt(3);
		if(n==0){
			c=(char)(r.nextInt(10)+48);
		}else if(n==1){
			c=(char)(r.nextInt(26)+97);
		}else if(n==2){
			c=(char)(r.nextInt(26)+65);
		}	
		
		return c;		
	}	
	public void doPost(HttpServletRequest request, HttpServletResponse response)
			throws ServletException, IOException {
		this.doGet(request, response);		
	}
}