利用ScriptEngineManager实现字符串公式灵活计算

2021-09-30 03:07:23

在开发中我们可能会出现大量的公式计算，而这些公式可能并不确定。

比如用户今天说a=b+c 然而下次说公式不正确应该是a=d+e

如果单纯的在代码中把这些公式写死实现，后期修改维护工作量显然会增加好多。

下面就简单的介绍一种实现方法的思路：

我们知道js的eval（）方法可以执行字符串的代码而恰好jdk6增加了对脚本语言的支持我们可以利用这个特性对计算实现简单化的处理

下面举个例子

例如有个公式 A+B*C 其中A=1,B=2,C=3

我们可以将公式的A B C替换成数字转换为 1+2*3 最后就可以得到结果了

刚刚接触到ScriptEngine这个东西的时候仅仅了解其eval（）方法，于是我利用了上述例子的思路去实现将字母替换成相应的数字去得到数字公式进而得到运算结果。

那么替换的方法当然是用正则去替换了，java中String对象有个replaceAll（）方法可以实现。

当时想这个正则可是琢么了一会儿呢。然而我这种实现却白忙活了，因为后期发现没有必要这么麻烦

具体怎么实现大家可以想想看，我就不具体介绍了，有了思路比什么都重要。

然而过了一段时间，我在面试的时候和一个前辈聊天时，前辈告诉我不用自己写正则去替换，js中本来就有对象，也支持对象的运算，所以直接往里放对象就可以。

后来我自己查了查资料，发现果然可以：

为了可以实现打印出中间参与计算的变量，我利用反射实现了获取值的方法

下面就是代码的实现：

   public static void Calculation(){
       Student stu=new Student();
       stu.setAge(10);
       stu.setName("zhangsan");
       stu.setSex(false);

// 上边是student对象

       Class claz=stu.getClass();
       String className= claz.getSimpleName();
       //String formula="Student.name+Student.age+10"; //1
       String formula="function test(){ if(Student.age==10){ return 12;} }"; //2
       System.out.println("the formula is:"+formula);

//获取对象名称和值
       for(Field field: claz.getDeclaredFields()){
           try {
               //打开私有访问
//field.setAccessible(true);
               String fieldName = field.getName();
               Method m = (Method) claz.getMethod("get" +getMethodName(fieldName));
               System.out.println(getMethodName(fieldName));
               System.out.println(field.getGenericType()+"-"+"fieldName:"+fieldName+"="+m.invoke(stu));
           } catch (Exception e) {
               e.printStackTrace();
           }
       }
       Object result=null; //计算结果
       ScriptEngineManager manager = new ScriptEngineManager(); //创建一个ScriptEngineManager对象
       ScriptEngine engine = manager.getEngineByName("js"); //通过ScriptEngineManager获得ScriptEngine对象
       engine.put(className, stu); //将student对象放到ScriptEngine 中为计算变量提供值

       try {
           //result =engine.eval(formula); //1 用ScriptEngine的eval方法执行脚本 String formula="Student.name+Student.age+10";

engine.eval(formula); //2
           Invocable inv = (Invocable) engine; //2
           result=inv.invokeFunction("test"); //2 执行字符串 js test（）
           System.out.println("the result is :"+result.toString());
       } catch (Exception e) {
           System.out.println("错误");
           e.printStackTrace();
       }
   }

   // 把一个字符串的第一个字母大写
   private static String getMethodName(String fildeName) throws Exception{
   byte[] items = fildeName.getBytes();
   items[0] = (byte) ((char) items[0] - 'a' + 'A');
   return new String(items);
   }

上述代码还可注释2 打开注释1 试试一样可以的不过方式不同而已

有了这个思想我们就可以把公式维护到数据库或文件中方便我们后期对公式的维护修改这样利用这个工具应该可以减少计算代码编写的复杂度

后期我也发现了好多表达式引擎如：Aviator、IKExpression等。

码农公寓

相关文章