no.nav.metrics.Metric Maven / Gradle / Ivy

Go to download
package no.nav.metrics;

import org.slf4j.MDC;

import java.util.HashMap;
import java.util.Map;
import java.util.Objects;

import static no.nav.log.MDCConstants.*;


abstract class Metric {
    protected final MetricsClient metricsClient;
    protected final String name;
    protected Map fields = new HashMap<>();
    protected Map tags = new HashMap<>();

    private static final String[] MDC_VARIABLES = {
            MDC_CALL_ID,
            MDC_USER_ID,
            MDC_REQUEST_ID,
            MDC_CONSUMER_ID
    };

    Metric(MetricsClient metricsClient, String name) {
        this.metricsClient = metricsClient;
        this.name = name;
        setSuccess();
        for (String mdcVariable : MDC_VARIABLES) {
            String mdcValue = MDC.get(mdcVariable);
            if(Objects.nonNull(mdcValue)) {
                addFieldToReport(mdcVariable, mdcValue);
            }
        }
    }

    public T addFieldToReport(String fieldName, Object value) {
        fields.put(fieldName, value);
        return self();
    }

    public T addTagToReport(String tagName, String value) {
        tags.put(tagName, value);
        return self();
    }

    public T setSuccess() {
        addFieldToReport("success", true);
        return self();
    }

    public T setFailed() {
        addFieldToReport("success", false);
        return self();
    }

    public abstract T report();

    protected abstract T self();

}