private TimelockService getFreshInstrumentedTimelockService() { timelockService = mock(TimelockService.class); TimelockService instrumentedTimelockService = new InstrumentedTimelockService( timelockService, metricRegistry); return instrumentedTimelockService; }
@Override public void tryUnlock(Set<LockToken> tokens) { executeWithRecord(() -> { timelockService.tryUnlock(tokens); return null; }); }
@Override public StartAtlasDbTransactionResponse startAtlasDbTransaction(IdentifiedTimeLockRequest request) { return executeWithRecord(() -> timelockService.startAtlasDbTransaction(request)); }
private static LockAndTimestampServices withMetrics( MetricsManager metricsManager, LockAndTimestampServices lockAndTimestampServices) { TimelockService timelockServiceWithBatching = lockAndTimestampServices.timelock(); TimelockService instrumentedTimelockService = new InstrumentedTimelockService( timelockServiceWithBatching, metricsManager.getRegistry()); return ImmutableLockAndTimestampServices.builder() .from(lockAndTimestampServices) .timestamp(new TimelockTimestampServiceAdapter(instrumentedTimelockService)) .timelock(instrumentedTimelockService) .build(); }
@Override public long getFreshTimestamp() { return executeWithRecord(() -> timelockService.getFreshTimestamp()); }
private static LockAndTimestampServices withMetrics( MetricsManager metricsManager, LockAndTimestampServices lockAndTimestampServices) { TimelockService timelockServiceWithBatching = lockAndTimestampServices.timelock(); TimelockService instrumentedTimelockService = new InstrumentedTimelockService( timelockServiceWithBatching, metricsManager.getRegistry()); return ImmutableLockAndTimestampServices.builder() .from(lockAndTimestampServices) .timestamp(new TimelockTimestampServiceAdapter(instrumentedTimelockService)) .timelock(instrumentedTimelockService) .build(); }
@Override public LockResponse lock(LockRequest request) { return executeWithRecord(() -> timelockService.lock(request)); }
@Override public WaitForLocksResponse waitForLocks(WaitForLocksRequest request) { return executeWithRecord(() -> timelockService.waitForLocks(request)); }
@Override public Set<LockToken> unlock(Set<LockToken> tokens) { return executeWithRecord(() -> timelockService.unlock(tokens)); }
@Override public LockImmutableTimestampResponse lockImmutableTimestamp( IdentifiedTimeLockRequest request) { return executeWithRecord(() -> timelockService.lockImmutableTimestamp(request)); }
@Override public Set<LockToken> refreshLockLeases( Set<LockToken> tokens) { return executeWithRecord(() -> timelockService.refreshLockLeases(tokens)); }
@Override public StartIdentifiedAtlasDbTransactionResponse startIdentifiedAtlasDbTransaction( StartIdentifiedAtlasDbTransactionRequest request) { return executeWithRecord(() -> timelockService.startIdentifiedAtlasDbTransaction(request)); }
@Override public TimestampRange getFreshTimestamps(int numTimestampsRequested) { return executeWithRecord(() -> timelockService.getFreshTimestamps(numTimestampsRequested)); }
@Override public long getImmutableTimestamp() { return executeWithRecord(() -> timelockService.getImmutableTimestamp()); }
@Override public long currentTimeMillis() { return executeWithRecord(() -> timelockService.currentTimeMillis()); }
@Override public long getFreshTimestamp() { return executeWithRecord(() -> timelockService.getFreshTimestamp()); }
@Override public LockImmutableTimestampResponse lockImmutableTimestamp( IdentifiedTimeLockRequest request) { return executeWithRecord(() -> timelockService.lockImmutableTimestamp(request)); }
@Override public StartIdentifiedAtlasDbTransactionResponse startIdentifiedAtlasDbTransaction( StartIdentifiedAtlasDbTransactionRequest request) { return executeWithRecord(() -> timelockService.startIdentifiedAtlasDbTransaction(request)); }
@Override public Set<LockToken> refreshLockLeases( Set<LockToken> tokens) { return executeWithRecord(() -> timelockService.refreshLockLeases(tokens)); }
@Override public void tryUnlock(Set<LockToken> tokens) { executeWithRecord(() -> { timelockService.tryUnlock(tokens); return null; }); }