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

fix named actor single process mode bug (#9652)

Browse Source
This commit is contained in:
fangfengbin
2020-07-23 20:28:50 +08:00
committed by GitHub
parent 5ab17e0dd8
commit 01d6edae9c
2 changed files with 29 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/runtime/src/main/java/io/ray/runtime/task/LocalModeTaskSubmitter.java
+8 -6
View File
@@ -174,9 +174,11 @@ public class LocalModeTaskSubmitter implements TaskSubmitter {
= new LocalModeActorHandle(actorId, getReturnIds(taskSpec).get(0));
actorHandles.put(actorId, actorHandle.copy());
if (StringUtils.isNotBlank(options.name)) {
Preconditions.checkArgument(!namedActors.containsKey(options.name),
String.format("Actor of name %s exists", options.name));
namedActors.put(options.name, actorHandle);
String fullName = options.global ? options.name :
String.format("%s-%s", Ray.getRuntimeContext().getCurrentJobId(), options.name);
Preconditions.checkArgument(!namedActors.containsKey(fullName),
String.format("Actor of name %s exists", fullName));
namedActors.put(fullName, actorHandle);
}
return actorHandle;
}
@@ -215,11 +217,11 @@ public class LocalModeTaskSubmitter implements TaskSubmitter {
public Optional<BaseActorHandle> getActor(String name, boolean global) {
String fullName = global ? name :
String.format("%s-%s", Ray.getRuntimeContext().getCurrentJobId(), name);
if (namedActors.containsKey(fullName)) {
return Optional.of(namedActors.get(fullName));
} else {
ActorHandle actorHandle = namedActors.get(fullName);
if (null == actorHandle) {
return Optional.empty();
}
return Optional.of(actorHandle);
}
public void shutdown() {