2019独角兽企业重金招聘Python工程师标准>>>
动态计算
应用中可能会遇到一些自定义公式功能, 比如,用户可能自定义输入一些公式 a +b, a%2, '000'+s.substr(0,3), 如果你对这样的需求比较头痛, 请往下看吧, 我们马上解决.
Groovy
groovy是一种基于JVM的敏捷开发语言, 代码能够和java很好的结合, groovy 还可以使用java的类型, 是不是能力超强, 基本上你想到的都可以解决.
如何使用GroovyShell
在使用GroovyShell之前,先Import
import groovy.lang.Binding;
import groovy.lang.GroovyShell;
import groovy.lang.Script;
先看看下面的例子:
GroovyShell shell = new GroovyShell();
Object result = shell.evaluate("(1+5)/2");
System.out.println(result);
这么简单的示例, 相信大家都不会出错, 我们继续深入
绑定参数
实际应用中公式{ (a+b)/2 }是用户定义的, 值是应用系统中的数值, 比如系统中的当前a=1, b=5, 如何将值绑定给公式呢, 一切看代码
Binding binding = new Binding();
binding.setVariable("p",3.1415926);
binding.setVariable("r",5);
GroovyShell shell = new GroovyShell(binding);
Object result = shell.evaluate("p*r*r");
System.out.println(result);
result: 78.539815
引用数学函数
Java中的数学函数很多, 见Math.xxx, 上面的计算结果因为小数位太长, 我们需求四舍五入, 可以直接用Math.round,
Object result = shell.evaluate("Math.round(p*r*r)");
System.out.println(result);
result: 79
这时候的结果可能并不满意, 我们需要保留2位小数, 这时候Math.xxx其实并没有这样的函数直接拿来即用, groovy代码可能是这么写的:
import java.math.RoundingMode
def result = 78.539815
BigDecimal bg = new BigDecimal(result)
bg.setScale(2,RoundingMode.HALF_DOWN).doubleValue()
要是能够将上面的代码写成一个function, 直接调用那就简单多了...
自定义函数
自定义函数, 这个其实可以的, 我们先创建一个Java Class
public class EvalBaseScript extends Script {
@Override
public Object run() {
Method[] methods = EvalBaseScript.class.getDeclaredMethods();
StringBuilder sb=new StringBuilder();
for (Method method : methods) {
sb.append(method);
}
return sb.substring(0, sb.length()-1);
}
public static double round(String value, int scale) {
BigDecimal bd = new BigDecimal(value);
return bd.setScale(scale, 4).doubleValue();
}
public static double round(double value, int scale) {
BigDecimal bd = new BigDecimal(value);
return bd.setScale(scale, 4).doubleValue();
}
下面我们看看groovy中如何使用上面的class
Binding binding = new Binding();
if(vars!=null) {
vars.keySet().forEach(name ->
binding.setVariable(name, vars.get(name))
);
}
CompilerConfiguration cfg = new CompilerConfiguration();
cfg.setScriptBaseClass(EvalBaseScript.class.getName());
GroovyShell shell = new GroovyShell(binding,cfg);
try {
Script script = shell.parse(buildScript());
return script.run();
} catch (Exception ex) {
logger.error("catch script execution error, script:"+buildScript(), ex);
return 0;
}
至此java中执行自定义script的能力已经足够了, 你可以定义出你想要的, 任何脚本.