com.outbrain.ob1k.client.endpoints.AbstractClientEndpoint Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ob1k-core Show documentation
The newest version!
package com.outbrain.ob1k.client.endpoints;

import com.outbrain.ob1k.HttpRequestMethodType;
import com.outbrain.ob1k.client.targets.TargetProvider;
import com.outbrain.ob1k.client.ctx.ClientRequestContext;
import com.outbrain.ob1k.common.marshalling.RequestMarshaller;
import com.outbrain.ob1k.common.marshalling.RequestMarshallerRegistry;
import com.outbrain.ob1k.http.HttpClient;
import com.outbrain.ob1k.http.RequestBuilder;
import com.outbrain.ob1k.http.TypedResponse;
import com.outbrain.ob1k.http.common.ContentType;
import com.outbrain.ob1k.common.marshalling.TypeHelper;
import com.outbrain.ob1k.http.marshalling.MarshallingStrategy;
import org.apache.commons.codec.EncoderException;


import java.io.IOException;
import java.lang.reflect.Method;
import java.lang.reflect.ParameterizedType;
import java.lang.reflect.Type;

/**
 * A common parent for sync and async endpoints used by the client to call the remote target.
 *
 * @author aronen
 */
public abstract class AbstractClientEndpoint {

  protected final RequestMarshallerRegistry marshallerRegistry;
  protected final HttpClient httpClient;
  public final Endpoint endpoint;

  protected AbstractClientEndpoint(final HttpClient httpClient, final RequestMarshallerRegistry marshallerRegistry,
                                   final Endpoint endpoint) {

    this.httpClient = httpClient;
    this.marshallerRegistry = marshallerRegistry;
    this.endpoint = endpoint;
  }

  protected Type extractResponseType() {
    return TypeHelper.extractReturnType(endpoint.getMethod());
  }

  protected RequestBuilder buildEndpointRequestBuilder(final ClientRequestContext ctx,
                                                       final MarshallingStrategy marshallingStrategy)
                                                       throws IOException,EncoderException {
    return ClientEndpointRequestBuilder.build(httpClient, endpoint, ctx, marshallingStrategy);
  }

  protected boolean isTypedResponse(final Type responseType) {
    return responseType instanceof ParameterizedType && ((ParameterizedType) responseType).getRawType() == TypedResponse.class;
  }

  protected byte[] marshallObject(final Object value) throws IOException {
    final RequestMarshaller marshaller = marshallerRegistry.getMarshaller(endpoint.getContentType().requestEncoding());
    return marshaller.marshallRequestParams((Object[]) value);
  }

  public abstract Object invoke(final TargetProvider remoteTarget, final Object[] params) throws Throwable;

  public static class Endpoint {

    private final Method method;
    private final Class serviceType;
    private final ContentType contentType;
    private final String methodPath;
    private final HttpRequestMethodType requestMethodType;

    public Endpoint(final Method method, final Class serviceType, final ContentType contentType, final String methodPath,
                    final HttpRequestMethodType requestMethodType) {

      this.method = method;
      this.serviceType = serviceType;
      this.contentType = contentType;
      this.methodPath = methodPath;
      this.requestMethodType = requestMethodType;
    }

    public Method getMethod() {
      return method;
    }

    public Class getServiceType() {
      return serviceType;
    }

    public ContentType getContentType() {
      return contentType;
    }

    public String getMethodPath() {
      return methodPath;
    }

    public HttpRequestMethodType getRequestMethodType() {
      return requestMethodType;
    }
  }
}