private void doPopulateDatabase() {
LOG.debug("==== DatabasePopulator Starting ====");
final OnmsDistPoller distPoller = getDistPoller("localhost", "127.0.0.1");
final NetworkBuilder builder = new NetworkBuilder(distPoller);
final OnmsNode node1 = buildNode1(builder);
getNodeDao().save(node1);
getNodeDao().flush();
OnmsNode node2 = buildNode2(builder);
getNodeDao().save(node2);
getNodeDao().flush();
setNode2(node2);
OnmsNode node3 = buildNode3(builder);
getNodeDao().save(node3);
getNodeDao().flush();
setNode3(node3);
OnmsNode node4 = buildNode4(builder);
getNodeDao().save(node4);
getNodeDao().flush();
setNode4(node4);
OnmsNode node5 = buildNode5(builder);
getNodeDao().save(node5);
getNodeDao().flush();
setNode5(node5);
OnmsNode node6 = buildNode6(builder);
getNodeDao().save(node6);
getNodeDao().flush();
setNode6(node6);
final OnmsEvent event = buildEvent(distPoller);
getEventDao().save(event);
getEventDao().flush();
final OnmsNotification notif = buildTestNotification(builder, event);
getNotificationDao().save(notif);
getNotificationDao().flush();
final OnmsUserNotification userNotif = buildTestUserNotification(notif);
getUserNotificationDao().save(userNotif);
getUserNotificationDao().flush();
final OnmsUserNotification userNotif2 = buildTestUser2Notification(notif);
getUserNotificationDao().save(userNotif2);
getUserNotificationDao().flush();
final OnmsMonitoredService svc =
getMonitoredServiceDao().get(node1.getId(), InetAddressUtils.addr("192.168.1.1"), "SNMP");
final OnmsOutage resolved =
new OnmsOutage(new Date(), new Date(), event, event, svc, null, null);
getOutageDao().save(resolved);
getOutageDao().flush();
final OnmsOutage unresolved = new OnmsOutage(new Date(), event, svc);
getOutageDao().save(unresolved);
getOutageDao().flush();
final OnmsAlarm alarm = buildAlarm(event);
getAlarmDao().save(alarm);
getAlarmDao().flush();
final OnmsMap map = new OnmsMap("DB_Pop_Test_Map", "admin");
map.setBackground("fake_background.jpg");
map.setAccessMode(OnmsMap.ACCESS_MODE_ADMIN);
map.setType(OnmsMap.USER_GENERATED_MAP);
map.setMapGroup("admin");
getOnmsMapDao().save(map);
getOnmsMapDao().flush();
final OnmsMapElement mapElement =
new OnmsMapElement(
map, 1, OnmsMapElement.NODE_TYPE, "Test Node", OnmsMapElement.defaultNodeIcon, 0, 10);
getOnmsMapElementDao().save(mapElement);
getOnmsMapElementDao().flush();
final DataLinkInterface dli =
new DataLinkInterface(node1, 1, node1.getId(), 1, StatusType.ACTIVE, new Date());
getDataLinkInterfaceDao().save(dli);
getDataLinkInterfaceDao().flush();
final DataLinkInterface dli2 =
new DataLinkInterface(node1, 2, node1.getId(), 1, StatusType.ACTIVE, new Date());
getDataLinkInterfaceDao().save(dli2);
getDataLinkInterfaceDao().flush();
final DataLinkInterface dli3 =
new DataLinkInterface(node2, 1, node1.getId(), 1, StatusType.ACTIVE, new Date());
getDataLinkInterfaceDao().save(dli3);
getDataLinkInterfaceDao().flush();
final OnmsAcknowledgment ack = new OnmsAcknowledgment();
ack.setAckTime(new Date());
ack.setAckType(AckType.UNSPECIFIED);
ack.setAckAction(AckAction.UNSPECIFIED);
ack.setAckUser("admin");
getAcknowledgmentDao().save(ack);
getAcknowledgmentDao().flush();
final OnmsMonitoringLocationDefinition def = new OnmsMonitoringLocationDefinition();
def.setName("RDU");
def.setArea("East Coast");
def.setPollingPackageName("example1");
def.setGeolocation("Research Triangle Park, NC");
def.setCoordinates("35.715751,-79.16262");
def.setPriority(1L);
m_locationMonitorDao.saveMonitoringLocationDefinition(def);
LOG.debug("= DatabasePopulatorExtension Populate Starting =");
for (Extension eachExtension : extensions) {
DaoSupport daoSupport = eachExtension.getDaoSupport();
OnmsDao dao = daoSupport != null ? daoSupport.getDao() : null;
Class<? super OnmsDao> daoClass = daoSupport != null ? daoSupport.getDaoClass() : null;
registerDao(daoClass, dao);
dao = lookupDao(daoClass);
eachExtension.onPopulate(this, dao);
if (dao != null) {
dao.flush();
}
}
LOG.debug("= DatabasePopulatorExtension Populate Finished =");
LOG.debug("==== DatabasePopulator Finished ====");
}
public void resetDatabase() {
LOG.debug("==== DatabasePopulator Reset ====");
for (final DataLinkInterface iface : m_dataLinkInterfaceDao.findAll()) {
m_dataLinkInterfaceDao.delete(iface);
}
for (final OnmsOutage outage : m_outageDao.findAll()) {
m_outageDao.delete(outage);
}
for (final OnmsUserNotification not : m_userNotificationDao.findAll()) {
m_userNotificationDao.delete(not);
}
for (final OnmsNotification not : m_notificationDao.findAll()) {
m_notificationDao.delete(not);
}
for (final OnmsAlarm alarm : m_alarmDao.findAll()) {
m_alarmDao.delete(alarm);
}
for (final OnmsEvent event : m_eventDao.findAll()) {
m_eventDao.delete(event);
}
for (final OnmsSnmpInterface iface : m_snmpInterfaceDao.findAll()) {
m_snmpInterfaceDao.delete(iface);
}
for (final OnmsIpInterface iface : m_ipInterfaceDao.findAll()) {
m_ipInterfaceDao.delete(iface);
}
for (final OnmsNode node : m_nodeDao.findAll()) {
m_nodeDao.delete(node);
}
for (final OnmsServiceType service : m_serviceTypeDao.findAll()) {
m_serviceTypeDao.delete(service);
}
LOG.debug("= DatabasePopulatorExtension Reset Starting =");
for (Extension eachExtension : extensions) {
DaoSupport daoSupport = eachExtension.getDaoSupport();
OnmsDao dao =
daoSupport != null && daoSupport.getDaoClass() != null
? lookupDao(daoSupport.getDaoClass())
: null;
eachExtension.onShutdown(this, dao);
if (dao != null) {
dao.flush();
}
}
LOG.debug("= DatabasePopulatorExtension Reset Finished =");
m_dataLinkInterfaceDao.flush();
m_outageDao.flush();
m_userNotificationDao.flush();
m_notificationDao.flush();
m_alarmDao.flush();
m_eventDao.flush();
m_snmpInterfaceDao.flush();
m_ipInterfaceDao.flush();
m_nodeDao.flush();
m_serviceTypeDao.flush();
LOG.debug("==== DatabasePopulator Reset Finished ====");
}
