java之使用Servlet生成驗證碼的原始碼分享

Evan1024發表於2024-03-06
import java.awt.Color;
import java.awt.Font;
import java.io.IOException;
import java.io.OutputStream;
import java.util.ArrayList;
import java.util.List;


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

import org.apache.commons.lang.math.RandomUtils;

import nl.captcha.Captcha;
import nl.captcha.Captcha.Builder;
import nl.captcha.gimpy.BlockGimpyRenderer;
import nl.captcha.text.producer.ChineseTextProducer;
import nl.captcha.text.producer.DefaultTextProducer;
import nl.captcha.text.renderer.DefaultWordRenderer;
import nl.captcha.text.renderer.WordRenderer;

public class CaptchaServlet extends HttpServlet implements Servlet {
    private static final long serialVersionUID = 1L;
    private static final String PARAM_HEIGHT = "height"; // 高度 預設為50


    private static final String PARAM_WIDTH = "width";// 寬度 預設為200

    private static final String PAEAM_NOISE = "noise";// 干擾線條 預設是沒有干擾線條

    private static final String PAEAM_TEXT = "text";// 文字

    protected int _width = 150;
    protected int _height = 40;
    protected boolean _noise = false;
    protected String _text = "number:1,word:1,number:1,number:1,word:1";

    /**
     * 初始化過濾器.將配置檔案的引數檔案賦值
     *
     * @throws ServletException
     */
    @Override
    public void init() throws ServletException {
        if (getInitParameter(PARAM_HEIGHT) != null) {
            _height = Integer.valueOf(getInitParameter(PARAM_HEIGHT));
        }

        if (getInitParameter(PARAM_WIDTH) != null) {
            _width = Integer.valueOf(getInitParameter(PARAM_WIDTH));
        }

        if (getInitParameter(PAEAM_NOISE) != null) {
            _noise = Boolean.valueOf(getInitParameter(PAEAM_NOISE));
        }

        if (getInitParameter(PAEAM_NOISE) != null) {
            _text = String.valueOf(getInitParameter(PAEAM_TEXT));
        }
    }

    /**
     * 因為獲取圖片只會有get方法
     */
    @Override
    public void doGet(HttpServletRequest req, HttpServletResponse resp) throws ServletException, IOException {

        Builder builder = new Captcha.Builder(_width, _height);
        // 增加邊框
        // builder.addBorder();
        // 是否增加干擾線條
        if (_noise == true) {
            builder.addNoise();
        }
        // ----------------自定義字型大小-----------
        // 自定義設定字型顏色和大小 最簡單的效果 多種字型隨機顯示
        List<Font> fontList = new ArrayList<Font>();
        fontList.add(new Font("微軟 雅黑", Font.LAYOUT_NO_START_CONTEXT, 30));// 可以設定斜體之類的
        DefaultWordRenderer dwr = new DefaultWordRenderer(getRandomColor(), fontList);
        // 加入多種顏色後會隨機顯示 字型空心

        WordRenderer wr = dwr;
        // 增加文字,預設為5個隨機字元.
        if (_text == null) {
            builder.addText();
        } else {
            String[] ts = _text.split(",");
            for (int i = 0; i < ts.length; i++) {
                String[] ts1 = ts[i].split(":");
                if ("chinese".equals(ts1[0])) {
                    builder.addText(new ChineseTextProducer(Integer.parseInt(ts1[1])), wr);
                } else if ("number".equals(ts1[0])) {
                    // 這裡沒有0和1是為了避免歧義 和字母I和O
                    char[] numberChar = new char[]{'2', '3', '4', '5', '6', '7', '8'};
                    builder.addText(new DefaultTextProducer(Integer.parseInt(ts1[1]), numberChar), wr);
                } else if ("word".equals(ts1[0])) {
                    // 原理同上
                    char[] numberChar = new char[]{'a', 'b', 'c', 'd', 'e', 'f', 'g', 'h', 'k', 'm', 'n', 'p', 'r',
                            'w', 'x', 'y'};
                    builder.addText(new DefaultTextProducer(Integer.parseInt(ts1[1]), numberChar), wr);
                } else {
                    builder.addText(new DefaultTextProducer(Integer.parseInt(ts1[1])), wr);
                }
            }

        }

        // --------------新增背景-------------
        // 設定背景漸進效果 以及顏色 form為開始顏色,to為結束顏色
        // GradiatedBackgroundProducer gbp=new GradiatedBackgroundProducer();
        // gbp.setFromColor(Color.yellow);
        // gbp.setToColor(Color.red);
        // 無漸進效果,只是填充背景顏色
        // FlatColorBackgroundProducer fbp=new
        // FlatColorBackgroundProducer(Color.red);
        // 加入網紋--一般不會用
        // SquigglesBackgroundProducer sbp=new SquigglesBackgroundProducer();
        // 沒發現有什麼用,可能就是預設的
        // TransparentBackgroundProducer tbp = new
        // TransparentBackgroundProducer();

        // builder.addBackground(gbp);
        // ---------裝飾字型---------------
        // 字型邊框齒輪效果 預設是3
        builder.gimp(new BlockGimpyRenderer(1));
        // 波紋渲染 相當於加粗
        // builder.gimp(new RippleGimpyRenderer());
        // 修剪--一般不會用
        // builder.gimp(new ShearGimpyRenderer(Color.red));
        // 加網--第一個引數是橫線顏色,第二個引數是豎線顏色
        // builder.gimp(new FishEyeGimpyRenderer(Color.red,Color.yellow));
        // 加入陰影效果 預設3,75
        // builder.gimp(new DropShadowGimpyRenderer());
        // 建立物件
        Captcha captcha = builder.build();

        req.getSession().setAttribute("simpleCaptcha", captcha);
        resp.setHeader("Cache-Control", "private,no-cache,no-store");
        resp.setContentType("image/png");
        try {
            OutputStream os = resp.getOutputStream();
            ImageIO.write(captcha.getImage(), "png", os);
            os.close();
        } catch (IOException e) {
            e.printStackTrace();
        }
    }

    public Color getRandomColor() {
        int n = RandomUtils.nextInt(100);

        if (n % 3 == 1) {
            return Color.RED;
        } else if (n % 3 == 2) {
            return Color.BLUE;
        }
        return Color.GREEN;
    }
}

相關文章