Examples with BatchResult org.folio.inventory.domain.BatchResult used on opensource projects

Example 1 with BatchResult

use of org.folio.inventory.domain.BatchResult in project mod-inventory by folio-org.

the class ExternalStorageModuleInstanceCollection method addBatch.

@Override
public void addBatch(List<Instance> items, Consumer<Success<BatchResult<Instance>>> resultCallback, Consumer<Failure> failureCallback) {
    List<JsonObject> jsonList = items.stream().map(this::mapToRequest).collect(Collectors.toList());
    JsonObject batchRequest = new JsonObject().put("instances", new JsonArray(jsonList)).put("totalRecords", jsonList.size());
    final var futureResponse = new CompletableFuture<AsyncResult<HttpResponse<Buffer>>>();
    final HttpRequest<Buffer> request = withStandardHeaders(webClient.postAbs(batchAddress));
    request.sendJsonObject(batchRequest, futureResponse::complete);
    futureResponse.thenCompose(this::mapAsyncResultToCompletionStage).thenAccept(response -> {
        if (isBatchResponse(response)) {
            try {
                JsonObject batchResponse = response.getJson();
                JsonArray createdInstances = batchResponse.getJsonArray("instances");
                List<Instance> instancesList = new ArrayList<>();
                for (int i = 0; i < createdInstances.size(); i++) {
                    instancesList.add(mapFromJson(createdInstances.getJsonObject(i)));
                }
                BatchResult<Instance> batchResult = new BatchResult<>();
                batchResult.setBatchItems(instancesList);
                batchResult.setErrorMessages(batchResponse.getJsonArray("errorMessages").getList());
                resultCallback.accept(new Success<>(batchResult));
            } catch (Exception e) {
                LOGGER.error(e);
                failureCallback.accept(new Failure(e.getMessage(), response.getStatusCode()));
            }
        } else {
            failureCallback.accept(new Failure(response.getBody(), response.getStatusCode()));
        }
    });
}