验证码案例

checkCodeServlet

package servlet;

import javax.imageio.ImageIO;
import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.awt.*;
import java.awt.image.BufferedImage;
import java.io.IOException;
import java.util.Random;

@WebServlet("/checkCodeServlet")
public class checkCodeServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        int width = 100;
        int height = 50;
        //创建验证码图片对象
        BufferedImage image = new BufferedImage(width,height,BufferedImage.TYPE_INT_RGB);//宽，高，颜色类型
        //美化图片
        //填充背景色
        Graphics graphics = image.getGraphics();//画笔对象
        graphics.setColor(Color.cyan);
        graphics.fillRect(0,0,width,height);
        //画边框
        graphics.setColor(Color.red);
        graphics.drawRect(0,0,width-1,height-1);
        //生成随机字母或数字
        String str = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789";
        StringBuilder stringBuilder = new StringBuilder();
        graphics.setColor(Color.black);
        Random random = new Random();
        for (int i = 1;i <= 4;i++){
            int index = random.nextInt(str.length());
            char ch = str.charAt(index);//获取字符
            stringBuilder.append(ch);
            graphics.drawString(ch+"",width/5*i,height/2);//写验证码
        }
        //将验证码存入session
        String checkCode_session = stringBuilder.toString();
        request.getSession().setAttribute("checkCode_session",checkCode_session);
        //生成干扰线,13条
        graphics.setColor(Color.GREEN);
        for (int i = 0;i < 10;i++){
            int x1 = random.nextInt(width);
            int y1 = random.nextInt(height);
            int x2 = random.nextInt(width);
            int y2 = random.nextInt(height);
            graphics.drawLine(x1,y1,x2,y2);
        }
        //输出展示
        ImageIO.write(image,"jpg",response.getOutputStream());
    }

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.doPost(request, response);
    }
}

LoginServlet

package servlet;

import javax.servlet.ServletException;
import javax.servlet.annotation.WebServlet;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.io.IOException;

@WebServlet("/LoginServlet")
public class LoginServlet extends HttpServlet {
    protected void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        //设置编码
        request.setCharacterEncoding("utf-8");
        //获取参数
        String user = request.getParameter("user");
        String password = request.getParameter("password");
        String checkCode = request.getParameter("checkCode");
        //获取生成的验证码,然后把session里的验证码删除
        HttpSession session = request.getSession();
        String checkCode_session = (String)session.getAttribute("checkCode_session");
        session.removeAttribute("checkCode_session");
        //判断验证码正确与否
        if(checkCode_session!=null && checkCode_session.equalsIgnoreCase(checkCode)){//忽略大小写比较
            //判断用户名与密码是否一致
            if ( "zhangsan".equals(user) && "123".equals(password)){
                //存储用户信息,重定向到success表单
                session.setAttribute("user",user);
                response.sendRedirect(request.getContextPath()+"/success.jsp");
            }else{
                request.setAttribute("login_error","用户名或密码错误");
                request.getRequestDispatcher("/Login.jsp").forward(request,response);
            }
        }else{
            //存储信息到request域，请求转发到登录页面
            request.setAttribute("check_error","验证码错误");
            request.getRequestDispatcher("/Login.jsp").forward(request,response);
        }
    }

    protected void doGet(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
        this.doPost(request, response);
    }
}