suguo
/
jetlinks-community
mirror of https://gitee.com/jetlinks/jetlinks-community.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

refactor: 优化nashorn支持

This commit is contained in:
zhouhao 2024-04-19 11:33:23 +08:00
parent 05abda43c7
commit a83ba95124
2 changed files with 169 additions and 35 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

169
jetlinks-components/script-component/src/main/java/org/jetlinks/community/script/nashorn/NashornScriptFactory.java
View File

@ -1,36 +1,113 @@
package org.jetlinks.community.script.nashorn;
import jdk.nashorn.api.scripting.ClassFilter;
import jdk.nashorn.api.scripting.JSObject;
import jdk.nashorn.api.scripting.NashornScriptEngineFactory;
import jdk.nashorn.internal.runtime.Undefined;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.jetlinks.community.script.jsr223.JavaScriptFactory;
import reactor.core.publisher.Flux;
import reactor.core.publisher.Mono;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineFactory;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Objects;
import java.util.stream.Collectors;
@Slf4j
public class NashornScriptFactory extends JavaScriptFactory implements ClassFilter {
public class NashornScriptFactory extends JavaScriptFactory {
static final Class<?> undefinedClass;
@Override
protected ScriptEngine createEngine() {
return new NashornScriptEngineFactory()
.getScriptEngine(new String[]{"-doe", "--language=es6", "--global-per-engine"},
NashornScriptFactory.class.getClassLoader(),
this);
static final Class<?> jsObjectClass;
static final Class<?> classFilterClass;
static final Class<?> scriptFactoryClass;
static final Method jsIsArrayMethod;
static final Method jsArrayValuesMethod;
static final Method getScriptEngineMethod;
static {
undefinedClass = loadClass(new String[]{
"org.openjdk.nashorn.internal.runtime.Undefined",
"jdk.nashorn.internal.runtime.Undefined"
});
jsObjectClass = loadClass(new String[]{
"org.openjdk.nashorn.api.scripting.JSObject",
"jdk.nashorn.api.scripting.JSObject"
});
classFilterClass = loadClass(new String[]{
"org.openjdk.nashorn.api.scripting.ClassFilter",
"jdk.nashorn.api.scripting.ClassFilter"
});
scriptFactoryClass = loadClass(new String[]{
"org.openjdk.nashorn.api.scripting.NashornScriptEngineFactory",
"jdk.nashorn.api.scripting.NashornScriptEngineFactory"
});
jsIsArrayMethod = getMethod(jsObjectClass, "isArray");
jsArrayValuesMethod = getMethod(jsObjectClass, "values");
getScriptEngineMethod = getMethod(scriptFactoryClass, "getScriptEngine", String[].class, ClassLoader.class, classFilterClass);
}
private static Method getMethod(Class<?> clazz, String name, Class<?>... parameters) {
if (clazz == null) {
return null;
}
try {
return clazz.getMethod(name, parameters);
} catch (NoSuchMethodException e) {
return null;
}
}
private static Class<?> loadClass(String[] classes) {
for (String clazz : classes) {
try {
return Class.forName(clazz);
} catch (ClassNotFoundException ignore) {
}
}
return null;
}
@Override
public boolean exposeToScripts(String s) {
return !isDenied(s);
@SneakyThrows
protected ScriptEngine createEngine() {
ScriptEngineFactory engineFactory = (ScriptEngineFactory) scriptFactoryClass
.getConstructor()
.newInstance();
Object filter = Proxy.newProxyInstance(
this.getClass().getClassLoader(),
new Class[]{classFilterClass},
(proxy, method, args) -> {
if (Objects.equals(method.getName(), "exposeToScripts")) {
return !isDenied((String) args[0]);
}
return method.invoke(proxy, args);
});
return (ScriptEngine) getScriptEngineMethod
.invoke(engineFactory,
new String[]{"-doe", "--language=es6", "--global-per-engine"},
NashornScriptFactory.class.getClassLoader(),
filter);
}
@Override
protected boolean valueIsUndefined(Object value) {
return value == null || value instanceof Undefined;
return valueIsUndefined0(value);
}
@Override
@ -38,34 +115,72 @@ public class NashornScriptFactory extends JavaScriptFactory implements ClassFilt
return convertToJavaObject(value);
}
public static Object convertToJavaObject(Object object) {
if (object instanceof JSObject) {
return convertJSObject(((JSObject) object));
private static boolean valueIsUndefined0(Object value) {
if (value == null) {
return true;
}
if (object instanceof Undefined) {
if (undefinedClass != null) {
return undefinedClass.isInstance(value);
}
return false;
}
@SuppressWarnings("all")
public static Object convertToJavaObject(Object object) {
if (valueIsUndefined0(object)) {
return null;
}
if (jsIsObject(object)) {
return convertJSObject(object);
}
if (object instanceof Flux) {
return ((Flux<?>) object)
.mapNotNull(NashornScriptFactory::convertToJavaObject);
}
if (object instanceof Mono) {
return ((Mono<?>) object)
.mapNotNull(NashornScriptFactory::convertToJavaObject);
}
return object;
}
public static Object convertJSObject(JSObject jsObject) {
if (jsObject.isArray()) {
return jsObject
.values()
private static boolean jsIsObject(Object obj) {
return jsObjectClass != null && jsObjectClass.isInstance(obj);
}
private static boolean jsIsArray(Object obj) {
if (jsIsArrayMethod == null) {
return false;
}
try {
return (boolean) jsIsArrayMethod.invoke(obj);
} catch (IllegalAccessException | InvocationTargetException ignore) {
}
return false;
}
@SneakyThrows
private static Object convertJSObject(Object jsObject) {
if (jsIsArray(jsObject)) {
@SuppressWarnings("all")
Collection<Object> values = (Collection) jsArrayValuesMethod.invoke(jsObject);
return values
.stream()
.map(obj -> {
if (obj instanceof JSObject) {
return convertJSObject(((JSObject) obj));
if (jsIsObject(obj)) {
return convertJSObject(obj);
}
return obj;
}).collect(Collectors.toList());
})
.collect(Collectors.toList());
}
if (jsObject instanceof Map) {
Map<Object, Object> newMap = new HashMap<>(((Map<?, ?>) jsObject).size());
for (Map.Entry<?, ?> entry : ((Map<?, ?>) jsObject).entrySet()) {
Object val = entry.getValue();
if (val instanceof JSObject) {
val = convertJSObject(((JSObject) val));
if (jsIsObject(val)) {
val = convertJSObject(val);
}
newMap.put(entry.getKey(), val);
}

35
jetlinks-standalone/pom.xml
View File

@ -52,6 +52,25 @@
</plugins>
</build>
<profiles>
<profile>
<id>java11</id>
<properties>
<maven.compiler.source>11</maven.compiler.source>
<maven.compiler.target>11</maven.compiler.target>
<java.version>11</java.version>
<project.build.jdk>11</project.build.jdk>
</properties>
<dependencies>
<dependency>
<groupId>org.openjdk.nashorn</groupId>
<artifactId>nashorn-core</artifactId>
<version>15.4</version>
</dependency>
</dependencies>
</profile>
</profiles>
<dependencies>
<dependency>
@ -152,20 +171,20 @@
<artifactId>spring-boot-starter-data-redis</artifactId>
</dependency>
<!-- <dependency>-->
<!-- <groupId>org.springframework.boot</groupId>-->
<!-- <artifactId>spring-boot-starter-actuator</artifactId>-->
<!-- </dependency>-->
<!-- <dependency>-->
<!-- <groupId>org.springframework.boot</groupId>-->
<!-- <artifactId>spring-boot-starter-actuator</artifactId>-->
<!-- </dependency>-->
<dependency>
<groupId>org.springframework.boot</groupId>
<artifactId>spring-boot-starter-data-r2dbc</artifactId>
</dependency>
<!-- <dependency>-->
<!-- <groupId>dev.miku</groupId>-->
<!-- <artifactId>r2dbc-mysql</artifactId>-->
<!-- </dependency>-->
<!-- <dependency>-->
<!-- <groupId>dev.miku</groupId>-->
<!-- <artifactId>r2dbc-mysql</artifactId>-->
<!-- </dependency>-->
<!-- https://github.com/mirromutth/r2dbc-mysql/issues/251 -->
<dependency>