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

[Java] Throw exception if Ray.init() is not called and users try to access ray API (#10497)

Browse Source
This commit is contained in:
chaokunyang
2020-09-05 10:09:19 +08:00
committed by GitHub
parent 8ee7c182f5
commit 7ee0fdba3d
6 changed files with 35 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
streaming/java/streaming-api/src/main/java/io/ray/streaming/api/context/ClusterStarter.java
+1 -1
View File
@@ -23,7 +23,7 @@ class ClusterStarter {
private static final String RAYLET_SOCKET_NAME = "/tmp/ray/raylet_socket";
static synchronized void startCluster(boolean isCrossLanguage, boolean isLocal) {
Preconditions.checkArgument(Ray.internal() == null);
Preconditions.checkArgument(!Ray.isInitialized());
RayConfig.reset();
if (!isLocal) {
System.setProperty("ray.raylet.config.num_workers_per_process_java", "1");
streaming/java/streaming-api/src/main/java/io/ray/streaming/api/context/StreamingContext.java
+2 -2
View File
@@ -63,7 +63,7 @@ public class StreamingContext implements Serializable {
jobGraph.printJobGraph();
LOG.info("JobGraph digraph\n{}", jobGraph.generateDigraph());
if (Ray.internal() == null) {
if (!Ray.isInitialized()) {
if (Config.MEMORY_CHANNEL.equalsIgnoreCase(jobConfig.get(Config.CHANNEL_TYPE))) {
Preconditions.checkArgument(!jobGraph.isCrossLanguageGraph());
ClusterStarter.startCluster(false, true);
@@ -102,7 +102,7 @@ public class StreamingContext implements Serializable {
}
public void stop() {
if (Ray.internal() != null) {
if (Ray.isInitialized()) {
ClusterStarter.stopCluster(jobGraph.isCrossLanguageGraph());
}
}