use of com.linkedin.databus.core.monitoring.mbean.DbusEventStatsCollectorsPartitioner in project databus by linkedin.
the class TestDbusEventStatsCollectorPartitioner method testStatsCollectorPartitionerMultipleDbs.
@Test
public void testStatsCollectorPartitionerMultipleDbs() {
DbusEventStatsCollectorsPartitioner collector = new DbusEventStatsCollectorsPartitioner(1, ":inbound", null);
TestStatsCollectorCallback callback = new TestStatsCollectorCallback();
collector.registerStatsCallback(callback);
// Add 2 collectors for DB1
PhysicalPartition p1 = new PhysicalPartition(1, "db1");
PhysicalPartition p2 = new PhysicalPartition(2, "db1");
DbusEventsStatisticsCollector c1 = new DbusEventsStatisticsCollector(1, "db1:1", true, false, null);
DbusEventsStatisticsCollector c2 = new DbusEventsStatisticsCollector(1, "db1:2", true, false, null);
collector.addStatsCollector(p1, c1);
collector.addStatsCollector(p2, c2);
// Add 2 collectors for DB2
PhysicalPartition p3 = new PhysicalPartition(1, "db2");
PhysicalPartition p4 = new PhysicalPartition(2, "db2");
DbusEventsStatisticsCollector c3 = new DbusEventsStatisticsCollector(1, "db2:1", true, false, null);
DbusEventsStatisticsCollector c4 = new DbusEventsStatisticsCollector(1, "db2:2", true, false, null);
collector.addStatsCollector(p3, c3);
collector.addStatsCollector(p4, c4);
// Add 2 collectors for DB3
PhysicalPartition p5 = new PhysicalPartition(3, "db3");
PhysicalPartition p6 = new PhysicalPartition(4, "db3");
DbusEventsStatisticsCollector c5 = new DbusEventsStatisticsCollector(1, "db3:3", true, false, null);
DbusEventsStatisticsCollector c6 = new DbusEventsStatisticsCollector(1, "db3:4", true, false, null);
collector.addStatsCollector(p5, c5);
collector.addStatsCollector(p6, c6);
StatsWriter w1 = new StatsWriter(c1);
StatsWriter w2 = new StatsWriter(c2);
StatsWriter w3 = new StatsWriter(c3);
StatsWriter w4 = new StatsWriter(c4);
StatsWriter w5 = new StatsWriter(c5);
StatsWriter w6 = new StatsWriter(c6);
w1.addEvents(2, 2, 100);
w2.addEvents(2, 2, 200);
w3.addEvents(3, 2, 300);
w4.addEvents(3, 2, 400);
w5.addEvents(4, 2, 500);
w6.addEvents(4, 2, 600);
// Verify DB1 collector
StatsCollectors<DbusEventsStatisticsCollector> col = collector.getDBStatsCollector("db1");
Assert.assertNotNull(col);
col.mergeStatsCollectors();
LOG.info("Merged Stats : " + col.getStatsCollector().getTotalStats());
LOG.info("C1 Stats : " + c1.getTotalStats());
LOG.info("C2 Stats : " + c2.getTotalStats());
DbusEventsTotalStats s = col.getStatsCollector().getTotalStats();
Assert.assertEquals("Total Events", 8, s.getNumDataEvents());
Assert.assertEquals("Sys Events", 4, s.getNumSysEvents());
Assert.assertEquals("Min Scn", 101, s.getMinScn());
Assert.assertEquals("Max Scn", 205, s.getMaxScn());
// Verify DB2 collector
col = collector.getDBStatsCollector("db2");
Assert.assertNotNull(col);
col.mergeStatsCollectors();
LOG.info("Merged Stats : " + col.getStatsCollector().getTotalStats());
LOG.info("C3 Stats : " + c3.getTotalStats());
LOG.info("C4 Stats : " + c4.getTotalStats());
s = col.getStatsCollector().getTotalStats();
Assert.assertEquals("Total Events", 12, s.getNumDataEvents());
Assert.assertEquals("Sys Events", 4, s.getNumSysEvents());
Assert.assertEquals("Min Scn", 301, s.getMinScn());
Assert.assertEquals("Max Scn", 407, s.getMaxScn());
// Verify DB3 collector
col = collector.getDBStatsCollector("db3");
Assert.assertNotNull(col);
col.mergeStatsCollectors();
LOG.info("Merged Stats : " + col.getStatsCollector().getTotalStats());
LOG.info("C3 Stats : " + c5.getTotalStats());
LOG.info("C4 Stats : " + c6.getTotalStats());
s = col.getStatsCollector().getTotalStats();
Assert.assertEquals("Total Events", 16, s.getNumDataEvents());
Assert.assertEquals("Sys Events", 4, s.getNumSysEvents());
Assert.assertEquals("Min Scn", 501, s.getMinScn());
Assert.assertEquals("Max Scn", 609, s.getMaxScn());
Assert.assertEquals("Num Stats Callback", 3, callback.getCollectorsAddedList().size());
collector.removeAllStatsCollector();
Assert.assertEquals("Num Stats Callback", 3, callback.getCollectorsRemovedList().size());
}
use of com.linkedin.databus.core.monitoring.mbean.DbusEventStatsCollectorsPartitioner in project databus by linkedin.
the class TestDbusEventStatsCollectorPartitioner method testStatsCollectorPartitionerSingleDb.
@Test
public void testStatsCollectorPartitionerSingleDb() {
DbusEventStatsCollectorsPartitioner collector = new DbusEventStatsCollectorsPartitioner(1, ":inbound", null);
TestStatsCollectorCallback callback = new TestStatsCollectorCallback();
collector.registerStatsCallback(callback);
// Add 2 collectors for DB1
PhysicalPartition p1 = new PhysicalPartition(1, "db1");
PhysicalPartition p2 = new PhysicalPartition(2, "db1");
DbusEventsStatisticsCollector c1 = new DbusEventsStatisticsCollector(1, "db1:1", true, false, null);
DbusEventsStatisticsCollector c2 = new DbusEventsStatisticsCollector(1, "db1:2", true, false, null);
collector.addStatsCollector(p1, c1);
collector.addStatsCollector(p2, c2);
StatsWriter w1 = new StatsWriter(c1);
StatsWriter w2 = new StatsWriter(c2);
w1.addEvents(2, 2, 100);
w2.addEvents(2, 2, 200);
StatsCollectors<DbusEventsStatisticsCollector> col = collector.getDBStatsCollector("db1");
Assert.assertNotNull(col);
col.mergeStatsCollectors();
LOG.info("Merged Stats : " + col.getStatsCollector().getTotalStats());
LOG.info("C1 Stats : " + c1.getTotalStats());
LOG.info("C2 Stats : " + c2.getTotalStats());
DbusEventsTotalStats s = col.getStatsCollector().getTotalStats();
Assert.assertEquals("Total Events", 8, s.getNumDataEvents());
Assert.assertEquals("Sys Events", 4, s.getNumSysEvents());
Assert.assertEquals("Min Scn", 101, s.getMinScn());
Assert.assertEquals("Max Scn", 205, s.getMaxScn());
Assert.assertEquals("Num Stats Callback", 1, callback.getCollectorsAddedList().size());
collector.removeAllStatsCollector();
Assert.assertEquals("Num Stats Callback", 1, callback.getCollectorsRemovedList().size());
}
Aggregations