@Override public boolean isComplete() { return totalResults > 0 ? queryBatch.getJobResultsSoFar() >= totalResults : false; }
@Override public String getProgressAsString() { String text; if (totalResults > 0) { text = String.format("Progress: %d of %d; time %fs", queryBatch.getJobResultsSoFar(), totalResults, timeSoFarInSeconds); } else { text = String.format("Progress: %d results so far; time %fs", queryBatch.getJobResultsSoFar(), timeSoFarInSeconds); } if (timeSoFarInSeconds > 0) { double rate = queryBatch.getJobResultsSoFar() / timeSoFarInSeconds; BigDecimal bd = new BigDecimal(rate); rate = bd.round(new MathContext(5)).doubleValue(); return text + "; " + rate + " records/s"; } return text; }
@Override public void processEvent(QueryBatch queryBatch) { String message = String.format("Processed batch number [%d]; job results so far: [%d]", queryBatch.getJobBatchNumber(), queryBatch.getJobResultsSoFar()); if (useLogger) { if (logger.isInfoEnabled()) { logger.info(message); } } else { System.out.println(message); } }
public void processEvent(QueryBatch batch) { final long jobResultsSoFar = batch.getJobResultsSoFar(); final long newResultsSoFar = this.resultsSoFar.updateAndGet(operand -> jobResultsSoFar > operand ? jobResultsSoFar : operand
@Override public void processFailure(QueryBatch batch, Throwable throwable) { boolean isHostUnavailableException = processException(batch.getBatcher(), throwable, batch.getClient().getHost()); if ( isHostUnavailableException == true ) { try { logger.warn("Retrying failed listener batch: {}, results so far: {}, uris: {}", batch.getJobBatchNumber(), batch.getJobResultsSoFar(), Arrays.toString(batch.getItems())); batch.getBatcher().retryListener(batch, queryBatchListener); } catch (RuntimeException e) { logger.error("Exception during listener retry", e); processFailure(batch, e); } } } }
.withTimestamp( batch.getTimestamp() ) .withJobBatchNumber( batch.getJobBatchNumber() ) .withJobResultsSoFar( batch.getJobResultsSoFar() ) .withForestBatchNumber( batch.getForestBatchNumber() ) .withForestResultsSoFar( batch.getForestResultsSoFar() )
@Test public void batchesInOrder() throws Exception { listener = new ProgressListener(4, consumer); listener.initializeListener(null); // Ensure that the total amount of time will be greater than zero for when it's tested below Thread.sleep(100); listener.processEvent(new FakeQueryBatch(2)); assertFalse(consumer.lastProgressUpdate.isComplete()); listener.processEvent(new FakeQueryBatch(4)); assertTrue(consumer.lastProgressUpdate.isComplete()); assertTrue(consumer.texts.get(0).startsWith("Progress: 2 of 4; time ")); assertTrue(consumer.texts.get(0).contains("records/s")); assertTrue(consumer.texts.get(1).startsWith("Progress: 4 of 4; time ")); assertTrue(consumer.texts.get(1).contains("records/s")); assertEquals(2, consumer.texts.size()); // Verify the values on the ProgressUpdate are correct assertEquals(4, consumer.lastProgressUpdate.getQueryBatch().getJobResultsSoFar()); assertEquals(4, consumer.lastProgressUpdate.getTotalResults()); assertTrue(consumer.lastProgressUpdate.getStartTime() > 0); assertTrue(consumer.lastProgressUpdate.getTimeSoFarInSeconds() > 0); }
@Override public void retryListener(QueryBatch batch, QueryBatchListener queryBatchListener) { // We get the batch and modify the client alone in order to make use // of the new forest client in case if the original host is unavailable. DatabaseClient client = null; Forest[] forests = batch.getBatcher().getForestConfig().listForests(); for(Forest forest : forests) { if(forest.equals(batch.getForest())) client = getMoveMgr().getForestClient(forest); } QueryBatchImpl retryBatch = new QueryBatchImpl() .withClient( client ) .withBatcher( batch.getBatcher() ) .withTimestamp( batch.getTimestamp() ) .withServerTimestamp( batch.getServerTimestamp() ) .withItems( batch.getItems() ) .withJobTicket( batch.getJobTicket() ) .withJobBatchNumber( batch.getJobBatchNumber() ) .withJobResultsSoFar( batch.getJobResultsSoFar() ) .withForestBatchNumber( batch.getForestBatchNumber() ) .withForestResultsSoFar( batch.getForestResultsSoFar() ) .withForest( batch.getForest() ) .withJobTicket( batch.getJobTicket() ); queryBatchListener.processEvent(retryBatch); }