com.commercetools.sunrise.framework.controllers.metrics.MetricsLogger Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of common_2.11 Show documentation
common
The newest version!
package com.commercetools.sunrise.framework.controllers.metrics;

import com.commercetools.sunrise.ctp.CtpLogUtils;
import io.sphere.sdk.client.SphereClient;
import io.sphere.sdk.client.metrics.ObservedTotalDuration;
import io.sphere.sdk.client.metrics.SimpleMetricsSphereClient;
import io.sphere.sdk.models.Base;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import play.mvc.Action;
import play.mvc.Http;
import play.mvc.Result;

import javax.inject.Inject;
import java.util.List;
import java.util.concurrent.CompletionStage;

import static java.util.stream.Collectors.joining;

/**
 * Action that logs metrics for every request.
 */
final class MetricsLogger extends Action.Simple {

    private static final Logger LOGGER = LoggerFactory.getLogger(MetricsLogger.class);

    private final SphereClient sphereClient;

    @Inject
    MetricsLogger(final SphereClient sphereClient) {
        this.sphereClient = sphereClient;
    }

    @Override
    public CompletionStage call(final Http.Context ctx) {
        if (LOGGER.isDebugEnabled()) {
            if (sphereClient instanceof SimpleMetricsSphereClient) {
                return callWithMetrics((SimpleMetricsSphereClient) sphereClient, ctx);
            } else {
                LOGGER.warn(ctx.request() + " enabled logging via @LogMetrics annotation without a SimpleMetricsSphereClient");
            }
        }
        return delegate.call(ctx);
    }

    private CompletionStage callWithMetrics(final SimpleMetricsSphereClient metricsSphereClient, final Http.Context ctx) {
        final long startTimestamp = System.currentTimeMillis();
        final SphereClientMetricsObserver observer = new SphereClientMetricsObserver();
        metricsSphereClient.getMetricObservable().addObserver(observer);
        final CompletionStage resultStage = delegate.call(ctx);
        return resultStage.thenApply(result -> {
            final long stopTimestamp = System.currentTimeMillis();
            logRequestData(observer.getMetrics(), ctx, stopTimestamp - startTimestamp);
            metricsSphereClient.getMetricObservable().deleteObserver(observer);
            return result;
        });
    }

    private static void logRequestData(final List metrics, final Http.Context ctx, final long totalDuration) {
        if (LOGGER.isTraceEnabled()) {
            logTraceRequestData(metrics);
        } else {
            logDebugRequestData(metrics, ctx, totalDuration);
        }
    }

    private static void logDebugRequestData(final List metrics, final Http.Context ctx, final long totalDuration) {
        String report = metrics.stream()
                .map(data -> String.format("(%dms) %s",
                        data.getDurationInMilliseconds(),
                        CtpLogUtils.printableRequest(data.getRequest())))
                .collect(joining("\n"));
        if (!report.isEmpty()) {
            report = ":\n" + report;
        }
        LOGGER.debug("({}ms) {} used {} CTP request(s){}",
                totalDuration,
                ctx.request(),
                metrics.size(),
                report);
    }

    private static void logTraceRequestData(final List metrics) {
        final String report = metrics.stream()
                .map(Base::toString)
                .collect(joining("\n"));
        if (!report.isEmpty()) {
            LOGGER.trace(report);
        }
    }
}