private AsyncResult<Void> awaitLocks(UUID requestId, Set<LockDescriptor> lockDescriptors, TimeLimit timeout) { OrderedLocks orderedLocks = locks.getAll(lockDescriptors); return lockAcquirer.waitForLocks(requestId, orderedLocks, timeout); }
private AsyncResult<Void> waitFor(AsyncLock... locks) { return lockAcquirer.waitForLocks(REQUEST_ID, OrderedLocks.fromOrderedList(ImmutableList.copyOf(locks)), TIMEOUT); }
@Test public void doesNotWaitForDuplicateRequests() { Set<LockDescriptor> descriptors = descriptors(LOCK_A); lockService.waitForLocks(REQUEST_ID, descriptors, DEADLINE); lockService.waitForLocks(REQUEST_ID, descriptors, DEADLINE); verify(acquirer, times(1)).waitForLocks(any(), any(), any()); verifyNoMoreInteractions(acquirer); }
@Before public void before() { when(acquirer.acquireLocks(any(), any(), any())).thenReturn(new AsyncResult<>()); when(acquirer.waitForLocks(any(), any(), any())).thenReturn(new AsyncResult<>()); when(locks.getAll(any())).thenReturn(OrderedLocks.fromSingleLock(newLock())); when(immutableTimestampTracker.getImmutableTimestamp()).thenReturn(Optional.empty()); when(immutableTimestampTracker.getLockFor(anyLong())).thenReturn(newLock()); }
@Test public void passesOrderedLocksToAcquirerWhenWaitingForLocks() { OrderedLocks expected = orderedLocks(newLock(), newLock()); Set<LockDescriptor> descriptors = descriptors(LOCK_A, LOCK_B); when(locks.getAll(descriptors)).thenReturn(expected); lockService.waitForLocks(REQUEST_ID, descriptors, DEADLINE); verify(acquirer).waitForLocks(REQUEST_ID, expected, DEADLINE); }
private AsyncResult<Void> awaitLocks(UUID requestId, Set<LockDescriptor> lockDescriptors, TimeLimit timeout) { OrderedLocks orderedLocks = locks.getAll(lockDescriptors); return lockAcquirer.waitForLocks(requestId, orderedLocks, timeout); }