co.elastic.apm.agent.tracer.GlobalTracer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of elastic-apm-agent Show documentation
There is a newer version: 1.52.1
/*
 * Licensed to Elasticsearch B.V. under one or more contributor
 * license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright
 * ownership. Elasticsearch B.V. licenses this file to you 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 co.elastic.apm.agent.tracer;

import co.elastic.apm.agent.tracer.dispatch.HeaderGetter;
import co.elastic.apm.agent.tracer.metrics.DoubleSupplier;
import co.elastic.apm.agent.tracer.metrics.Labels;
import co.elastic.apm.agent.tracer.pooling.ObjectPoolFactory;
import co.elastic.apm.agent.tracer.reference.ReferenceCounted;
import co.elastic.apm.agent.tracer.reference.ReferenceCountedMap;
import co.elastic.apm.agent.tracer.service.Service;
import co.elastic.apm.agent.tracer.service.ServiceInfo;
import com.dslplatform.json.JsonWriter;

import javax.annotation.Nullable;
import java.util.Set;
import java.util.concurrent.TimeUnit;

public class GlobalTracer implements Tracer {

    private static final GlobalTracer INSTANCE = new GlobalTracer();
    private volatile Tracer tracer = NoopTracer.INSTANCE;

    private GlobalTracer() {
    }

    public static Tracer get() {
        return INSTANCE;
    }

    public static synchronized void setNoop() {
        boolean running = INSTANCE.tracer.isRunning();
        if (running) {
            throw new IllegalStateException("Can't override tracer as current tracer is already running");
        }
        INSTANCE.tracer = NoopTracer.INSTANCE;
    }

    public static synchronized void init(Tracer tracer) {
        if (!isNoop()) {
            throw new IllegalStateException("Tracer is already initialized");
        }
        INSTANCE.tracer = tracer;
    }

    public static boolean isNoop() {
        return INSTANCE.tracer == NoopTracer.INSTANCE;
    }

    @Override
    public boolean isRunning() {
        return tracer.isRunning();
    }

    @Nullable
    @Override
    public  T probe(Class type) {
        return tracer.probe(type);
    }

    @Override
    public  T require(Class type) {
        return tracer.require(type);
    }

    @Override
    public  T getConfig(Class configuration) {
        return tracer.getConfig(configuration);
    }

    @Override
    public ObjectPoolFactory getObjectPoolFactory() {
        return tracer.getObjectPoolFactory();
    }

    @Override
    public  ReferenceCountedMap newReferenceCountedMap() {
        return tracer.newReferenceCountedMap();
    }

    @Override
    public Set getTraceHeaderNames() {
        return tracer.getTraceHeaderNames();
    }

    @Override
    public TraceState currentContext() {
        return tracer.currentContext();
    }

    @Nullable
    @Override
    public AbstractSpan getActive() {
        return tracer.getActive();
    }

    @Nullable
    @Override
    public Transaction currentTransaction() {
        return tracer.currentTransaction();
    }

    @Nullable
    @Override
    public ErrorCapture getActiveError() {
        return tracer.getActiveError();
    }

    @Nullable
    @Override
    public Transaction startRootTransaction(@Nullable ClassLoader initiatingClassLoader) {
        return tracer.startRootTransaction(initiatingClassLoader);
    }

    @Nullable
    @Override
    public  Transaction startChildTransaction(@Nullable C headerCarrier, HeaderGetter headerGetter, @Nullable ClassLoader initiatingClassLoader) {
        return tracer.startChildTransaction(headerCarrier, headerGetter, initiatingClassLoader);
    }

    @Nullable
    @Override
    public ErrorCapture captureException(@Nullable Throwable e, @Nullable ClassLoader initiatingClassLoader) {
        return tracer.captureException(e, initiatingClassLoader);
    }

    @Override
    public void reportLog(String log) {
        tracer.reportLog(log);
    }

    @Override
    public void reportLog(byte[] log) {
        tracer.reportLog(log);
    }

    @Nullable
    @Override
    public Service createService(String ephemeralId) {
        return tracer.createService(ephemeralId);
    }

    @Nullable
    @Override
    public Throwable redactExceptionIfRequired(@Nullable Throwable original) {
        return tracer.redactExceptionIfRequired(original);
    }

    @Override
    public void flush() {
        tracer.flush();
    }

    @Override
    public void completeMetaData(String name, String version, String id, String region) {
        tracer.completeMetaData(name, version, id, region);
    }

    @Override
    public void removeGauge(String name, Labels.Immutable labels) {
        tracer.removeGauge(name, labels);
    }

    @Override
    public void addGauge(String name, Labels.Immutable labels, DoubleSupplier supplier) {
        tracer.addGauge(name, labels, supplier);
    }

    @Override
    public void submit(Runnable job) {
        tracer.submit(job);
    }

    @Override
    public void schedule(Runnable job, long interval, TimeUnit timeUnit) {
        tracer.schedule(job, interval, timeUnit);
    }

    @Override
    public void addShutdownHook(AutoCloseable hook) {
        tracer.addShutdownHook(hook);
    }

    @Override
    public void reportMetric(JsonWriter metrics) {
        tracer.reportMetric(metrics);
    }

    @Nullable
    @Override
    public ServiceInfo getServiceInfoForClassLoader(@Nullable ClassLoader initiatingClassLoader) {
        return tracer.getServiceInfoForClassLoader(initiatingClassLoader);
    }

    @Override
    public void setServiceInfoForClassLoader(@Nullable ClassLoader classLoader, ServiceInfo serviceInfo) {
        tracer.setServiceInfoForClassLoader(classLoader, serviceInfo);
    }

    @Override
    public ServiceInfo autoDetectedServiceInfo() {
        return tracer.autoDetectedServiceInfo();
    }
}