[Java] Add fetchLocal parameter in Ray.wait() (#13604)

This commit is contained in:
Kai Yang
2021-01-22 17:55:00 +08:00
committed by GitHub
parent 00c14ce4a4
commit 90f1e408de
14 changed files with 78 additions and 87 deletions
@@ -25,7 +25,7 @@ public class PlasmaFreeTest extends BaseTest {
() ->
!TestUtils.getRuntime()
.getObjectStore()
.wait(ImmutableList.of(((ObjectRefImpl<String>) helloId).getId()), 1, 0)
.wait(ImmutableList.of(((ObjectRefImpl<String>) helloId).getId()), 1, 0, true)
.get(0),
50);
if (TestUtils.isSingleProcessMode()) {
@@ -119,7 +119,7 @@ public class ReferenceCountingTest extends BaseTest {
TestUtils.getRuntime().getObjectStore().getRaw(ImmutableList.of(objectId), Long.MAX_VALUE);
} else {
List<Boolean> result =
TestUtils.getRuntime().getObjectStore().wait(ImmutableList.of(objectId), 1, 100);
TestUtils.getRuntime().getObjectStore().wait(ImmutableList.of(objectId), 1, 100, true);
Assert.assertFalse(result.get(0));
}
}