write transformer in javascript

src/main/scala/spark.scala

@@ -7,7 +7,6 @@ import java.util.concurrent.atomic.AtomicInteger
 import java.util.{Map => JMap}
 import javax.script.{Compilable, CompiledScript => JCompiledScript, ScriptEngineManager}
 
-import jdk.nashorn.api.scripting.ScriptObjectMirror
 import org.apache.spark.api.java.function.{FlatMapFunction, MapFunction}
 import org.apache.spark.sql.{Dataset, Encoder, Encoders, SparkSession}
 
@@ -217,32 +216,12 @@ class JavaScriptEngine extends ScriptEngine {
         bindings.asInstanceOf[JMap[String, Any]].putAll(args);
 
         val value = compiled.eval(bindings);
-        if (value.isInstanceOf[java.lang.Integer]
-          || value.isInstanceOf[java.lang.String]
-          || value.isInstanceOf[java.lang.Double]
-          || value.isInstanceOf[java.lang.Boolean]) {
-          value;
-        }
-
-        else if (value.isInstanceOf[ScriptObjectMirror]) {
-          if (value.asInstanceOf[ScriptObjectMirror].isArray) {
-            value.asInstanceOf[ScriptObjectMirror].values().asInstanceOf[java.util.Collection[_]]
-          }
-        }
-
-        else {
-          throw new UnrecognizedValueException(value);
-        }
+        value;
       }
     }
   }
 }
 
-class UnrecognizedValueException(value: Any)
-  extends RuntimeException(s"unrecognized value: $value") {
-
-}
-
 object Tool {
   def ops(s: String) = new StringOps(s);
 }

src/test/scala/FlowTest.scala

@@ -52,16 +52,10 @@ class FlowTest {
     val fg = new SparkETLProcess();
     val s1 = fg.loadStream(TextFile("./out/honglou.txt", "text"));
     val s2 = fg.transform(s1, DoMap(
-      """
-      function(s){
-        return s.replaceAll("[\\x00-\\xff]|，|。|：|．|“|”|？|！|　", "");
-      }""", classOf[String]));
+      SparkETLTest.SCRIPT_1, classOf[String]));
 
     val s3 = fg.transform(s2, DoFlatMap(
-      """
-      function(s){
-        return s.zip(s.drop(1)).map(t => "" + t._1 + t._2);
-      }""", classOf[String]));
+      SparkETLTest.SCRIPT_2, classOf[String]));
 
     fg.writeStream(s3, TextFile("./out/wordcount", "json"));
 

src/test/scala/ScriptEngineTest.scala

@@ -17,14 +17,8 @@ class ScriptEngineTest {
     Assert.assertEquals("BLUEJOE", s);
     Assert.assertEquals(classOf[String], s.getClass);
 
-    engine.eval("function arr(){return [1,2,3];}");
-
-    val s2 = engine.asInstanceOf[Invocable].invokeFunction("arr");
-    val som = s2.asInstanceOf[ScriptObjectMirror];
-    Assert.assertEquals(true, som.isArray);
-    println(som.values());
-    println(som.values().getClass);
-    println(som.values().asInstanceOf[util.Collection[_]].iterator());
+    val s2 = engine.eval("(function (){return java.util.Arrays.asList([1,2,3]);})();");
+    println(s2);
 
     println(engine.eval("1;").getClass);
     println(engine.eval("'abc';").getClass);
@@ -42,6 +36,15 @@ class ScriptEngineTest {
     println(m);
   }
 
+  @Test
+  def testTs(): Unit = {
+    val engine = new ScriptEngineManager().getEngineByName("typescript");
+    val s = engine.eval("""["1","2","3"]""");
+    println(s);
+    val m = ScriptUtils.convert(s, classOf[Array[String]]);
+    println(m);
+  }
+
   @Test
   def testScala(): Unit = {
     val engine = new ScriptEngineManager().getEngineByName("scala");

src/test/scala/SparkETLTest.scala

@@ -14,23 +14,10 @@ class SparkETLTest {
     val fg = new SparkETLProcess();
     val s1 = fg.loadStream(TextFile("./out/honglou.txt", "text"));
     val s2 = fg.transform(s1, DoMap(
-      """
-      function(s){
-        return s.replaceAll("[\\x00-\\xff]|，|。|：|．|“|”|？|！|　", "");
-      }""", classOf[String]));
+      SparkETLTest.SCRIPT_1, classOf[String]));
 
     val s3 = fg.transform(s2, DoFlatMap(
-      """
-	function (s) {
-		var arr = Array();
-		var len = s.length;
-		for (var i = 0; i < s.length - 1; i++) {
-			arr.push(s.substring(i, i + 2));
-		}
-
-		return arr;
-	}
-      """, classOf[String]));
+      SparkETLTest.SCRIPT_2, classOf[String]));
 
     fg.writeStream(s3, TextFile("./out/wordcount", "json"));
 
@@ -38,3 +25,23 @@ class SparkETLTest {
     fg.run(new SparkProcessContext());
   }
 }
+
+object SparkETLTest {
+  val SCRIPT_1 =
+    """
+      function(s){
+        return s.replaceAll("[\\x00-\\xff]|，|。|：|．|“|”|？|！|　", "");
+      }""";
+  val SCRIPT_2 =
+    """
+	function (s) {
+		var arr = Array();
+		var len = s.length;
+		for (var i = 0; i < s.length - 1; i++) {
+			arr.push(s.substring(i, i + 2));
+		}
+
+		return java.util.Arrays.asList(arr);
+	}
+    """;
+}