@After
public void tearDown() throws Exception {
LOGGER.debug("Doing tearDown");
// Stops the threaded executor JmxCollector is using to poll MBean attribute
if (jmxCollector != null) {
jmxCollector.destroy();
}
if (rrdDb == null) return;
if (!rrdDb.isClosed()) {
rrdDb.close();
}
String path = rrdDb.getPath();
File rrdFile = new File(path);
if (rrdFile.exists()) {
boolean status = rrdFile.delete();
if (status) {
LOGGER.debug("Successfully deleted rrdFile " + path);
} else {
LOGGER.debug("Unable to delete rrdFile " + path);
}
} else {
LOGGER.debug("rrdFile " + path + " does not exist - cannot delete");
}
}
@Test
public void testManyUpdatesInRapidSuccession() throws Exception {
createJmxCollector("Uptime", JmxCollector.COUNTER_DATA_SOURCE_TYPE, 1);
// Set high update delta time so that samples will be skipped
jmxCollector.setMinimumUpdateTimeDelta(3);
// Sleep long enough for some data to be collected
long numRrdSamples = 4;
Thread.sleep(numRrdSamples * 1000);
// Expected skip count is 2 because first sample is always accepted and
// num samples is one more than minimum update time delta
long expectedSampleSkipCount = numRrdSamples - 2;
assertThat(jmxCollector.getSampleSkipCount(), is(expectedSampleSkipCount));
}
/** *************************************************************************************** */
private void createJmxCollector(String mbeanAttributeName, String dataSourceType, int sampleRate)
throws Exception {
dataSourceName = mbeanAttributeName.toLowerCase();
rrdPath = dataSourceName + ".rrd";
jmxCollector = new JmxCollector();
jmxCollector.setMbeanName("java.lang:type=Runtime");
jmxCollector.setMbeanAttributeName(mbeanAttributeName);
jmxCollector.setRrdPath(rrdPath);
jmxCollector.setRrdDataSourceName(dataSourceName);
jmxCollector.setRrdDataSourceType(dataSourceType);
jmxCollector.setSampleRate(sampleRate);
jmxCollector.setMinimumUpdateTimeDelta(0);
jmxCollector.setMetricsDir(TEST_DIR);
// Simulates what Spring beans container would do
jmxCollector.configureCollector();
}
@Test(expected = CollectorException.class)
public void testInaccessibleMbeanAttribute() throws IOException, CollectorException {
String mbeanAttributeName = "Invalid";
dataSourceName = mbeanAttributeName.toLowerCase();
rrdPath = dataSourceName + ".rrd";
jmxCollector = new JmxCollector();
jmxCollector.setMbeanName("java.lang:type=Runtime");
jmxCollector.setMbeanAttributeName(mbeanAttributeName);
jmxCollector.setRrdPath(rrdPath);
jmxCollector.setRrdDataSourceName(dataSourceName);
jmxCollector.setRrdDataSourceType(JmxCollector.COUNTER_DATA_SOURCE_TYPE);
jmxCollector.setSampleRate(1);
jmxCollector.setMetricsDir(TEST_DIR);
jmxCollector.setMbeanTimeoutMillis(50);
// Simulates what Spring beans container would do
jmxCollector.configureCollector();
}
@Test
public void testRrdFileCreationForGaugeDataSource() throws Exception {
// Set sample rate to 1 sec (default is 60 seconds) so that unit test runs quickly
int sampleRate = 1;
createJmxCollector("Uptime", JmxCollector.GAUGE_DATA_SOURCE_TYPE, sampleRate);
String rrdFilename = jmxCollector.getRrdPath();
assertThat(rrdFilename, is(TEST_DIR + rrdPath));
rrdDb = new RrdDb(rrdFilename);
assertThat(rrdDb, not(nullValue()));
assertThat(rrdDb.isClosed(), is(false));
Header header = rrdDb.getHeader();
assertThat(header, not(nullValue()));
assertThat(header.getStep(), is((long) sampleRate)); // verify 60 second sample rate
assertThat(rrdDb.getDsCount(), is(1));
Datasource dataSource = rrdDb.getDatasource(dataSourceName);
assertThat(dataSource, not(nullValue()));
DsType dataSourceType = dataSource.getType();
assertThat(dataSourceType, is(DsType.GAUGE));
assertThat(rrdDb.getArcCount(), is(8));
Archive archive = rrdDb.getArchive(ConsolFun.MIN, 1);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(60));
archive = rrdDb.getArchive(ConsolFun.MIN, 15);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(JmxCollector.ONE_YEAR_IN_15_MINUTE_STEPS));
archive = rrdDb.getArchive(ConsolFun.MAX, 1);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(60));
archive = rrdDb.getArchive(ConsolFun.MAX, 15);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(JmxCollector.ONE_YEAR_IN_15_MINUTE_STEPS));
archive = rrdDb.getArchive(ConsolFun.AVERAGE, 1);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(60));
archive = rrdDb.getArchive(ConsolFun.AVERAGE, 15);
assertThat(archive, not(nullValue()));
assertThat(archive.getRows(), is(JmxCollector.ONE_YEAR_IN_15_MINUTE_STEPS));
}
@Test(expected = CollectorException.class)
public void testUnsupportedDataSourceType() throws IOException, CollectorException {
String mbeanAttributeName = "Uptime";
dataSourceName = mbeanAttributeName.toLowerCase();
rrdPath = dataSourceName + ".rrd";
jmxCollector = new JmxCollector();
jmxCollector.setMbeanName("java.lang:type=Runtime");
jmxCollector.setMbeanAttributeName(mbeanAttributeName);
jmxCollector.setRrdPath(rrdPath);
jmxCollector.setRrdDataSourceName(dataSourceName);
jmxCollector.setRrdDataSourceType("DERIVE");
jmxCollector.setSampleRate(1);
jmxCollector.setMetricsDir(TEST_DIR);
// Simulates what Spring beans container would do
jmxCollector.configureCollector();
}
@Test(expected = CollectorException.class)
public void testNullDataSourceName() throws IOException, CollectorException {
String mbeanAttributeName = "Uptime";
dataSourceName = null;
rrdPath = "uptime.rrd";
jmxCollector = new JmxCollector();
jmxCollector.setMbeanName("java.lang:type=Runtime");
jmxCollector.setMbeanAttributeName(mbeanAttributeName);
jmxCollector.setRrdPath(rrdPath);
jmxCollector.setRrdDataSourceName(dataSourceName);
jmxCollector.setRrdDataSourceType(JmxCollector.COUNTER_DATA_SOURCE_TYPE);
jmxCollector.setSampleRate(1);
jmxCollector.setMetricsDir(TEST_DIR);
// Simulates what Spring beans container would do
jmxCollector.configureCollector();
}
private void collectData(int numRrdStepIterations) throws Exception {
String rrdFilename = jmxCollector.getRrdPath();
rrdDb = new RrdDb(rrdFilename);
Header header = rrdDb.getHeader();
// Wait for "n" iterations of RRDB's sample rate, then see if MBean value was collected
LOGGER.debug("Sleeping for " + (header.getStep() * numRrdStepIterations) + " seconds");
Thread.sleep((header.getStep() * numRrdStepIterations) * 1000);
// LOGGER.debug(rrdDb.dump());
long endTime = Calendar.getInstance().getTimeInMillis() / 1000;
// +1 because the fetch gets data for times inclusively, e.g.,
// endTime=12345, so startTime=12345-4=12341,
// then fetch data for timestamps 12341, 12342, 12343, 12344, 12345 (which is 5 values)
long startTime = endTime - numRrdStepIterations + 1;
LOGGER.debug("startTime = " + startTime + ", endTime = " + endTime);
FetchRequest fetchRequest = rrdDb.createFetchRequest(ConsolFun.TOTAL, startTime, endTime);
FetchData fetchData = fetchRequest.fetchData();
double[] values = fetchData.getValues(dataSourceName);
assertThat(values.length, is(numRrdStepIterations));
logFetchData(fetchData, "TOTAL");
fetchRequest = rrdDb.createFetchRequest(ConsolFun.AVERAGE, startTime, endTime);
fetchData = fetchRequest.fetchData();
values = fetchData.getValues(dataSourceName);
assertThat(values.length, is(numRrdStepIterations));
logFetchData(fetchData, "AVERAGE");
fetchRequest = rrdDb.createFetchRequest(ConsolFun.MIN, startTime, endTime);
fetchData = fetchRequest.fetchData();
values = fetchData.getValues(dataSourceName);
assertThat(values.length, is(numRrdStepIterations));
logFetchData(fetchData, "MIN");
fetchRequest = rrdDb.createFetchRequest(ConsolFun.MAX, startTime, endTime);
fetchData = fetchRequest.fetchData();
values = fetchData.getValues(dataSourceName);
assertThat(values.length, is(numRrdStepIterations));
logFetchData(fetchData, "MAX");
}
@Test
@Ignore
public void testRrdFileCreationWhenRrdFileAlreadyExists() throws Exception {
// Set sample rate to 1 sec (default is 60 seconds) so that unit test runs quickly
int sampleRate = 1;
createJmxCollector("Uptime", JmxCollector.COUNTER_DATA_SOURCE_TYPE, sampleRate);
String rrdFilename1 = jmxCollector.getRrdPath();
assertThat(rrdFilename1, is(TEST_DIR + rrdPath));
rrdDb = new RrdDb(rrdFilename1);
assertThat(rrdDb, not(nullValue()));
assertThat(rrdDb.isClosed(), is(false));
// Attempt to create again
LOGGER.debug("Creating JmxCollector again ...");
dataSourceName = "uptime";
rrdPath = dataSourceName + ".rrd";
JmxCollector jmxCollector2 = new JmxCollector();
jmxCollector2.setMbeanName("java.lang:type=Runtime");
jmxCollector2.setMbeanAttributeName("Uptime");
jmxCollector2.setRrdPath(rrdPath);
jmxCollector2.setRrdDataSourceName(dataSourceName);
jmxCollector2.setRrdDataSourceType(JmxCollector.COUNTER_DATA_SOURCE_TYPE);
jmxCollector2.setSampleRate(sampleRate);
jmxCollector2.setMetricsDir(TEST_DIR);
// Simulates what Spring beans container would do
jmxCollector2.configureCollector();
// Verify the 2 JMX Collectors are using the same RRD file
String rrdFilename2 = jmxCollector2.getRrdPath();
assertThat(rrdFilename2, is(TEST_DIR + rrdPath));
assertThat(rrdFilename1, equalTo(rrdFilename2));
jmxCollector2.destroy();
}
@Test
public void testConstruction() {
JmxCollector jmxCollector = new JmxCollector();
assertThat(jmxCollector, not(nullValue()));
assertThat(jmxCollector.getRrdDataSourceType(), is(JmxCollector.COUNTER_DATA_SOURCE_TYPE));
}
