wassname/ray
1
0
Fork 0
mirror of https://github.com/wassname/ray.git synced 2026-06-28 04:07:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

[Java] Fix output parsing in RunManager (#12968)

Browse Source 
* Fix output parsing in RunManager

* change log level

Co-authored-by: 灵洵 <fengbin.ffb@antgroup.com>
This commit is contained in:
Kai Yang
2020-12-19 10:22:12 +08:00
committed by GitHub
parent 6ece291f35
commit ac5ea2c13d
3 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime/src/main/java/io/ray/runtime/runner/RunManager.java
+6 -2
View File
@@ -75,14 +75,18 @@ public class RunManager {
// address info of the local node.
String script = String.format("import ray;"
+ " print(ray._private.services.get_address_info_from_redis("
+ "'%s', '%s', redis_password='%s', no_warning=True))",
+ "'%s', '%s', redis_password='%s'))",
rayConfig.getRedisAddress(), rayConfig.nodeIp, rayConfig.redisPassword);
List<String> command = Arrays.asList("python", "-c", script);
String output = null;
try {
output = runCommand(command);
JsonObject addressInfo = new JsonParser().parse(output).getAsJsonObject();
// NOTE(kfstorm): We only parse the last line here in case there are some warning
// messages appear at the beginning.
String[] lines = output.split(System.lineSeparator());
String lastLine = lines[lines.length - 1];
JsonObject addressInfo = new JsonParser().parse(lastLine).getAsJsonObject();
rayConfig.rayletSocketName = addressInfo.get("raylet_socket_name").getAsString();
rayConfig.objectStoreSocketName = addressInfo.get("object_store_address").getAsString();
rayConfig.nodeManagerPort = addressInfo.get("node_manager_port").getAsInt();