Java自定义公式计算

代码

import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import java.math.BigDecimal;
import java.math.RoundingMode;
import java.util.ArrayList;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

public class RegexTest {

    public static void main(String[] args) {
        Map<Integer, Double> map = new LinkedHashMap<>();
        map.put(0,1.0);
        map.put(1,1.0);
        String formulas = "([q1] + [q2]) / 2 ";
        double calculate = calculate(formulas, map);
        System.out.println("calculate:  " + calculate);
    }


    /**
     * 计算分数
     */
    public static double calculate(String formulas, Map<Integer, Double> questionScore) {

        System.out.println("自定义分数计算，formulas={}" + formulas +"questionScore={}"  + questionScore);
        String regex = "\\[q\\d+]";
        Pattern compile = Pattern.compile(regex);
        Matcher m = compile.matcher(formulas);
        System.out.println("m={}" +  m);
        List<Integer> qSeq = new ArrayList<>();
        while (m.find()) {
            qSeq.add(Integer.valueOf(formulas.substring(m.start() + 2, m.end() - 1)));
        }
        List<Double> qScores = new ArrayList<>();
        //题目排序后的值列表
        questionScore.forEach((key, value) -> qScores.add(value));

        System.out.println("qSeq=" + qSeq+ "qScores=" +  qScores);
        String evalStr = null;
        for (int i = 0; i < qSeq.size(); i++) {
            Integer index = qSeq.get(i);
            if (i == 0) {
                evalStr = formulas.replace("[q" + index + "]", String.valueOf(qScores.get(index - 1)));
            } else {
                evalStr = evalStr.replace("[q" + index + "]", String.valueOf(qScores.get(index - 1)));
            }
        }
        ScriptEngineManager manager = new ScriptEngineManager();
        ScriptEngine engine = manager.getEngineByName("js");
        Object result = 0;
        try {
            result = engine.eval(evalStr);
        } catch (ScriptException e) {
            System.out.println("转换分数异常");
        }
        BigDecimal two = BigDecimal.valueOf((Double) result);
        return two.setScale(2, RoundingMode.HALF_UP).doubleValue();
    }
}

结果