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

Fix Java exception serialization (#10568)

Browse Source
This commit is contained in:
fyrestone
2020-09-05 11:06:14 +08:00
committed by GitHub
parent 523705ac0f
commit 77efad8616
4 changed files with 46 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
java/test/src/main/java/io/ray/test/FailureTest.java
+21
View File
@@ -4,9 +4,11 @@ import io.ray.api.ActorHandle;
import io.ray.api.ObjectRef;
import io.ray.api.Ray;
import io.ray.api.function.RayFunc0;
import io.ray.api.id.ObjectId;
import io.ray.runtime.exception.RayActorException;
import io.ray.runtime.exception.RayTaskException;
import io.ray.runtime.exception.RayWorkerException;
import io.ray.runtime.exception.UnreconstructableException;
import java.time.Duration;
import java.time.Instant;
import java.util.Arrays;
@@ -145,5 +147,24 @@ public class FailureTest extends BaseTest {
}
}
}
public void testExceptionSerialization() {
RayTaskException ex1 = Assert.expectThrows(RayTaskException.class, () -> {
Ray.put(new RayTaskException("xxx", new RayActorException())).get();
});
Assert.assertEquals(ex1.getCause().getClass(), RayActorException.class);
RayTaskException ex2 = Assert.expectThrows(RayTaskException.class, () -> {
Ray.put(new RayTaskException("xxx", new RayWorkerException())).get();
});
Assert.assertEquals(ex2.getCause().getClass(), RayWorkerException.class);
ObjectId objectId = ObjectId.fromRandom();
RayTaskException ex3 = Assert.expectThrows(RayTaskException.class, () -> {
Ray.put(new RayTaskException("xxx", new UnreconstructableException(objectId)))
.get();
});
Assert.assertEquals(ex3.getCause().getClass(), UnreconstructableException.class);
Assert.assertEquals(((UnreconstructableException) ex3.getCause()).objectId, objectId);
}
}