/**
* 执行函数脚本
*
* @since author: l00050286 date: 2023/4/10 15:26
* @param scriptStr 函数脚本字符串
* @param functionName 函数名
* @return java.lang.Object
*/
public static Object executeFunctionScript(String scriptStr,String functionName) throws Exception {
ScriptEngine engine = new ScriptEngineManager().getEngineByName("JavaScript");
engine.eval(scriptStr);
Invocable jsInvoke = (Invocable) engine;
return jsInvoke.invokeFunction(functionName);
}
/**
* 获取可执行函数Str
*
* @since author: l00050286 date: 2023/4/10 14:46
* @param sourceFormula 计算规则
* @param endSign 计算规则结束标识
* @param functionName 可执行函数名称
* @param factorValueMap 计算规则计算因子的值
* @return java.lang.String
*/
public static String getScriptStr(String sourceFormula, String endSign, String functionName,
Map<String, String> factorValueMap) {
int length = endSign.length();
String column = StringUtils.trimAllWhitespace(sourceFormula);
String splitStr = "IF|THEN|OTHERWISE|AND|OR|>=|<=|>|<|==|\\+|-|\\*|/|\\?|:";
if (column.indexOf("IF") == 0) {
String logicScript = column.substring(0, column.length() - length); // 去掉结束标记
Map<String, String> factorMap = new HashMap<>();
// 1、设置需要替换的最小计算因数
setFactorMap(factorMap, logicScript, splitStr);
// 2、格式化逻辑脚本
splitStr = "IF|THEN|OTHERWISE|AND|OR|>=|<=|>|<|==|\\+|-|\\*|/|\\;|\\?|:|" + endSign;
String formatFormula = CommonUtil.formatLogicFormula(column, endSign, splitStr);
// 3、格式化逻辑脚本转换成可执行函数
return scriptTransferFunction(formatFormula, factorMap, functionName, endSign, factorValueMap);
} else {
Map<String, String> factorMap = new HashMap<>();
// 1、设置需要替换的最小计算因数
setFactorMap(factorMap, column, splitStr);
// 2、格式化逻辑脚本
splitStr = "IF|THEN|OTHERWISE|AND|OR|>=|<=|>|<|==|\\+|-|\\*|/|\\;|\\?|:|" + endSign;
String formatFormula = CommonUtil.formatLogicFormula(column, endSign, splitStr);
// 3、格式化逻辑脚本转换成可执行函数
return scriptTransferFunction(formatFormula, factorMap, functionName, endSign, factorValueMap);
}
}
/**
* 设置需要替换的最小计算因数
*
* @since author: l00050286 date: 2023/4/10 14:40
* @param factorMap 保存需要替换的最小计算因数Map
* @param logicScript 逻辑脚本
* @param splitStr 分隔符
* @return void
*/
public static void setFactorMap(Map<String,String> factorMap,String logicScript,String splitStr){
String[] elementArray = logicScript.split(";");
int serialIndex=0;
// 解析公式获取需要替换的最小计算因数
for (String tempElement : elementArray) {
String[] split = tempElement.split(splitStr);
for (String minCalculationFactor : split) {
if(!CommonUtil.isEmpty(minCalculationFactor)){
int j = minCalculationFactor.indexOf("[");
int k = minCalculationFactor.indexOf("]");
if (j > -1 && k > -1) {
// 考虑到 ( ) 的情况
int l = minCalculationFactor.indexOf(")");
String element = (l > -1 ? minCalculationFactor.substring(j, l) : minCalculationFactor.substring(j));
if(factorMap.get(element) == null) {
serialIndex++;
factorMap.put(element, "param" + serialIndex);
// 考虑到存在()的情况
int m = minCalculationFactor.indexOf("(");
if (m > -1 || l > -1) {
factorMap.put(minCalculationFactor, element);
}
}else{
// 考虑到(,((,(((...和),)),)))...的情况
if(factorMap.get(minCalculationFactor) == null){
factorMap.put(minCalculationFactor, element);
}
}
}
}
}
}
}
/**
* 格式化逻辑公式
*
* @since author: l00050286 date: 2023/3/31 18:28
* @param column 计算规则公式
* @param endSign 原始字符串结束标识
* @param splitStr 分隔字符串
* "IF|THEN|OTHERWISE|AND|OR|>=|<=|>|<|==|\\+|-|\\*|/|\\;|"
* "IF|THEN|OTHERWISE|AND|OR|>=|<=|>|<|==|\\+|-|\\*|/"
* @return java.lang.String
*/
public static String formatLogicFormula(String column,String endSign,String splitStr) {
List<String> separatorList = new ArrayList<>();
String[] split1 = column.split(splitStr);
int i = 0;
int j = 1;
int length = split1.length;
while (column.length() > 0 && i < length && j < length) {
column = column.substring(split1[i].length());
int i1 = column.indexOf(split1[j]);
separatorList.add(column.substring(0, i1));
column = column.substring(i1);
i++;
j++;
}
StringBuilder stringBuilder = new StringBuilder();
int size = separatorList.size();
for (int i1 = 0; i1 < length; i1++) {
if(!CommonUtil.isEmpty(split1[i1])){
stringBuilder.append(split1[i1]).append(" ");
}
if (i1 < size) {
String tmp = separatorList.get(i1);
if(tmp.length()>1 && tmp.indexOf(";")==0){
stringBuilder.append(";").append(" ").append(tmp.substring(1)).append(" ");
}else{
stringBuilder.append(separatorList.get(i1)).append(" ");
}
}
}
stringBuilder.append(endSign);
return stringBuilder.toString();
}
/**
* 格式化逻辑脚本转换成可执行函数
*
* @since author: l00050286 date: 2023/4/10 15:05
* @param formatFormula 格式化逻辑脚本
* @param factorMap 需要替换的最小计算因数Map
* @param functionName 可执行函数名称
* @param endSign 格式化逻辑脚本结束标识
* @return java.lang.String
*/
public static String scriptTransferFunction(String formatFormula, Map<String, String> factorMap,
String functionName, String endSign,
Map<String, String> factorValueMap) {
String[] elementAllArray = formatFormula.split(" ");
StringBuilder stringBuilderFormula = new StringBuilder();
factorMap.forEach((key, value) -> {
if (key.indexOf("(") == -1 && key.indexOf(")") == -1) {
if (factorValueMap.get(key) == null) {
stringBuilderFormula.append("var ").append(value).append(" = 10;\n");
} else {
stringBuilderFormula.append("var ").append(value).append(" = ").append(factorValueMap.get(key)).append(";\n");
}
}
});
stringBuilderFormula.append("function ");
if (formatFormula.indexOf("IF") == 0) {
stringBuilderFormula.append(functionName).append("(){\n var result = 0;\n");
} else {
stringBuilderFormula.append(functionName).append("(){\n var result = ( \n");
}
for (String element : elementAllArray) {
switch (element) {
case "IF":
stringBuilderFormula.append(" if (");
break;
case "THEN":
stringBuilderFormula.append(") {\n\t result = ");
break;
case "AND":
stringBuilderFormula.append(" && ");
break;
case "OR":
stringBuilderFormula.append(" || ");
break;
case "OTHERWISE":
stringBuilderFormula.append(";\n } else {\n\t result = ");
break;
case ";":
stringBuilderFormula.append(";\n } else");
break;
default:
if (factorMap.get(element) == null) {
if (element.equals(endSign)) {
if(formatFormula.indexOf("IF") == 0){
stringBuilderFormula.append(";\n }\n return result ;\n}");
} else{
stringBuilderFormula.append(" );\n return result ;\n}");
}
} else {
stringBuilderFormula.append(" ").append(element);
}
} else {
String factor = factorMap.get(element);
String parameter = factorMap.get(factor);
if (parameter != null) {
element = element.replace(factor, parameter);
stringBuilderFormula.append(element);
} else {
stringBuilderFormula.append(" ").append(factorMap.get(element));
}
}
break;
}
}
return stringBuilderFormula.toString();
}
