org.metricssampler.reader.AbstractMetricsReader Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of metrics-sampler-core Show documentation
The newest version!
package org.metricssampler.reader;

import org.apache.commons.beanutils.BeanUtils;
import org.metricssampler.config.InputConfig;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.lang.reflect.InvocationTargetException;
import java.util.*;
import java.util.Map.Entry;

public abstract class AbstractMetricsReader implements MetricsReader {
	protected static final String CONFIG_VAR_PREFIX = "input";
	protected static final Set IGNORED_CONFIG_PROPERTIES = new HashSet<>(Arrays.asList("class", "variables"));
	protected final T config;
	protected final Logger logger;
	protected final Logger timingsLogger;
	protected final Map variables;

	protected AbstractMetricsReader(final T config) {
		this.config = config;
		this.logger = LoggerFactory.getLogger("reader." + config.getName());
		this.timingsLogger = LoggerFactory.getLogger("timings.reader");
		variables = prepareVariables();
	}

	private Map prepareVariables() {
		final Map result = new HashMap<>();
		result.putAll(config.getVariables());
		try {
			@SuppressWarnings("unchecked")
			final Map configProperties = BeanUtils.describe(config);

			for (final Entry entry : configProperties.entrySet()) {
				final String name = entry.getKey();
				if (!IGNORED_CONFIG_PROPERTIES.contains(name)) {
					result.put(CONFIG_VAR_PREFIX + "." + name, entry.getValue());
				}
			}
		} catch (final IllegalAccessException e) {
			logger.warn("Failed to introspect configuration bean: " + config, e);
		} catch (final InvocationTargetException e) {
			logger.warn("Failed to introspect configuration bean: " + config, e);
		} catch (final NoSuchMethodException e) {
			logger.warn("Failed to introspect configuration bean: " + config, e);
		}
		defineCustomVariables(result);
		return Collections.unmodifiableMap(result);
	}

	protected void defineCustomVariables(final Map variables) {
		// no custom variables by default
	}

	@Override
	public Map getVariables() {
		return variables;
	}

	@Override
	public String toString() {
		return getClass().getSimpleName() + "[" + config.getName() + "]";
	}

	@Override
	public void reset() {
		// do nothing in this case
	}
	
	
}