Spider/spider-flow
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

自定义函数返回数组、日期类型不对的问题

Browse Source
This commit is contained in:
mxd 2020-03-26 18:51:37 +08:00
parent 9cef96a04e
commit 41c0efb639
1 changed files with 25 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
spider-flow-core/src/main/java/org/spiderflow/core/script/ScriptManager.java
View File

@ -1,5 +1,6 @@
package org.spiderflow.core.script;
import jdk.nashorn.api.scripting.ScriptObjectMirror;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.spiderflow.core.expression.ExpressionTemplate;
@ -9,6 +10,7 @@ import javax.script.Invocable;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import java.util.Date;
import java.util.HashSet;
import java.util.Set;
import java.util.concurrent.locks.ReadWriteLock;
@ -107,9 +109,31 @@ public class ScriptManager {
}
try{
lock.readLock().lock();
return ((Invocable) scriptEngine).invokeFunction(functionName,args);
return convertObject(((Invocable) scriptEngine).invokeFunction(functionName, args));
} finally{
lock.readLock().unlock();
}
}
private static Object convertObject(Object object){
if(object instanceof ScriptObjectMirror){
ScriptObjectMirror mirror = (ScriptObjectMirror) object;
if(mirror.isArray()){
int size = mirror.size();
Object[] array = new Object[size];
for (int i = 0; i < size; i++) {
array[i] = convertObject(mirror.getSlot(i));
}
return array;
}else{
String className = mirror.getClassName();
if("Date".equalsIgnoreCase(className)){
return new Date(mirror.to(Long.class));
}
//其它类型待处理
}
}
return object;
}
}