com.expediagroup.streamplatform.streamregistry.graphql.GraphQLMetricHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of stream-registry-graphql-api Show documentation
The newest version!
/**
 * Copyright (C) ${license.git.copyrightYears} Expedia, Inc.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.expediagroup.streamplatform.streamregistry.graphql;

import static lombok.AccessLevel.PACKAGE;

import java.lang.reflect.InvocationHandler;
import java.lang.reflect.InvocationTargetException;
import java.lang.reflect.Method;
import java.util.function.Supplier;

import com.google.common.base.Stopwatch;

import io.micrometer.core.instrument.MeterRegistry;
import io.micrometer.core.instrument.Tags;

import org.springframework.security.authentication.AnonymousAuthenticationToken;
import org.springframework.security.core.Authentication;
import org.springframework.security.core.context.SecurityContextHolder;

import lombok.Getter;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;

@Slf4j
@RequiredArgsConstructor(access = PACKAGE)
@Getter(PACKAGE)
class GraphQLMetricHandler implements InvocationHandler {
  private final Object delegate;
  private final MeterRegistry registry;
  private final Supplier authenticationSupplier;

  public GraphQLMetricHandler(Object delegate, MeterRegistry registry) {
    this(delegate, registry, () -> SecurityContextHolder.getContext().getAuthentication());
  }

  @Override
  public Object invoke(Object proxy, Method method, Object[] args) throws Throwable {
    boolean success = false;
    Stopwatch stopwatch = Stopwatch.createStarted();
    try {
      Object result = method.invoke(delegate, args);
      success = true;
      return result;
    } catch (InvocationTargetException e) {
      if (e.getCause() != null) {
        throw e.getCause();
      }
      throw e;
    } finally {
      Tags tags = Tags
          .of("api", method.getDeclaringClass().getSimpleName())
          .and("method", method.getName())
          .and("result", success ? "success" : "failure")
          .and("authentication_group", AuthenticationGroup.of(authenticationSupplier.get()).name());
      registry.timer("graphql_api", tags).record(stopwatch.elapsed());
    }
  }

  enum AuthenticationGroup {
    AUTHENTICATED,
    ANONYMOUS;

    static AuthenticationGroup of(Authentication authentication) {
      if (authentication == null || authentication instanceof AnonymousAuthenticationToken) {
        return ANONYMOUS;
      }
      return AUTHENTICATED;
    }
  }
}