/** * Can be overridden by the subclass to prepare the QueryBatcher before the job is started. * * @param queryBatcher */ protected void prepareQueryBatcher(QueryBatcher queryBatcher) { super.prepareBatcher(queryBatcher); if (consistentSnapshot) { queryBatcher.withConsistentSnapshot(); } if (urisReadyListeners != null) { for (QueryBatchListener listener : urisReadyListeners) { queryBatcher.onUrisReady(listener); } } if (queryFailureListeners != null) { for (QueryFailureListener listener : queryFailureListeners) { queryBatcher.onQueryFailure(listener); } } }
public void testIteratorExceptions(List<String> uris, int expectedSuccesses, int expectedFailures) { QueryBatcher uriListBatcher = moveMgr.newQueryBatcher(uris.iterator()) .onUrisReady( batch -> { throw new InternalError(errorMessage); } ) .onQueryFailure( queryThrowable -> { throw new InternalError(errorMessage); } ); testExceptions(uriListBatcher, expectedSuccesses, expectedFailures); uriListBatcher = moveMgr.newQueryBatcher(uris.iterator()) .onUrisReady( batch -> { throw new RuntimeException(errorMessage); } ) .onQueryFailure( queryThrowable -> { throw new RuntimeException(errorMessage); } ); testExceptions(uriListBatcher, expectedSuccesses, expectedFailures); }
public List<String> testQueryExceptions(QueryDefinition query, int expectedSuccesses, int expectedFailures) { QueryBatcher queryBatcher = newQueryBatcher(query) .onUrisReady( batch -> { throw new InternalError(errorMessage); } ) .onQueryFailure( queryThrowable -> { throw new InternalError(errorMessage); } ); testExceptions(queryBatcher, expectedSuccesses, expectedFailures); // collect the uris this time List<String> matchingUris = Collections.synchronizedList(new ArrayList<>()); queryBatcher = newQueryBatcher(query) .onUrisReady( batch -> matchingUris.addAll(Arrays.asList(batch.getItems())) ) .onUrisReady( batch -> { throw new RuntimeException(errorMessage); } ) .onQueryFailure( queryThrowable -> { throw new RuntimeException(errorMessage); } ); testExceptions(queryBatcher, expectedSuccesses, expectedFailures); return matchingUris; }
public void testExceptions(QueryBatcher queryBatcher, int expectedSuccesses, int expectedFailures) { final AtomicInteger successfulBatchCount = new AtomicInteger(); final AtomicInteger failureBatchCount = new AtomicInteger(); queryBatcher .withBatchSize(1) .onUrisReady( batch -> successfulBatchCount.incrementAndGet() ) .onQueryFailure( queryThrowable -> failureBatchCount.incrementAndGet() ); moveMgr.startJob(queryBatcher); queryBatcher.awaitCompletion(); moveMgr.stopJob(queryBatcher); assertEquals(expectedSuccesses, successfulBatchCount.get()); assertEquals(expectedFailures, failureBatchCount.get()); }
.onQueryFailure((QueryBatchException failure) -> { failedBatches.addAndGet(1); failedEvents.addAndGet(batchSize);
private void testListenerException(QueryBatchListener listener) { final AtomicInteger failureBatchCount = new AtomicInteger(); Iterator<String> iterator = Arrays.asList(new String[] {uri1}).iterator(); QueryBatcher queryBatcher = moveMgr.newQueryBatcher(iterator) .onUrisReady( batch -> logger.debug("uri={}", batch.getItems()[0]) ) .onUrisReady(listener) .onQueryFailure( queryThrowable -> failureBatchCount.incrementAndGet() ); moveMgr.startJob(queryBatcher); queryBatcher.awaitCompletion(); moveMgr.stopJob(queryBatcher); // there should be no failure sent to the QueryBatcher onQueryFailure listeners assertEquals(0, failureBatchCount.get()); }
batcher.onQueryFailure(throwable -> throwable.printStackTrace());
.onUrisReady( new UrisToWriterListener(writer) ) .onUrisReady(batch -> successDocs1.addAndGet(batch.getItems().length)) .onQueryFailure( throwable -> { throwable.printStackTrace(); failures.append("ERROR:[" + throwable + "]\n"); .withBatchSize(19) .onUrisReady(batch -> successDocs2.addAndGet(batch.getItems().length)) .onQueryFailure( throwable -> { throwable.printStackTrace(); failures2.append("ERROR:[" + throwable + "]\n");
.withBatchSize(5000) .onUrisReady( batch -> uris.addAll(Arrays.asList(batch.getItems())) ) .onQueryFailure( throwable -> { throwable.printStackTrace(); failures.append("ERROR:[" + throwable + "]\n"); .onUrisReady(batch -> successDocs.addAndGet(batch.getItems().length)) .onUrisReady(batch -> uris2.addAll(Arrays.asList(batch.getItems()))) .onQueryFailure( throwable -> { throwable.printStackTrace(); failures2.append("ERROR:[" + throwable + "]\n");
.onBatchFailure((batch, throwable) -> throwable.printStackTrace()) .onQueryFailure(throwable -> throwable.printStackTrace()); JobTicket ticket = moveMgr.startJob( batcher ); batcher.awaitCompletion();
QueryBatcher queryBatcher = moveMgr.newQueryBatcher(query) .onUrisReady(new CloseBatchListener()) .onQueryFailure(new CloseFailureListener());
.onQueryFailure( throwable -> { failureBatchCount.incrementAndGet();
.onQueryFailure(failListener); moveMgr.startJob(exportBatcher); .onUrisReady(new DeleteListener()) .onUrisReady(batch -> logger.info("Deleting {} docs", batch.getItems().length) ) .onQueryFailure(failListener); moveMgr.startJob(deleteBatcher); deleteBatcher.awaitCompletion();
.withThreadCount(5) .onUrisReady(batch -> retrievedUris.addAll(Arrays.asList(batch.getItems())) ) .onQueryFailure(exception -> exception.printStackTrace() ); moveMgr.startJob(getUris); getUris.awaitCompletion();
.withBatchSize(5000) .onUrisReady( new UrisToWriterListener(writer) ) .onQueryFailure(exception -> exception.printStackTrace()); JobTicket getUrisTicket = dataMovementManager.startJob(getUris); getUris.awaitCompletion(); dataMovementManager.newQueryBatcher(reader.lines().iterator()) .onUrisReady(new DeleteListener()) .onQueryFailure(exception-> exception.printStackTrace()); JobTicket ticket = dataMovementManager.startJob(performDelete); performDelete.awaitCompletion();
.onQueryFailure(exception -> exception.printStackTrace()); moveMgr.startJob(qb); qb.awaitCompletion();
.withBatchSize(10) .onUrisReady(exportListener) .onQueryFailure( throwable -> throwable.printStackTrace() ); moveMgr.startJob( queryJob );
.withBatchSize(100) .onUrisReady(batch -> count3.addAndGet(batch.getItems().length)) .onQueryFailure((throwable) -> throwable.printStackTrace()); JobTicket ticket3 = moveMgr.startJob( batcher3 ); batcher3.awaitCompletion();
.onQueryFailure( throwable -> { anyFailure.append("error: " + throwable + "\n"); throwable.printStackTrace();
@Test public void testQueryBatcher() { client.newDocumentManager().writeAs(collection + "/test1.json", meta, "[true]"); client.newDocumentManager().writeAs(collection + "/test1.xml", meta, "<xml/>"); client.newDocumentManager().writeAs(collection + "/test1.txt", meta, "text"); assertEquals(3, client.newQueryManager().search(collectionQuery, new SearchHandle()).getTotalResults()); StructuredQueryDefinition query = collectionQuery; // begin copy from "Using QueryBatcher" in src/main/java/com/marklogic/datamovement/package-info.java QueryBatcher qhb = dataMovementManager.newQueryBatcher(query) .withBatchSize(1000) .withThreadCount(20) .withConsistentSnapshot() .onUrisReady(batch -> { for ( String uri : batch.getItems() ) { if ( uri.endsWith(".txt") ) { client.newDocumentManager().delete(uri); } } }) .onQueryFailure(queryBatchException -> queryBatchException.printStackTrace()); JobTicket ticket = dataMovementManager.startJob(qhb); qhb.awaitCompletion(); dataMovementManager.stopJob(ticket); // end copy from "Using QueryBatcher" in src/main/java/com/marklogic/datamovement/package-info.java SearchHandle results = client.newQueryManager().search(collectionQuery, new SearchHandle()); assertEquals(2, results.getTotalResults()); for ( MatchDocumentSummary match : results.getMatchResults() ) { assertTrue(match.getUri().matches(".*/test1.(json|xml)")); } }