@Test public void testGetLeaseTimesOut() throws Exception { TimeoutException te = setRuleForDependencyTimeout(); when(dynamoDbClient.getItem(any(GetItemRequest.class))).thenReturn(mockGetItemFuture); when(mockGetItemFuture.get(anyLong(), any())).thenThrow(te); when(leaseSerializer.getDynamoHashKey(anyString())).thenReturn(Collections.emptyMap()); verifyCancel(mockGetItemFuture, () -> leaseRefresher.getLease("test")); }
@Override public Lease getLease(@NonNull final String leaseKey) throws DependencyException, InvalidStateException, ProvisionedThroughputException { log.debug("Getting lease with key {}", leaseKey); GetItemRequest request = GetItemRequest.builder().tableName(table).key(serializer.getDynamoHashKey(leaseKey)) .consistentRead(consistentReads).build(); final AWSExceptionManager exceptionManager = createExceptionManager(); try { try { GetItemResponse result = FutureUtils.resolveOrCancelFuture(dynamoDBClient.getItem(request), dynamoDbRequestTimeout); Map<String, AttributeValue> dynamoRecord = result.item(); if (CollectionUtils.isNullOrEmpty(dynamoRecord)) { log.debug("No lease found with key {}, returning null.", leaseKey); return null; } else { final Lease lease = serializer.fromDynamoRecord(dynamoRecord); log.debug("Got lease {}", lease); return lease; } } catch (ExecutionException e) { throw exceptionManager.apply(e.getCause()); } catch (InterruptedException e) { throw new DependencyException(e); } } catch (DynamoDbException | TimeoutException e) { throw convertAndRethrowExceptions("get", leaseKey, e); } }
DynamoDBLeaseRefresher implements LeaseRefresher { @Override public Lease getLease(@NonNull final String leaseKey) throws DependencyException, InvalidStateException, ProvisionedThroughputException { log.debug("Getting lease with key {}", leaseKey); GetItemRequest request = GetItemRequest.builder().tableName(table).key(serializer.getDynamoHashKey(leaseKey)) .consistentRead(consistentReads).build(); final AWSExceptionManager exceptionManager = createExceptionManager(); try { try { GetItemResponse result = FutureUtils.resolveOrCancelFuture(dynamoDBClient.getItem(request), dynamoDbRequestTimeout); Map<String, AttributeValue> dynamoRecord = result.item(); if (CollectionUtils.isNullOrEmpty(dynamoRecord)) { log.debug("No lease found with key {}, returning null.", leaseKey); return null; } else { final Lease lease = serializer.fromDynamoRecord(dynamoRecord); log.debug("Got lease {}", lease); return lease; } } catch (ExecutionException e) { throw exceptionManager.apply(e.getCause()); } catch (InterruptedException e) { throw new DependencyException(e); } } catch (DynamoDbException | TimeoutException e) { throw convertAndRethrowExceptions("get", leaseKey, e); } } }
DynamoDBLeaseRefresher implements LeaseRefresher { @Override public Lease getLease(@NonNull final String leaseKey) throws DependencyException, InvalidStateException, ProvisionedThroughputException { log.debug("Getting lease with key {}", leaseKey); GetItemRequest request = GetItemRequest.builder().tableName(table).key(serializer.getDynamoHashKey(leaseKey)) .consistentRead(consistentReads).build(); final AWSExceptionManager exceptionManager = createExceptionManager(); try { try { GetItemResponse result = FutureUtils.resolveOrCancelFuture(dynamoDBClient.getItem(request), dynamoDbRequestTimeout); Map<String, AttributeValue> dynamoRecord = result.item(); if (CollectionUtils.isNullOrEmpty(dynamoRecord)) { log.debug("No lease found with key {}, returning null.", leaseKey); return null; } else { final Lease lease = serializer.fromDynamoRecord(dynamoRecord); log.debug("Got lease {}", lease); return lease; } } catch (ExecutionException e) { throw exceptionManager.apply(e.getCause()); } catch (InterruptedException e) { throw new DependencyException(e); } } catch (DynamoDbException | TimeoutException e) { throw convertAndRethrowExceptions("get", leaseKey, e); } } @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
            final LeaseSerializer serializer, final boolean consistentReads); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout);  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout,
                                  final BillingMode billingMode); }
