从Spring批处理器调用异步REST api

Question

我写了一个处理列表列表的Spring批处理作业。

Reader返回List of List。处理器处理每个ListItem并返回已处理的List。Writer将内容写入DB并从List of List中sftp。

我有一个从spring批处理程序调用异步REST api的用例。在ListenableFuture响应上，我实现了LitenableFutureCallback来处理成功和失败，这正如预期的那样，但在异步调用返回某些内容之前，ItemProcessor不会等待来自异步api的回调，而是将对象（列表）返回给编写器。

我不知道如何实现和处理异步调用从项目处理器。

我确实读过关于AsyncItem处理机和AsyncItemWriter，但我不确定这是否是我应该在这个场景中使用的东西。

我还想过从AsyncRestTemplate对ListenableFuture响应调用get（），但根据文档，它将阻止当前线程，直到收到响应为止。

我正在寻求一些关于如何实现这一点的帮助。下面是代码片段：

处理器：

public class MailDocumentProcessor implements ItemProcessor<List<MailingDocsEntity>, List<MailingDocsEntity>> {

... Initialization code

@Override
public List<MailingDocsEntity> process(List<MailingDocsEntity> documentsList) throws Exception {
    logger.info("Entering MailingDocsEntity processor");


    List<MailingDocsEntity> synchronizedList = Collections.synchronizedList(documentsList);


    for (MailingDocsEntity mailingDocsEntity : synchronizedList) {
        System.out.println("Reading Mailing id: " + mailingDocsEntity.getMailingId());

       ..code to get the file

         //If the file is not a pdf convert it
         String fileExtension = readFromSpResponse.getFileExtension();
         String fileName = readFromSpResponse.getFileName();
         byte[] fileBytes = readFromSpResponse.getByteArray();

         try {

             //Do checks to make sure PDF file is being sent
             if (!"pdf".equalsIgnoreCase(fileExtension)) {
                 //Only doc, docx and xlsx conversions are supported

                     ...Building REquest object
                     //make async call to pdf conversion service
            pdfService.convertDocxToPdf(request, mailingDocsEntity);

                 } else {
                     logger.error("The file cannot be converted to a pdf.\n"
                        );

                 }
             }


         } catch (Exception ex){
             logger.error("There has been an exception while processing data", ex);

         }

    }
    return synchronizedList;
}

}

异步PdfConversion服务类：

@Service
public class PdfService{


   @Autowired
   @Qualifier("MicroServiceAsyncRestTemplate")
   AsyncRestTemplate microServiceAsyncRestTemplate;

   public ConvertDocxToPdfResponse convertDocxToPdf(ConvertDocxToPdfRequest request, MailingDocsEntity mailingDocsEntity){

        ConvertDocxToPdfResponse pdfResponse = new ConvertDocxToPdfResponse();


            try {

                HttpHeaders headers = new HttpHeaders();
                headers.setContentType(MediaType.APPLICATION_JSON);

                HttpEntity<?> entity = new HttpEntity<>(request, headers);



                ListenableFuture<ResponseEntity<ConvertDocxToPdfResponse>> microServiceResponse = microServiceAsyncRestTemplate.postForEntity(batchMailProcessingConfiguration.getPdfUrl(), entity, ConvertDocxToPdfResponse.class);

                ConvertDocxToPdfResponse resultBody = microServiceResponse.get().getBody();
                microServiceResponse.addCallback(new ListenableFutureCallback<ResponseEntity<ConvertDocxToPdfResponse>>()  {

                    @Override
                    public void onSuccess(ResponseEntity<ConvertDocxToPdfResponse> result) {
                        ...code to do stuff on success


                    }

                    @Override
                    public void onFailure(Throwable ex) {
                        pdfResponse.setMessage("Exception while retrieving response");

                    }
                });

            } catch (Exception e) {
                String message = "There has been an error while issuing a pdf generate request to the pdf micro service";
                pdfResponse.setMessage(message);
                logger.error(message, e);
            }


        return pdfResponse;
    }

}

我最初的批处理作业是同步的，我正在转换为异步的，以加快处理速度。我确实试图寻找类似的问题，但找不到足够的信息。非常感谢您的指点和帮助。

非常感谢。

Answer 1

我确实读过关于AsyncItem处理机和AsyncItemWriter，但我不确定这是否是我应该在这个场景中使用的东西。

是的，AsyncItemProccher和AsyncItemWriter适合您的用例。AsyncItemProccher将在新线程上为一个项目执行委托ItemProcencer的逻辑（您的Rest调用）。一旦项目完成，结果的Future将传递给要写入的AsynchItemWriter。然后AsynchItemWriter将打开Future并写入该项目。这些组件的优点是您不必自己处理Future的包装，展开等。

您可以找到：

• 更多详情请点击此处：https://docs.spring.io/spring-batch/4.0.x/reference/html/spring-batch-integration.html#asynchronous-处理器
• 举个例子：https://github.com/mminella/scaling-demos/blob/master/single-jvm-demos/src/main/java/io/spring/batch/scalingdemos/asyncprocessor/AsyncProcessorJobApplication.java

希望这有帮助。