//
// This file is part of the OpenNMS(R) Application.
//
// OpenNMS(R) is Copyright (C) 2002-2003 The OpenNMS Group, Inc. All rights reserved.
// OpenNMS(R) is a derivative work, containing both original code, included code and modified
// code that was published under the GNU General Public License. Copyrights for modified
// and included code are below.
//
// OpenNMS(R) is a registered trademark of The OpenNMS Group, Inc.
//
// Modifications:
//
// 2008 Mar 20: Remove uncommented code (particularly System.out.println) and
// de-nest a block. - dj@opennms.org
// 2007 May 12: Reorganize, use Java 5 generics and loops. - dj@opennms.org
//
// Copyright (C) 1999-2001 Oculan Corp. All rights reserved.
//
// This program is free software; you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation; either version 2 of the License, or
// (at your option) any later version.
//
// This program is distributed in the hope that it will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
// GNU General Public License for more details.
//
// You should have received a copy of the GNU General Public License
// along with this program; if not, write to the Free Software
// Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
//
// For more information contact:
// OpenNMS Licensing <license@opennms.org>
// http://www.opennms.org/
// http://www.opennms.com/
//
package org.opennms.web.admin.pollerConfig;
import java.io.FileInputStream;
import java.io.FileOutputStream;
import java.io.FileWriter;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Collection;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import java.util.ListIterator;
import java.util.Map;
import java.util.Properties;
import javax.servlet.ServletConfig;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServlet;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.apache.log4j.Category;
import org.exolab.castor.xml.MarshalException;
import org.exolab.castor.xml.Marshaller;
import org.exolab.castor.xml.ValidationException;
import org.opennms.core.utils.ThreadCategory;
import org.opennms.netmgt.ConfigFileConstants;
import org.opennms.netmgt.config.CapsdConfig;
import org.opennms.netmgt.config.CapsdConfigFactory;
import org.opennms.netmgt.config.PollerConfig;
import org.opennms.netmgt.config.PollerConfigFactory;
import org.opennms.netmgt.config.capsd.CapsdConfiguration;
import org.opennms.netmgt.config.capsd.ProtocolPlugin;
import org.opennms.netmgt.config.poller.Monitor;
import org.opennms.netmgt.config.poller.PollerConfiguration;
import org.opennms.netmgt.config.poller.Service;
/**
* A servlet that handles managing or unmanaging interfaces and services on a
* node
*
* @author <A HREF="mailto:jacinta@opennms.org">Jacinta Remedios </A>
* @author <A HREF="http://www.opennms.org/">OpenNMS </A>
*/
public class PollerConfigServlet extends HttpServlet {
private static final long serialVersionUID = 1L;
private PollerConfiguration m_pollerConfig = null;
private CapsdConfiguration m_capsdConfig = null;
protected String m_redirectSuccess;
private Map<String, Service> m_pollerServices = new HashMap<String, Service>();
private Map<String, ProtocolPlugin> m_capsdProtocols = new HashMap<String, ProtocolPlugin>();
private List<ProtocolPlugin> m_capsdColl = new ArrayList<ProtocolPlugin>();
private org.opennms.netmgt.config.poller.Package m_pkg = null;
private List<ProtocolPlugin> m_pluginColl = null;
private Properties m_props = new Properties();
private PollerConfig m_pollerFactory = null;
private CapsdConfig m_capsdFactory = null;
public void init() throws ServletException {
getInitParameters();
loadPollerConfProperties();
initPollerConfigFactory();
initCapsdConfigFactory();
initPollerServices();
initCapsdProtocols();
}
private void getInitParameters() throws ServletException {
ServletConfig config = getServletConfig();
m_redirectSuccess = config.getInitParameter("redirect.success");
if (m_redirectSuccess == null) {
throw new ServletException("Missing required init parameter: redirect.success");
}
}
private void initCapsdConfigFactory() throws ServletException {
try {
CapsdConfigFactory.init();
} catch (Exception e) {
throw new ServletException(e);
}
m_capsdFactory = CapsdConfigFactory.getInstance();
m_capsdConfig = m_capsdFactory.getConfiguration();
if (m_capsdConfig == null) {
throw new ServletException("Capsd Configuration file is empty");
}
}
private void initPollerConfigFactory() throws ServletException {
try {
PollerConfigFactory.init();
} catch (Exception e) {
throw new ServletException(e);
}
m_pollerFactory = PollerConfigFactory.getInstance();
m_pollerConfig = m_pollerFactory.getConfiguration();
if (m_pollerConfig == null) {
throw new ServletException("Poller Configuration file is empty");
}
}
private void loadPollerConfProperties() throws ServletException {
try {
m_props.load(new FileInputStream(ConfigFileConstants.getFile(ConfigFileConstants.POLLER_CONF_FILE_NAME)));
} catch (Exception e) {
throw new ServletException(e);
}
//String[] protocols = BundleLists.parseBundleList(m_props.getProperty("services"));
}
public void reloadFiles() throws ServletException {
ServletConfig config = getServletConfig();
try {
loadPollerConfProperties();
initPollerConfigFactory();
initCapsdConfigFactory();
} catch (Exception e) {
throw new ServletException(e);
}
initPollerServices();
initCapsdProtocols();
m_redirectSuccess = config.getInitParameter("redirect.success");
if (m_redirectSuccess == null) {
throw new ServletException("Missing required init parameter: redirect.success");
}
}
public void initCapsdProtocols() {
m_pluginColl = getCapsdProtocolPlugins();
if (m_pluginColl != null) {
Iterator<ProtocolPlugin> pluginiter = m_pluginColl.iterator();
while (pluginiter.hasNext()) {
ProtocolPlugin plugin = pluginiter.next();
m_capsdColl.add(plugin);
m_capsdProtocols.put(plugin.getProtocol(), plugin);
}
}
}
@SuppressWarnings("unchecked")
private List<ProtocolPlugin> getCapsdProtocolPlugins() {
return (List<ProtocolPlugin>) m_capsdConfig.getProtocolPluginCollection();
}
public void initPollerServices() {
Collection packageColl = m_pollerConfig.getPackageCollection();
if (packageColl != null) {
Iterator pkgiter = packageColl.iterator();
if (pkgiter.hasNext()) {
m_pkg = (org.opennms.netmgt.config.poller.Package) pkgiter.next();
Collection svcColl = m_pkg.getServiceCollection();
Iterator svcIter = svcColl.iterator();
Service svcProp = null;
while (svcIter.hasNext()) {
svcProp = (Service) svcIter.next();
m_pollerServices.put(svcProp.getName(), svcProp);
}
}
}
}
public void doPost(HttpServletRequest request, HttpServletResponse response) throws ServletException, IOException {
reloadFiles();
List<String> checkedList = new ArrayList<String>();
List<String> deleteList = new ArrayList<String>();
m_props.store(new FileOutputStream(ConfigFileConstants.getFile(ConfigFileConstants.POLLER_CONF_FILE_NAME)), null);
String[] requestActivate = request.getParameterValues("activate");
String[] requestDelete = request.getParameterValues("delete");
if (requestActivate != null) {
for (int i = 0; i < requestActivate.length; i++) {
modifyPollerInfo("on", requestActivate[i]);
checkedList.add(requestActivate[i]);
}
}
if (requestDelete != null) {
for (int j = 0; j < requestDelete.length; j++) {
deleteList.add(requestDelete[j]);
}
}
adjustNonChecked(checkedList);
deleteThese(deleteList);
FileWriter poller_fileWriter = new FileWriter(ConfigFileConstants.getFile(ConfigFileConstants.POLLER_CONFIG_FILE_NAME));
FileWriter capsd_fileWriter = new FileWriter(ConfigFileConstants.getFile(ConfigFileConstants.CAPSD_CONFIG_FILE_NAME));
try {
Marshaller.marshal(m_pollerConfig, poller_fileWriter);
Marshaller.marshal(m_capsdConfig, capsd_fileWriter);
} catch (MarshalException e) {
log().error("Could not marshal config object when writing config file: " + e, e);
throw new ServletException(e);
} catch (ValidationException e) {
log().error("Could not validate config object when writing config file: " + e, e);
throw new ServletException(e);
}
String redirectPage = request.getParameter("redirect");
if (redirectPage == null) {
redirectPage = m_redirectSuccess;
}
response.sendRedirect(redirectPage);
}
public void deleteCapsdInfo(String name) {
if (m_capsdProtocols.get(name) != null) {
ProtocolPlugin tmpproto = (ProtocolPlugin) m_capsdProtocols.get(name);
m_capsdProtocols.remove(name);
m_pluginColl = new ArrayList<ProtocolPlugin>(m_capsdProtocols.values());
m_capsdColl.remove(tmpproto);
m_capsdConfig.setProtocolPluginCollection(new ArrayList<ProtocolPlugin>(m_pluginColl));
}
}
public void adjustNonChecked(List<String> checkedList) {
if (m_pkg != null) {
Collection svcColl = m_pkg.getServiceCollection();
Service svc = null;
if (svcColl != null) {
Iterator svcIter = svcColl.iterator();
while (svcIter.hasNext()) {
svc = (Service) svcIter.next();
if (svc != null) {
if (!checkedList.contains(svc.getName())) {
if (svc.getStatus().equals("on")) {
svc.setStatus("off");
}
}
}
}
}
}
}
public void deleteThese(List<String> deleteServices) throws IOException {
ListIterator<String> lstIter = deleteServices.listIterator();
while (lstIter.hasNext()) {
String svcname = lstIter.next();
if (m_pkg != null) {
boolean flag = false;
Collection svcColl = m_pkg.getServiceCollection();
if (svcColl != null) {
Iterator svcIter = svcColl.iterator();
Service svc = null;
while (svcIter.hasNext()) {
svc = (Service) svcIter.next();
if (svc != null) {
if (svc.getName().equals(svcname)) {
flag = true;
break;
}
}
}
if (flag) {
m_pkg.removeService(svc);
log().info("Package removed " + svc.getName());
removeMonitor(svc.getName());
deleteCapsdInfo(svc.getName());
m_props.remove("service." + svc.getName() + ".protocol");
m_props.store(new FileOutputStream(ConfigFileConstants.getFile(ConfigFileConstants.POLLER_CONF_FILE_NAME)), null);
}
}
}
}
}
public void removeMonitor(String service) {
// Add the new monitor with the protocol.
Collection monitorColl = m_pollerConfig.getMonitorCollection();
Monitor newMonitor = new Monitor();
if (monitorColl != null) {
Iterator monitoriter = monitorColl.iterator();
while (monitoriter.hasNext()) {
Monitor mon = (Monitor) monitoriter.next();
if (mon != null) {
if (mon.getService().equals(service)) {
newMonitor.setService(service);
newMonitor.setClassName(mon.getClassName());
newMonitor.setParameterCollection(mon.getParameterCollection());
break;
}
}
}
monitorColl.remove(newMonitor);
}
}
public void modifyPollerInfo(String bPolled, String protocol) {
if (m_pkg != null) {
Collection svcColl = m_pkg.getServiceCollection();
if (svcColl != null) {
Iterator svcIter = svcColl.iterator();
while (svcIter.hasNext()) {
Service svc = (Service) svcIter.next();
if (svc != null) {
if (svc.getName().equals(protocol)) {
svc.setStatus(bPolled);
break;
}
}
}
}
}
}
/**
* @return logger for this servlet
*/
private Category log() {
return ThreadCategory.getInstance(getClass());
}
}
|