Add test to test raylet client connection when raylet crashes. (#3518)

This commit is contained in:
Yuhong Guo
2018-12-14 15:40:50 +08:00
committed by Robert Nishihara
parent e7b51cbd1b
commit a4abe6c0fe
5 changed files with 21 additions and 7 deletions
@@ -6,7 +6,6 @@ import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import org.ray.api.RayObject;
import org.ray.api.WaitResult;
import org.ray.api.exception.RayException;
import org.ray.api.id.UniqueId;
import org.ray.runtime.RayDevRuntime;
import org.ray.runtime.objectstore.MockObjectStore;
@@ -68,7 +67,7 @@ public class MockRayletClient implements RayletClient {
@Override
public void fetchOrReconstruct(List<UniqueId> objectIds, boolean fetchOnly,
UniqueId currentTaskId) throws RayException {
UniqueId currentTaskId) {
}
@@ -3,7 +3,6 @@ package org.ray.runtime.raylet;
import java.util.List;
import org.ray.api.RayObject;
import org.ray.api.WaitResult;
import org.ray.api.exception.RayException;
import org.ray.api.id.UniqueId;
import org.ray.runtime.task.TaskSpec;
@@ -16,8 +15,7 @@ public interface RayletClient {
TaskSpec getTask();
void fetchOrReconstruct(List<UniqueId> objectIds, boolean fetchOnly, UniqueId currentTaskId)
throws RayException;
void fetchOrReconstruct(List<UniqueId> objectIds, boolean fetchOnly, UniqueId currentTaskId);
void notifyUnblocked(UniqueId currentTaskId);
@@ -90,7 +90,7 @@ public class RayletClientImpl implements RayletClient {
@Override
public void fetchOrReconstruct(List<UniqueId> objectIds, boolean fetchOnly,
UniqueId currentTaskId) throws RayException {
UniqueId currentTaskId) {
if (RayLog.core.isDebugEnabled()) {
RayLog.core.debug("Blocked on objects for task {}, object IDs are {}",
UniqueIdUtil.computeTaskId(objectIds.get(0)), objectIds);