MVEL实现java直接根据公式计算结果

工具类

import java.math.BigDecimal;
import java.util.Map;
import java.util.Map.Entry;
import java.util.stream.Collectors;

import org.mvel2.CompileException;
import org.mvel2.MVEL;
import org.mvel2.PropertyAccessException;


/**
 * 计算工具
 * 
 * @author valsong
 * @date Jul 24, 2017
 *
 */
public class CalculateUtils {

    /**
     * 根据传入的公式和参数进行计算
     * 
     * @param formula
     * @param variables
     * @return
     */
    public static  BigDecimal calculate(String formula, Map variables) {
        if (SimpleStringUtils.isBlank(formula)) {
            throw new CalculateException("MVEL formula can't be null! formula : " + formula); // 公式不能为空
        }
        if (variables == null || variables.size() == 0) {
            throw new CalculateException("MVEL variables can't be null! variables : " + String.valueOf(variables)); // 参数不能为空
        }

        try {
            // 将公式中的变量全部转化为BigDecimal类型
            variables.entrySet().stream().filter(e -> e != null && e.getKey() != null && e.getValue() != null)
                    .map(CalculateUtils::convert).collect(Collectors.toMap(Entry::getKey, Entry::getValue));

        } catch (NumberFormatException e) {
            throw new CalculateException(
                    "MVEL can't convert to BigDecimal, please check the variables : " + String.valueOf(variables) + "!",
                    e);
        } catch (Exception e) {
            throw e;
        }
        BigDecimal result = null;
        try {
            result = (BigDecimal) MVEL.eval(formula, variables);
        } catch (PropertyAccessException pae) {
            throw new CalculateException(
                    "MVEL please check the formula :" + formula + " & variables : " + String.valueOf(variables) + "!",
                    pae);
        } catch (CompileException ce) {
            throw new CalculateException("MVEL calculate error! ", ce);
        } catch (Exception e) {
            throw e;
        }
        return result;
    }


    /**
     * 将参数转化为Bigdecimal类型
     * 
     * @param entry
     * @return
     */
    @SuppressWarnings("unchecked")
    public static  Entry convert(Entry entry) {
        if (entry != null) {
            BigDecimal value = null;
            if (entry.getValue() instanceof BigDecimal) {
                value = (BigDecimal) entry.getValue();
            } else {
                value = NumberUtils.getNum(NumberUtils.getNum(entry.getValue()));
            }
            entry.setValue((T) value);
        }
        return entry;
    }

}

异常类

/**
 * 计算异常
 * 
 * @author Val Song
 * @date   Jul 28, 2017
 *
 */
public class CalculateException extends RuntimeException {
    private static final long serialVersionUID = 5162710183389028792L;

    /**
     * Constructs a {@code CalculateException} with no detail message.
     */
    public CalculateException() {
        super();
    }

    /**
     * Constructs a {@code CalculateException} with the specified detail
     * message.
     *
     * @param message
     *            the detail message.
     */
    public CalculateException(String message) {
        super(message);
    }

    /**
     * Constructs a {@code CalculateException} with the specified detail
     * message & cause
     * 
     * @param message
     * @param cause
     */
    public CalculateException(String message, Throwable cause) {
        super(message, cause);
    }

}

字符串工具类

/**
 * 字符串工具
 * 
 * @author valsong
 *
 */
public class SimpleStringUtils {

    /**
     * 判断字符串不为空
     * 
     * @param str
     * @return
     */
    public static boolean isNotBlank(String str) {
        return !isBlank(str);
    }

    /**
     * 判断字符串为空
     * 
     * @param str
     * @return
     */
    public static boolean isBlank(String str) {
        if (str == null || "".equals(str.trim())) {
            return true;
        }
        return false;
    }

    /**
     *  去首尾空格
     * 
     * @param s
     * @return
     */
    public static  String trim(T s) {
        if (s instanceof String) {
            return s == null ? null : ((String) s).trim();
        } else {
            return s == null ? null : String.valueOf(s).trim();
        }
    }

    /**
     * 下划线命名转驼峰式
     * 
     * @param str
     * @return
     */
    public static String toCamel(String str) {
        if (SimpleStringUtils.isBlank(str)) {
            return str;
        }
        StringBuffer buffer = new StringBuffer();
        str = str.toLowerCase().trim();
        char[] charArray = str.toCharArray();
        if (charArray != null) {
            for (int i = 0; i < charArray.length; i++) {
                if ('_' == charArray[i]) {
                    i = i + 1;
                    buffer.append(Character.toUpperCase(charArray[i]));
                } else {
                    buffer.append(charArray[i]);
                }
            }
        }
        return buffer.toString();
    }

    /**
     * 驼峰转下划线
     * 
     * @param str
     * @return
     */
    public static String toUnderline(String str) {
        if (SimpleStringUtils.isBlank(str)) {
            return str;
        }
        StringBuffer buffer = new StringBuffer();
        str = str.trim();
        char[] charArray = str.toCharArray();
        if (charArray != null) {
            for (int i = 0; i < charArray.length; i++) {
                if (Character.isUpperCase(charArray[i])) {
                    buffer.append("_");
                    buffer.append(Character.toLowerCase(charArray[i]));
                } else {
                    buffer.append(charArray[i]);
                }
            }
        }
        return buffer.toString();
    }

}

测试类简单粗暴String Double Integer Bigdecimal 都能支持

public class CalculateUtilsTest {

    /** 
     * 20 * 100 - ( 30 -20 ) + 20 ^ 3 + 20/2
     */
    @Test
    public void calculateTest4() {

        String formula = "A * B - (C -D) + E.pow(F) + G / H";

        Map variables = new HashMap<>();
        variables.put("A", "20");
        variables.put("B", 100L);
        variables.put("C", 30.0D);
        variables.put("D", 20);
        variables.put("E", new BigDecimal("20"));
        variables.put("F", "3");
        variables.put("G", "20");
        variables.put("H", "2");

        BigDecimal result = CalculateUtils.calculate(formula, variables);
        Assert.assertTrue(new BigDecimal("10000.0").compareTo(result) == 0);
    }

}