如何让主线程等待执行器服务线程完成

Question

Here is my main thread, inside each process i am calling executor service threads. At line 4, before updating the status, i want to wait for my all threads to complete

 line 1- missingStrategyContext.process(FileType.CARD, programId, fromDate, toDate, channel);
 line 2- missingStrategyContext.process(FileType.TRANSACTION, programId, fromDate, toDate, channel);
 line 3- missingStrategyContext.process(FileType.REFUND, programId, fromDate, toDate, channel);

 line 4- processingLog.setProcessingStatus(TransactionState.SUCCESS.name());

Inside each process i am deciding on file type value which strategy i have to call

public void process(FileType fileType, String programId, Long fromDate, Long toDate, String channel){
    map.get(fileType).process(programId, fromDate, toDate, channel, fileType);
}

and then depending on file type i have implemented my process method and called executor service in each file type implementation

@Override
public void process(String programId, Long fromDate, Long toDate, String channel, FileType fileType) {

    MultiTenantTxMgmt multiTenantTxMgmt = MultiTenantTxMgmtUtil.get(programId);
    EntityManager entityManager = null;

    List<MissingReason> reasonList = new ArrayList<>();
    try {
        entityManager = multiTenantTxMgmt.getEntityManager();
        reasonList = missingDataRepository.getDistinctMissingReasonForFileType(entityManager, fileType, TransactionState.READYTOATTEMPT);
    }catch (Exception exception) {
        logger.error(exception.getMessage(), exception);
        throw new UnkownBatchSaveException();
    } finally {
        entityManager.close();
    }

    reasonList.forEach(
            missingReason -> deExecutorService.dotask(() ->
                    missingTransactionStrategyContext.processMissingV3(missingReason, programId, fromDate, toDate, channel, fileType)
            )
    );

}

Answer 1

You can use CountDownLatch#await . Eg copied from docs:

CountDownLatch startSignal = new CountDownLatch(1);
CountDownLatch doneSignal = new CountDownLatch(N);

for (int i = 0; i < N; ++i) // create and start threads
    new Thread(new Worker(startSignal, doneSignal)).start();

doSomethingElse();            // don't let run yet
startSignal.countDown();      // let all threads proceed
doSomethingElse();
doneSignal.await();           // wait for all to finish

Answer 2

You could make doTask method let return a (completable) future (if it does not already) and return a list of futures in method process , eg replace

reasonList.forEach(
            missingReason -> deExecutorService.dotask(() ->
                    missingTransactionStrategyContext.processMissingV3(missingReason, programId, fromDate, toDate, channel, fileType)
            )
    );

with

    final List<Future> futures = reasonList.stream()
        .map(missingReason -> deExecutorService.dotask(() ->
            missingTransactionStrategyContext
                .processMissingV3(missingReason, programId, fromDate, toDate, channel, fileType)))
        .collect(Collectors.toList());

  }
return futures;

Then in your calling part you can wait on all these futures to terminate before continuing.