RequestsBatchAPI.java

package org.folio.rest.impl;

import static io.vertx.core.Future.succeededFuture;
import static org.folio.rest.impl.util.RequestsApiUtil.hasSamePositionConstraintViolated;
import static org.folio.rest.impl.util.RequestsApiUtil.samePositionInQueueError;
import static org.folio.rest.jaxrs.resource.RequestStorageBatch.PostRequestStorageBatchRequestsResponse.respond201;
import static org.folio.rest.jaxrs.resource.RequestStorageBatch.PostRequestStorageBatchRequestsResponse.respond422WithApplicationJson;
import static org.folio.rest.jaxrs.resource.RequestStorageBatch.PostRequestStorageBatchRequestsResponse.respond500WithTextPlain;
import static org.folio.rest.tools.utils.TenantTool.tenantId;

import java.util.Map;

import javax.ws.rs.core.Response;

import org.folio.rest.jaxrs.model.RequestsBatch;
import org.folio.rest.jaxrs.resource.RequestStorageBatch;
import org.folio.rest.persist.PgUtil;
import org.folio.service.BatchResourceService;
import org.folio.service.request.RequestBatchResourceService;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import io.vertx.core.AsyncResult;
import io.vertx.core.Context;
import io.vertx.core.Handler;

public class RequestsBatchAPI implements RequestStorageBatch {
  private static final Logger LOG = LoggerFactory.getLogger(RequestsBatchAPI.class);

  @Override
  public void postRequestStorageBatchRequests(
    RequestsBatch entity, Map<String, String> okapiHeaders,
    Handler<AsyncResult<Response>> asyncResultHandler, Context context) {

    BatchResourceService batchUpdateService = new BatchResourceService(
      PgUtil.postgresClient(context, okapiHeaders)
    );

    RequestBatchResourceService requestBatchUpdateService =
      new RequestBatchResourceService(tenantId(okapiHeaders), batchUpdateService);

    requestBatchUpdateService.executeRequestBatchUpdate(entity.getRequests(),
      updateResult -> {
        // Successfully updated
        if (updateResult.succeeded()) {
          LOG.debug("Batch update executed successfully");
          asyncResultHandler.handle(succeededFuture(respond201()));
          return;
        }

        // Update failed due to can not have more then one request in the same position
        if (hasSamePositionConstraintViolated(updateResult.cause())) {
          LOG.warn("Same position constraint violated", updateResult.cause());
          asyncResultHandler.handle(succeededFuture(
            respond422WithApplicationJson(samePositionInQueueError(null, null))
          ));
        } else {
          // Other failure occurred
          LOG.warn("Unhandled error occurred during update", updateResult.cause());
          asyncResultHandler.handle(succeededFuture(
            respond500WithTextPlain(updateResult.cause().getMessage())
          ));
        }
      });
  }
}