ai.promoted.delivery.client.DeliveryRequestState Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of deliveryclient Show documentation

A Java Client to contact the Promoted.ai Delivery API.

There is a newer version: 3.2.0

package ai.promoted.delivery.client;

import java.util.ArrayList;
import java.util.List;
import java.util.logging.Logger;

import ai.promoted.proto.delivery.Insertion;
import ai.promoted.proto.delivery.Request;
import ai.promoted.proto.delivery.Response;

/**
 * Holds state of a request during processing.
 */
class DeliveryRequestState {

  private static final Logger LOGGER = Logger.getLogger(DeliveryRequestState.class.getName());

  /** The delivery request we're processing */
  private final DeliveryRequest deliveryRequest;

  /**
   * Creates a state object for the given request.
   * 
   * @param deliveryRequest the delivery request we are processing.
   */
  public DeliveryRequestState(DeliveryRequest deliveryRequest) {
    this.deliveryRequest = deliveryRequest;
  }
  
  /**
   * Gets the response to return to the client with any necessary post-processing.
   * @param resp the response from Delivery API.
   * @return the post-processed response
   */
  public Response getResponseToReturn(Response resp) {
    return resp;
  }

  /**
   * Gets the request to send to Delivery API with any necessary preprocessing.
   * 
   * @param maxRequestInsertions the maximum number of request insertions, which we will trim to if necessary.
   * @return the pre-processed request.
   */
  public Request getRequestToSend(int maxRequestInsertions) {
    Request.Builder reqBuilder = deliveryRequest.getRequestBuilder();
    
    // Trim the list if necessary.
    if (reqBuilder.getInsertionCount() > maxRequestInsertions) {
      LOGGER.warning("Too many request insertions, truncating at " + maxRequestInsertions);
      List truncatedList = new ArrayList<>(reqBuilder.getInsertionList().subList(0, maxRequestInsertions));
      reqBuilder.clearInsertion();
      reqBuilder.addAllInsertion(truncatedList);
    }

    return reqBuilder.build();
  }
}