/
Scripting.java
69 lines (57 loc) · 2.68 KB
/
Scripting.java
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
package com.zendesk.maxwell.scripting;
import javax.script.ScriptEngine;
import javax.script.ScriptEngineManager;
import javax.script.ScriptException;
import java.io.IOException;
import java.nio.file.Files;
import java.nio.file.Paths;
import com.zendesk.maxwell.row.HeartbeatRowMap;
import com.zendesk.maxwell.row.RowMap;
import com.zendesk.maxwell.schema.ddl.DDLMap;
import jdk.nashorn.api.scripting.ScriptObjectMirror;
import jdk.nashorn.api.scripting.ScriptUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class Scripting {
static final Logger LOGGER = LoggerFactory.getLogger(Scripting.class);
private final ScriptObjectMirror processRowFunc, processHeartbeatFunc, processDDLFunc;
private ScriptObjectMirror getFunc(ScriptEngine engine, String fName, String filename) {
ScriptObjectMirror f = (ScriptObjectMirror) engine.get(fName);
if ( f == null )
return null;
else if ( !f.isFunction() ) {
throw new RuntimeException("Expected " + fName + " to be a function!");
} else {
LOGGER.info("using function " + fName + " from " + filename);
}
return f;
}
public Scripting(String filename) throws IOException, ScriptException, NoSuchMethodException {
ScriptEngineManager manager = new ScriptEngineManager();
ScriptEngine engine = manager.getEngineByName("nashorn");
String externJS = new String(Files.readAllBytes(Paths.get(filename)));
engine.put("logger", LOGGER);
engine.eval(externJS);
processRowFunc = getFunc(engine, "process_row", filename);
processHeartbeatFunc = getFunc(engine, "process_heartbeat", filename);
processDDLFunc = getFunc(engine, "process_ddl", filename);
if ( processRowFunc == null && processHeartbeatFunc == null && processDDLFunc == null )
LOGGER.warn("expected " + filename + " to define at least one of: process_row,process_heartbeat,process_ddl");
}
public void invoke(RowMap row) {
if ( row instanceof HeartbeatRowMap && processHeartbeatFunc != null )
processHeartbeatFunc.call(null, new WrappedHeartbeatMap((HeartbeatRowMap) row));
else if ( row instanceof DDLMap && processDDLFunc != null )
processDDLFunc.call(null, new WrappedDDLMap((DDLMap) row));
else if ( row instanceof RowMap && processRowFunc != null )
processRowFunc.call(null, new WrappedRowMap(row));
}
private static ThreadLocal<ScriptEngine> stringifyEngineThreadLocal = ThreadLocal.withInitial(() -> {
ScriptEngineManager manager = new ScriptEngineManager();
return manager.getEngineByName("nashorn");
});
public static String stringify(ScriptObjectMirror mirror) throws ScriptException {
ScriptObjectMirror json = (ScriptObjectMirror) stringifyEngineThreadLocal.get().eval("JSON");
return (String) json.callMember("stringify", mirror);
}
}