io.smallrye.graphql.cdi.tracing.TracingService Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of smallrye-graphql-cdi Show documentation
Allow lookup up of GraphQL Beans via CDI
There is a newer version: 2.11.0
package io.smallrye.graphql.cdi.tracing;

import java.util.List;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;

import jakarta.annotation.Priority;
import jakarta.enterprise.inject.spi.CDI;

import org.jboss.logging.Logger;

import io.opentelemetry.api.trace.Span;
import io.opentelemetry.api.trace.StatusCode;
import io.opentelemetry.api.trace.Tracer;
import io.opentelemetry.context.Scope;
import io.smallrye.graphql.api.Context;
import io.smallrye.graphql.config.ConfigKey;
import io.smallrye.graphql.execution.event.Priorities;
import io.smallrye.graphql.spi.EventingService;

/**
 * Listening for operation start/end event and create traces from it
 *
 * @author Jan Martiska ([email protected])
 * @author Phillip Kruger ([email protected])
 */
@Priority(Priorities.FIRST_IN_LAST_OUT)
public class TracingService implements EventingService {

    private Logger LOG = Logger.getLogger(TracingService.class);
    private static final Map spans = new ConcurrentHashMap<>();
    private static final Map scopes = new ConcurrentHashMap<>();

    private Tracer tracer;

    @Override
    public void beforeExecute(Context context) {
        String operationName = getOperationName(context);
        Span span = getTracer().spanBuilder(operationName)
                .setAttribute("graphql.executionId", context.getExecutionId())
                .setAttribute("graphql.operationType", getOperationNameString(context.getRequestedOperationTypes()))
                .setAttribute("graphql.operationName", context.getOperationName().orElse(EMPTY))
                .startSpan();
        LOG.trace("Start span " + span.getSpanContext().getSpanId());

        spans.put(context.getExecutionId(), span);
        scopes.put(context.getExecutionId(), span.makeCurrent());
    }

    @Override
    public void afterExecute(Context context) {
        Span span = spans.remove(context.getExecutionId());

        if (span != null) {
            LOG.trace("Finish span " + span.getSpanContext().getSpanId());
            scopes.remove(context.getExecutionId()).close();
            span.end();
        }
    }

    @Override
    public void errorExecute(Context context, Throwable t) {
        Span span = spans.remove(context.getExecutionId());
        if (span != null) {
            LOG.trace("Exceptionally finish span " + span.getSpanContext().getSpanId());
            span.recordException(t);
            span.setStatus(StatusCode.ERROR);
            scopes.remove(context.getExecutionId()).close();
            span.end();
        }
    }

    @Override
    public String getConfigKey() {
        return ConfigKey.ENABLE_TRACING;
    }

    private Tracer getTracer() {
        if (tracer == null) {
            this.tracer = CDI.current().select(Tracer.class).get();
        }
        return tracer;
    }

    private static String getOperationName(Context context) {
        if (context.getOperationName().isPresent()) {
            return PREFIX + ":" + context.getOperationName().get();
        }
        return PREFIX;
    }

    private String getOperationNameString(List types) {
        return String.join(UNDERSCORE, types);
    }

    private static final String UNDERSCORE = "_";
    private static final String EMPTY = "";
    private static final String PREFIX = "GraphQL";
}