Merge pull request byzer-org#793 from allwefantasy/ISSUE-792

fix excape problem in ps cluster warning
ilibx · Dec 17, 2018 · 14e09ae · 14e09ae
2 parents 20957f4 + df7b5ba
commit 14e09ae
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/streamingpro-mlsql/src/main/java/streaming/core/strategy/platform/SparkRuntime.scala b/streamingpro-mlsql/src/main/java/streaming/core/strategy/platform/SparkRuntime.scala
@@ -82,12 +82,12 @@ class SparkRuntime(_params: JMap[Any, Any]) extends StreamingRuntime with Platfo
            |${MLSQLConf.MLSQL_CLUSTER_PS_ENABLE.key} is enabled. Please make sure
            |you have the uber-jar of mlsql placed in
            |1. --jars
-           |2. --conf "spark.executor.extraClassPath=[your jar name in jars]" \
+           |2. --conf "spark.executor.extraClassPath=[your jar name in jars]"
            |
            |for exmaple:
            |
-           |--jars ./streamingpro-mlsql-spark_2.x-x.x.x-SNAPSHOT.jar \
-           |--conf "spark.executor.extraClassPath=streamingpro-mlsql-spark_2.x-x.x.x-SNAPSHOT.jar" \
+           |--jars ./streamingpro-mlsql-spark_2.x-x.x.x-SNAPSHOT.jar
+           |--conf "spark.executor.extraClassPath=streamingpro-mlsql-spark_2.x-x.x.x-SNAPSHOT.jar" 
            |
            |Otherwise the executor will
            |fail to start and the whole application will fails.

diff --git a/streamingpro-mlsql/src/main/java/streaming/rest/RestController.scala b/streamingpro-mlsql/src/main/java/streaming/rest/RestController.scala
@@ -477,8 +477,8 @@ class RestController extends ApplicationController {
   @At(path = Array("/test"), types = Array(GET, POST))
   def test = {
     val psDriverBackend = runtime.asInstanceOf[SparkRuntime].psDriverBackend
-    psDriverBackend.psDriverRpcEndpointRef.send(Message.TensorFlowModelClean("/tmp/ok"))
-    render("{}")
+    val res = psDriverBackend.psDriverRpcEndpointRef.ask[Boolean](Message.CopyModelToLocal(param("hdfs"), param("local")))
+    render(s"""{"msg":${res}""")
   }
 
   @At(path = Array("/instance/resource"), types = Array(GET, POST))