DynamoDBLeaseRefresher implements LeaseRefresher { @Override public Lease getLease(@NonNull final String leaseKey) throws DependencyException, InvalidStateException, ProvisionedThroughputException { log.debug("Getting lease with key {}", leaseKey); GetItemRequest request = GetItemRequest.builder().tableName(table).key(serializer.getDynamoHashKey(leaseKey)) .consistentRead(consistentReads).build(); final AWSExceptionManager exceptionManager = createExceptionManager(); try { try { GetItemResponse result = FutureUtils.resolveOrCancelFuture(dynamoDBClient.getItem(request), dynamoDbRequestTimeout); Map<String, AttributeValue> dynamoRecord = result.item(); if (CollectionUtils.isNullOrEmpty(dynamoRecord)) { log.debug("No lease found with key {}, returning null.", leaseKey); return null; } else { final Lease lease = serializer.fromDynamoRecord(dynamoRecord); log.debug("Got lease {}", lease); return lease; } } catch (ExecutionException e) { throw exceptionManager.apply(e.getCause()); } catch (InterruptedException e) { throw new DependencyException(e); } } catch (DynamoDbException | TimeoutException e) { throw convertAndRethrowExceptions("get", leaseKey, e); } } @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
            final LeaseSerializer serializer, final boolean consistentReads); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout);  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout,
                                  final BillingMode billingMode); @Override boolean createLeaseTableIfNotExists(@NonNull final Long readCapacity, @NonNull final Long writeCapacity); @Override boolean leaseTableExists(); @Override boolean waitUntilLeaseTableExists(long secondsBetweenPolls, long timeoutSeconds); @Override List<Lease> listLeasesForStream(StreamIdentifier streamIdentifier); @Override List<Lease> listLeases(); @Override boolean isLeaseTableEmpty(); @Override boolean createLeaseIfNotExists(@NonNull final Lease lease); @Override Lease getLease(@NonNull final String leaseKey); @Override boolean renewLease(@NonNull final Lease lease); @Override boolean takeLease(@NonNull final Lease lease, @NonNull final String owner); @Override boolean evictLease(@NonNull final Lease lease); void deleteAll(); @Override void deleteLease(@NonNull final Lease lease); @Override boolean updateLease(@NonNull final Lease lease); @Override void updateLeaseWithMetaInfo(Lease lease, UpdateField updateField); @Override ExtendedSequenceNumber getCheckpoint(String leaseKey); }
DynamoDBLeaseRefresher implements LeaseRefresher { @Override public Lease getLease(@NonNull final String leaseKey) throws DependencyException, InvalidStateException, ProvisionedThroughputException { log.debug("Getting lease with key {}", leaseKey); GetItemRequest request = GetItemRequest.builder().tableName(table).key(serializer.getDynamoHashKey(leaseKey)) .consistentRead(consistentReads).build(); final AWSExceptionManager exceptionManager = createExceptionManager(); try { try { GetItemResponse result = FutureUtils.resolveOrCancelFuture(dynamoDBClient.getItem(request), dynamoDbRequestTimeout); Map<String, AttributeValue> dynamoRecord = result.item(); if (CollectionUtils.isNullOrEmpty(dynamoRecord)) { log.debug("No lease found with key {}, returning null.", leaseKey); return null; } else { final Lease lease = serializer.fromDynamoRecord(dynamoRecord); log.debug("Got lease {}", lease); return lease; } } catch (ExecutionException e) { throw exceptionManager.apply(e.getCause()); } catch (InterruptedException e) { throw new DependencyException(e); } } catch (DynamoDbException | TimeoutException e) { throw convertAndRethrowExceptions("get", leaseKey, e); } } @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
            final LeaseSerializer serializer, final boolean consistentReads); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback); @Deprecated  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout);  DynamoDBLeaseRefresher(final String table, final DynamoDbAsyncClient dynamoDBClient,
                                  final LeaseSerializer serializer, final boolean consistentReads,
                                  @NonNull final TableCreatorCallback tableCreatorCallback, Duration dynamoDbRequestTimeout,
                                  final BillingMode billingMode); @Override boolean createLeaseTableIfNotExists(@NonNull final Long readCapacity, @NonNull final Long writeCapacity); @Override boolean leaseTableExists(); @Override boolean waitUntilLeaseTableExists(long secondsBetweenPolls, long timeoutSeconds); @Override List<Lease> listLeasesForStream(StreamIdentifier streamIdentifier); @Override List<Lease> listLeases(); @Override boolean isLeaseTableEmpty(); @Override boolean createLeaseIfNotExists(@NonNull final Lease lease); @Override Lease getLease(@NonNull final String leaseKey); @Override boolean renewLease(@NonNull final Lease lease); @Override boolean takeLease(@NonNull final Lease lease, @NonNull final String owner); @Override boolean evictLease(@NonNull final Lease lease); void deleteAll(); @Override void deleteLease(@NonNull final Lease lease); @Override boolean updateLease(@NonNull final Lease lease); @Override void updateLeaseWithMetaInfo(Lease lease, UpdateField updateField); @Override ExtendedSequenceNumber getCheckpoint(String leaseKey);  }