利用ScriptEngineManager實現字符串公式靈活計算

在開發中我們可能會遇到好多不固定的公式計算 如有時候我們需要將excel中大量的計算公式轉換成java語言進行實現

如果說單純的用java的基本方法實現計算 我們就需要的分析excel中的公式將其一一併轉換成java代碼 這樣對我們開發人員來說工作量有點太大了 而且在轉換過程中很可能會出錯

下面就介紹一種簡單的實現方法

我們知道js的eval()方法可以執行字符串的代碼 而恰好jdk6增加了對腳本語言的支持 我們可以利用這個特性對計算實現簡單化的處理 

下面舉個例子

加入有個公式 

A+B*C 

A=1,B=2,C=3

我們可以將公式的A B C替換成數字 轉換爲 1+2*3 最後就可以得到結果了

 

接下來 我麼就探討下實現方法

直接以代嗎的形式來寫吧

public Double Calculation(String formula){
Double result=null;  //計算結果
ScriptEngineManager manager = new ScriptEngineManager();   //創建一個ScriptEngineManager對象
ScriptEngine engine = manager.getEngineByName("js");  //通過ScriptEngineManager獲得ScriptEngine對象
try {
result = (Double) engine.eval(formula);  //ScriptEngineeval方法執行腳本
} catch (ScriptException e) {
this.isSuccess=false;
result=Double.NaN;
}  
return result;
}

利用上面的方法我們就可以將一個數學表達式計算出結果

Calculation(“1+2*3”)-->得到結果7

顯然這樣是不能實現公式的靈活計算的 怎麼實現A+B*C  此時我們應該會想到將參數ABC替換成數字就可以實現計算結果了

接下來我們就探討如何實現替換的功能

說到替換 我們不難想到String的replaceAll方法,但是這樣我們會遇到一個問題就是替換形如A+AA+AAA的問題

如果str.replaceAll("A",2) ;結果就是2+22+222了

所以我想到了利用正則去準確匹配替換 方法如下

參數1:替換公式字的符串 參數2: 公式中要替換的字母如上邊的A  參數3:要替換成的數值

/**
 * 精確替換字符 防止出現 匹配A 時將AA匹配的情況
 */

public static String replaceStr(String sourceStr,String replaceKey,String replaceValue){
String replaceStrReg="";
for(char str_char:replaceKey.toCharArray()){
replaceStrReg+="[";
replaceStrReg+=str_char;
replaceStrReg+="]";
}
String startReg="^"+replaceStrReg+"([\\+\\-\\*/,)])";
String endReg="([\\+\\-\\*/,(])"+replaceStrReg+"$";
String reg="([^a-zA-Z])("+replaceStrReg+")"+"([^a-zA-Z])";
String endStr=sourceStr;
while(matcheStr(endStr,replaceKey)){
endStr=endStr.replaceAll(startReg, replaceValue+"$1");
endStr=endStr.replaceAll(reg, "$1"+replaceValue+"$3");
endStr=endStr.replaceAll(endReg, "$1"+replaceValue);
}
return endStr;
}


/**
* 精確匹配字符 防止出現 匹配A 時將AA匹配的情況
*/
public static Boolean matcheStr(String sourceStr,String matchStr){
String replaceStrReg="";
for(char str_char:matchStr.toCharArray()){
replaceStrReg+="[";
replaceStrReg+=str_char;
replaceStrReg+="]";
}
String startReg="^"+replaceStrReg+"([\\+\\-\\*/,)])[\\s\\S]*";
String endReg="[\\s\\S]*([\\+\\-\\*/,(])"+replaceStrReg+"$";
String reg="[\\s\\S]*([^A-Za-z])("+replaceStrReg+")"+"([^A-Za-z])[\\s\\S]*";
if(sourceStr.matches(startReg)||sourceStr.matches(reg)||sourceStr.matches(endReg)){
return true;
}else{
return false;
}
}

上述的正則表達式就是利用了數學公式中出現+-*/運算符號的規律


這樣我們就可以先替換 再計算了

我們的公式中可以實現計算平方等 如Math.pow(2,2) 只要js支持的數學公式都可以寫進去

至於怎麼將公式中的參數全部替換 我們可以用循環遍歷替換參數 這裏就不寫了 有了上面的方法就ok了


我有這樣的想法就是因爲開發中頻繁編寫公式 可能會出錯 另一方面純用java實現公式後期維護修改也麻煩 利用這種方法我們在後期修改公式時只需修改字符串公式即可

這樣我們還可以讓用戶自定義公式 


當然我們還可以利用ScriptEngineManager他實現執行js代碼塊 如if語句等 下面是個例子 可以參考下

public Map<String,Object> myFunction(String formula){
ScriptEngineManager manager = new ScriptEngineManager();  
ScriptEngine engine = manager.getEngineByName("js"); 
Object result;
try {
engine.eval(formula);
Invocable inv = (Invocable) engine;
result=inv.invokeFunction("js");  
Object function = engine.get("useFunction");
map.put("result", result);
map.put("function", function);
} catch (Exception e) {
result=Double.NaN;

return map;
}


大家如果有什麼好的想法可以與我交流下 共同進步嗎

歡迎大家與我交流


發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章