cloud.prefab.client.ClientAuthenticationInterceptor Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of prefab-cloud-java Show documentation

API Client for https://prefab.cloud: rate limits, feature flags and semaphores as a service

The newest version!

package cloud.prefab.client;

import cloud.prefab.client.util.MavenInfo;
import io.grpc.CallOptions;
import io.grpc.Channel;
import io.grpc.ClientCall;
import io.grpc.ClientInterceptor;
import io.grpc.ForwardingClientCall;
import io.grpc.ForwardingClientCallListener;
import io.grpc.Metadata;
import io.grpc.MethodDescriptor;

public class ClientAuthenticationInterceptor implements ClientInterceptor {

  public static final Metadata.Key CUSTOM_HEADER_KEY = Metadata.Key.of(
    "auth",
    Metadata.ASCII_STRING_MARSHALLER
  );

  public static final Metadata.Key CLIENT_HEADER_KEY = Metadata.Key.of(
    "client",
    Metadata.ASCII_STRING_MARSHALLER
  );

  private static final String CLIENT_HEADER_VALUE = String.format(
    "%s.%s",
    MavenInfo.getInstance().getArtifactId(),
    MavenInfo.getInstance().getVersion()
  );

  private final String apikey;

  public ClientAuthenticationInterceptor(String apikey) {
    this.apikey = apikey;
  }

  @Override
  public  ClientCall interceptCall(
    MethodDescriptor methodDescriptor,
    CallOptions callOptions,
    Channel channel
  ) {
    return new ForwardingClientCall.SimpleForwardingClientCall(
      channel.newCall(methodDescriptor, callOptions)
    ) {
      @Override
      public void start(Listener responseListener, Metadata headers) {
        headers.put(CLIENT_HEADER_KEY, CLIENT_HEADER_VALUE);
        headers.put(CUSTOM_HEADER_KEY, apikey);
        super.start(
          new ForwardingClientCallListener.SimpleForwardingClientCallListener(
            responseListener
          ) {},
          headers
        );
      }
    };
  }
}