com.pig4cloud.captcha.base.ArithmeticCaptchaAbstract Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of captcha-core Show documentation

There is a newer version: 2.2.4

package com.pig4cloud.captcha.base;

import com.googlecode.aviator.AviatorEvaluator;
import com.pig4cloud.captcha.engine.Symbol;

/**
 * 算术验证码抽象类 Created by 王帆 on 2019-08-23 上午 10:08.
 */
public abstract class ArithmeticCaptchaAbstract extends Captcha {
	// 计算公式
	private String arithmeticString;

	protected static int difficulty = 10;
	protected static int algorithmSign = 4;

	public ArithmeticCaptchaAbstract() {
		setLen(2);
	}

	/**
	 * 生成随机验证码
	 *
	 * @return 验证码字符数组
	 */
	@Override
	protected char[] alphas() {
		StringBuilder sb = new StringBuilder();
		for (int i = 0; i < len; i++) {
			sb.append(num(difficulty));
			if (i < len - 1) {
				int type = num(1, algorithmSign);
				if (type == 1) {
					sb.append(Symbol.ADD.getValue());
				} else if (type == 2) {
					sb.append(Symbol.SUB.getValue());
				} else if (type == 3) {
					sb.append(Symbol.MUL.getValue());
				}
			}
		}

		chars = String.valueOf(AviatorEvaluator.execute(sb.toString().replace("x", "*")));
		sb.append("=?");
		arithmeticString = sb.toString();
		return chars.toCharArray();
	}

	public String getArithmeticString() {
		checkAlpha();
		return arithmeticString;
	}

	public void setArithmeticString(String arithmeticString) {
		this.arithmeticString = arithmeticString;
	}

	public void setDifficulty(int difficulty) {
		ArithmeticCaptchaAbstract.difficulty = difficulty;
	}

	/**
	 * algorithmSign
	 * 2 : 支持加法 algorithmSign 3 : 支持加减法 algorithmSign 4 : 支持加减乘法
	 *
	 * @param algorithmSign 计算公式标示
	 */
	public void supportAlgorithmSign(int algorithmSign) {
		ArithmeticCaptchaAbstract.algorithmSign = algorithmSign;
	}

}