org.odata4j.cxf.consumer.CxfConsumerQueryLinksRequest Maven / Gradle / Ivy

Go to download
package org.odata4j.cxf.consumer;

import org.core4j.Enumerable;
import org.core4j.Func1;
import org.odata4j.consumer.ODataClientRequest;
import org.odata4j.core.OEntityId;
import org.odata4j.core.OEntityIds;
import org.odata4j.core.OEntityKey;
import org.odata4j.edm.EdmDataServices;
import org.odata4j.format.FormatType;
import org.odata4j.format.SingleLink;

class CxfConsumerQueryLinksRequest extends CxfConsumerQueryRequestBase {

  private final String targetNavProp;

  CxfConsumerQueryLinksRequest(FormatType formatType, String serviceRootUri, EdmDataServices metadata, OEntityId sourceEntity, String targetNavProp) {
    super(formatType, serviceRootUri, metadata, OEntityIds.toKeyString(sourceEntity));
    this.targetNavProp = targetNavProp;
  }

  public static Func1 linksPath(final String targetNavProp, final Object[] targetKeyValues) {
    return new Func1() {
      public String apply(String input) {
        String keyString = targetKeyValues == null || targetKeyValues.length == 0
            ? "" : OEntityKey.create(targetKeyValues).toKeyString();
        return input + "/$links/" + targetNavProp + keyString;
      }
    };
  }

  @Override
  public Enumerable execute() {
    ODataCxfClient client = new ODataCxfClient(this.getFormatType());
    try {
      ODataClientRequest request = buildRequest(linksPath(targetNavProp, null));
      return Enumerable.create(client.getLinks(request)).select(new Func1() {
        @Override
        public OEntityId apply(SingleLink link) {
          return OEntityIds.parse(getServiceRootUri(), link.getUri());
        }
      });
    } finally {
      client.shuttdown();
    }
  }
}