org.elasticsearch.telemetry.TestTelemetryPlugin Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of framework Show documentation
Elasticsearch subproject :test:framework
There is a newer version: 8.16.0
/*
 * Copyright Elasticsearch B.V. and/or licensed to Elasticsearch B.V. under one
 * or more contributor license agreements. Licensed under the Elastic License
 * 2.0 and the Server Side Public License, v 1; you may not use this file except
 * in compliance with, at your election, the Elastic License 2.0 or the Server
 * Side Public License, v 1.
 */

package org.elasticsearch.telemetry;

import org.elasticsearch.common.settings.Settings;
import org.elasticsearch.plugins.Plugin;
import org.elasticsearch.plugins.TelemetryPlugin;
import org.elasticsearch.telemetry.metric.Instrument;
import org.elasticsearch.telemetry.metric.MeterRegistry;
import org.elasticsearch.telemetry.tracing.Tracer;

import java.util.ArrayList;
import java.util.List;

/**
 * TelemetryPlugin that uses RecordingMeterRegistry to record meter calls
 * and exposes measurement getters.
 */
public class TestTelemetryPlugin extends Plugin implements TelemetryPlugin {

    protected final RecordingMeterRegistry meter = new RecordingMeterRegistry();

    Registration getRegistration(Instrument instrument) {
        return meter.getRecorder().getRegistration(instrument);
    }

    public List getMetrics(Instrument instrument) {
        return meter.getRecorder().getMeasurements(instrument);
    }

    public List getDoubleCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_COUNTER, name);
    }

    public List getLongCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_COUNTER, name);
    }

    public List getLongAsyncCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_ASYNC_COUNTER, name);
    }

    public List getDoubleUpDownCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_UP_DOWN_COUNTER, name);
    }

    public List getLongUpDownCounterMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_UP_DOWN_COUNTER, name);
    }

    public List getDoubleGaugeMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_GAUGE, name);
    }

    public List getLongGaugeMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_GAUGE, name);
    }

    public List getDoubleHistogramMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.DOUBLE_HISTOGRAM, name);
    }

    public List getLongHistogramMeasurement(String name) {
        return meter.getRecorder().getMeasurements(InstrumentType.LONG_HISTOGRAM, name);
    }

    public void collect() {
        meter.getRecorder().collect();
    }

    public void resetMeter() {
        meter.getRecorder().resetCalls();
    }

    public ArrayList getRegisteredMetrics(InstrumentType instrumentType) {
        return meter.getRecorder().getRegisteredMetrics(instrumentType);
    }

    @Override
    public TelemetryProvider getTelemetryProvider(Settings settings) {
        return new TelemetryProvider() {
            @Override
            public Tracer getTracer() {
                return Tracer.NOOP;
            }

            @Override
            public MeterRegistry getMeterRegistry() {
                return meter;
            }
        };
    }
}