net.thisptr.java.prometheus.metrics.agent.misc.AttributeNamePattern Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of java-prometheus-metrics-agent Show documentation

Java agent for collecting and reporting metrics to Prometheus

There is a newer version: 0.0.5

Show newest version

package net.thisptr.java.prometheus.metrics.agent.misc;

import java.util.Collections;
import java.util.LinkedHashMap;
import java.util.Map;
import java.util.regex.Pattern;

import javax.management.ObjectName;

import net.thisptr.java.prometheus.metrics.agent.javacc.AttributeNamePatternParser;

public class AttributeNamePattern {
	public final Pattern domain;
	public final Map keys;
	public final Pattern attribute;

	public AttributeNamePattern(final String domain, final Map keys, final String attribute) {
		this.domain = domain != null ? Pattern.compile(domain) : null;
		final Map tmp = new LinkedHashMap<>();
		keys.forEach((k, v) -> {
			tmp.put(k, Pattern.compile(v));
		});
		this.keys = Collections.unmodifiableMap(tmp);
		this.attribute = attribute != null ? Pattern.compile(attribute) : null;
	}

	public boolean matches(final ObjectName name, final String attribute) {
		if (domain != null && !domain.matcher(name.getDomain()).matches())
			return false;

		final Map target = name.getKeyPropertyList();
		for (final Map.Entry patternEntry : keys.entrySet()) {
			String targetValue = target.get(patternEntry.getKey());
			if (targetValue == null)
				return false;
			if (targetValue.startsWith("\""))
				targetValue = ObjectName.unquote(targetValue);
			if (!patternEntry.getValue().matcher(targetValue).matches())
				return false;
		}

		if (this.attribute != null && (attribute == null || !this.attribute.matcher(attribute).matches()))
			return false;

		return true;
	}

	public static AttributeNamePattern compile(final String patternText) {
		return AttributeNamePatternParser.parse(patternText);
	}
}