com.ringcentral.platform.metrics.producers.AbstractMetricsProducer Maven / Gradle / Ivy

Go to download
package com.ringcentral.platform.metrics.producers;

import com.ringcentral.platform.metrics.MetricMod;
import com.ringcentral.platform.metrics.MetricModBuilder;
import com.ringcentral.platform.metrics.labels.Label;
import com.ringcentral.platform.metrics.names.MetricName;
import com.ringcentral.platform.metrics.var.configs.builders.AbstractVarConfigBuilder;
import com.ringcentral.platform.metrics.var.configs.builders.VarConfigBuilder;
import com.ringcentral.platform.metrics.var.doubleVar.configs.builders.DoubleVarConfigBuilder;
import com.ringcentral.platform.metrics.var.longVar.configs.builders.LongVarConfigBuilder;
import com.ringcentral.platform.metrics.var.objectVar.configs.builders.ObjectVarConfigBuilder;
import com.ringcentral.platform.metrics.var.stringVar.configs.builders.StringVarConfigBuilder;

import java.util.function.Supplier;
import java.util.regex.Pattern;

import static com.ringcentral.platform.metrics.var.doubleVar.configs.builders.DoubleVarConfigBuilder.doubleVarConfigBuilder;
import static com.ringcentral.platform.metrics.var.longVar.configs.builders.LongVarConfigBuilder.longVarConfigBuilder;
import static com.ringcentral.platform.metrics.var.objectVar.configs.builders.ObjectVarConfigBuilder.objectVarConfigBuilder;
import static com.ringcentral.platform.metrics.var.stringVar.configs.builders.StringVarConfigBuilder.stringVarConfigBuilder;
import static java.util.Objects.requireNonNull;

public abstract class AbstractMetricsProducer implements MetricsProducer {

    protected static final Pattern WHITESPACE_PATTERN = Pattern.compile("[\\s]+");

    private final MetricName namePrefix;
    private final MetricMod metricMod;

    protected AbstractMetricsProducer(MetricName namePrefix, MetricModBuilder metricModBuilder) {
        this.namePrefix = requireNonNull(namePrefix);
        this.metricMod = metricModBuilder != null ? metricModBuilder.build() : null;
    }

    protected MetricName namePrefix() {
        return namePrefix;
    }

    protected MetricName nameWithSuffix(MetricName suffix) {
        return namePrefix.isEmpty() ? suffix : MetricName.of(namePrefix, suffix);
    }

    protected MetricName nameWithSuffix(String... suffix) {
        return namePrefix.isEmpty() ? MetricName.of(suffix) : MetricName.of(namePrefix, suffix);
    }

    protected Supplier objectVarConfigBuilderSupplier(String description, Label... labels) {
        final var builder = objectVarConfigBuilder();
        setDescriptionAndLabels(builder, description, labels);
        return () -> modified(builder);
    }

    protected Supplier longVarConfigBuilderSupplier(String description, Label... labels) {
        final var builder = longVarConfigBuilder();
        setDescriptionAndLabels(builder, description, labels);
        return () -> modified(builder);
    }

    protected Supplier doubleVarConfigBuilderSupplier(String description, Label... labels) {
        final var builder = doubleVarConfigBuilder();
        setDescriptionAndLabels(builder, description, labels);
        return () -> modified(builder);
    }

    protected Supplier stringVarConfigBuilderSupplier(String description, Label... labels) {
        final var builder = stringVarConfigBuilder();
        setDescriptionAndLabels(builder, description, labels);
        return () -> modified(builder);
    }

    private void setDescriptionAndLabels(AbstractVarConfigBuilder builder, String description, Label... labels) {
        builder.description(description);

        if (labels.length != 0) {
            builder.labels(labels);
        }
    }

    private > CB modified(CB builder) {
        if (metricMod != null) {
            if (metricMod.hasVarConfigBuilder()) {
                builder.modify(metricMod.varConfigBuilder());
            } else if (metricMod.hasMetricConfigBuilder()) {
                builder.modify(metricMod.metricConfigBuilder());
            }
        }

        return builder;
    }
}