com.spotify.confidence.FlagResolverClientImpl Maven / Gradle / Ivy

Go to download
package com.spotify.confidence;

import com.google.protobuf.Struct;
import com.google.protobuf.Value;
import com.spotify.confidence.shaded.flags.resolver.v1.ResolveFlagsResponse;
import java.time.Duration;
import java.time.Instant;
import java.util.concurrent.CompletableFuture;
import javax.annotation.Nullable;

class FlagResolverClientImpl implements FlagResolverClient {
  public static final String OPEN_FEATURE_RESOLVE_CONTEXT_KEY = "open-feature";
  private final FlagResolver grpcFlagResolver;
  private final @Nullable Telemetry telemetry;
  private final boolean isProvider;

  public FlagResolverClientImpl(FlagResolver grpcFlagResolver, @Nullable Telemetry telemetry) {
    this.grpcFlagResolver = grpcFlagResolver;
    this.telemetry = telemetry;
    this.isProvider = telemetry != null && telemetry.isProvider();
  }

  public FlagResolverClientImpl(FlagResolver grpcFlagResolver) {
    this(grpcFlagResolver, null);
  }

  public CompletableFuture resolveFlags(
      String flagName, ConfidenceValue.Struct context) {
    final Instant start = Instant.now();

    final Struct.Builder evaluationContextBuilder = context.toProto().getStructValue().toBuilder();
    if (context.asMap().containsKey(OPEN_FEATURE_RESOLVE_CONTEXT_KEY)) {
      final Value openFeatureEvaluationContext =
          context.asMap().get(OPEN_FEATURE_RESOLVE_CONTEXT_KEY).toProto();

      evaluationContextBuilder.putAllFields(
          openFeatureEvaluationContext.getStructValue().getFieldsMap());
      evaluationContextBuilder.removeFields(OPEN_FEATURE_RESOLVE_CONTEXT_KEY);
    }

    return this.grpcFlagResolver
        .resolve(flagName, evaluationContextBuilder.build(), isProvider)
        .thenApply(
            response -> {
              if (telemetry != null) {
                final Instant end = Instant.now();
                telemetry.appendLatency(Duration.between(start, end).toMillis());
              }
              return response;
            });
  }

  @Override
  public void close() {
    this.grpcFlagResolver.close();
  }
}