org.metricssampler.extensions.jdbc.JdbcExtension Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of metrics-sampler-extension-jdbc Show documentation

There is a newer version: 0.9.1

Show newest version

package org.metricssampler.extensions.jdbc;

import java.sql.DriverManager;
import java.util.Collection;
import java.util.LinkedList;
import java.util.List;

import org.metricssampler.config.ConfigurationException;
import org.metricssampler.config.InputConfig;
import org.metricssampler.config.SharedResourceConfig;
import org.metricssampler.reader.MetricsReader;
import org.metricssampler.resources.SharedResource;
import org.metricssampler.service.AbstractExtension;

public class JdbcExtension extends AbstractExtension {
	@Override
	public Collection> getXBeans() {
		final List> result = new LinkedList>();
		result.add(JdbcInputXBean.class);
		result.add(JdbcConnectionPoolXBean.class);
		return result;
	}

	@Override
	public void initialize() {
		/**
		 * WTF: load the drivers in the caller thread
		 */
		DriverManager.getDrivers();
	}

	@Override
	public boolean supportsInput(final InputConfig config) {
		return config instanceof JdbcInputConfig;
	}

	@Override
	protected MetricsReader doNewReader(final InputConfig config) {
		final JdbcInputConfig jdbcConfig = (JdbcInputConfig) config;
		final SharedResource sharedResource = getGlobalFactory().getSharedResource(jdbcConfig.getPool());
		if (sharedResource instanceof JdbcConnectionPool) {
			return new JdbcMetricsReader(jdbcConfig, (JdbcConnectionPool) sharedResource);
		} else {
			throw new ConfigurationException(jdbcConfig.getPool() + " is not a JDBC connection pool: " + sharedResource);
		}
	}

	@Override
	public boolean supportsSharedResource(final SharedResourceConfig config) {
		return config instanceof JdbcConnectionPoolConfig;
	}

	@Override
	protected SharedResource doNewSharedResource(final SharedResourceConfig config) {
		final JdbcConnectionPoolConfig poolConfig = (JdbcConnectionPoolConfig) config;
		final JdbcConnectionPool result = new JdbcConnectionPool(poolConfig);
		return result;
	}
}