responsetimemodel.java

来自「opennms得相关源码 请大家看看」· Java 代码 · 共 502 行 · 第 1/2 页

//
// 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:
//
// 12 Nov 2002: Added response time reports to webUI.
//
// Original coda base 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       <[e-mail]>
//      http://www.opennms.org/
//      http://www.opennms.com/
//

package org.opennms.web.response;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Statement;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collection;
import java.util.HashSet;
import java.util.Iterator;
import java.util.LinkedList;
import java.util.List;
import java.util.Map;
import java.util.Properties;
import java.util.Set;

import org.opennms.core.resource.Vault;
import org.opennms.core.utils.IntSet;
import org.opennms.netmgt.utils.IfLabel;
import org.opennms.netmgt.utils.RrdFileConstants;
import org.opennms.web.Util;
import org.opennms.web.graph.PrefabGraph;

/**
 * Encapsulates all SNMP performance reporting for the web user interface.
 * 
 * @author <a href="mailto:[e-mail]">Tarus Balog </a>
 * @author <a href="mailto:[e-mail]">Seth Leger </a>
 * @author <a href="mailto:[e-mail]">Lawrence Karnowski </a>
 * @author <a href="http://www.opennms.org">OpenNMS </a>
 */
public class ResponseTimeModel extends Object {
    public static final String RRDTOOL_GRAPH_PROPERTIES_FILENAME = "/etc/response-graph.properties";

    public static final String INTERFACE_GRAPH_TYPE = "interface";

    public static final String NODE_GRAPH_TYPE = "node";

    protected Properties props;

    protected PrefabGraph[] queries;

    protected Map reportMap;

    protected File rrdDirectory;

    protected String infoCommand;

    /**
     * Create a new instance.
     * 
     * @param homeDir
     *            the OpenNMS home directory, see {@link Vault#getHomeDir
     *            Vault.getHomeDir}.
     */
    public ResponseTimeModel(String homeDir) throws IOException {
        if (homeDir == null) {
            throw new IllegalArgumentException("Cannot take null parameters.");
        }

        this.props = new java.util.Properties();
        this.props.load(new FileInputStream(homeDir + RRDTOOL_GRAPH_PROPERTIES_FILENAME));

        this.rrdDirectory = new File(this.props.getProperty("command.input.dir"));
        this.infoCommand = this.props.getProperty("info.command");

        this.reportMap = PrefabGraph.getPrefabGraphDefinitions(props);
    }

    public File getRrdDirectory() {
        return this.rrdDirectory;
    }

    public PrefabGraph getQuery(String queryName) {
        return (PrefabGraph) this.reportMap.get(queryName);
    }

    /**
     * Return a list of all known prefabricated graph definitions.
     */
    public PrefabGraph[] getQueries() {
        if (this.queries == null) {
            Collection values = this.reportMap.values();
            Iterator iter = values.iterator();

            PrefabGraph[] graphs = new PrefabGraph[values.size()];

            for (int i = 0; i < graphs.length; i++) {
                graphs[i] = (PrefabGraph) iter.next();
            }

            this.queries = graphs;
        }

        return (this.queries);
    }

    public PrefabGraph[] getQueries(int nodeId) {
        return this.getQueries(String.valueOf(nodeId));
    }

    public PrefabGraph[] getQueries(String nodeId) {
        if (nodeId == null) {
            throw new IllegalArgumentException("Cannot take null parameters.");
        }

        // create a temporary list of queries to return
        ArrayList returnList = new ArrayList();

        // get the full list of all possible queries
        PrefabGraph[] queries = this.getQueries();

        // get all the data sources supported by node
        List availDataSourceList = this.getDataSourceList(nodeId);

        // for each query, see if all the required data sources are available
        // in the available data source list, if so, add that query to the
        // returnList
        for (int i = 0; i < queries.length; i++) {
            List requiredList = Arrays.asList(queries[i].getColumns());

            if (availDataSourceList.containsAll(requiredList)) {
                returnList.add(queries[i]);
            }
        }

        // put the queries in returnList into an array
        PrefabGraph[] availQueries = (PrefabGraph[]) returnList.toArray(new PrefabGraph[returnList.size()]);

        return availQueries;
    }

    public PrefabGraph[] getQueries(int nodeId, String intf, boolean includeNodeQueries) {
        return this.getQueries(String.valueOf(nodeId), intf, includeNodeQueries);
    }

    public PrefabGraph[] getQueries(String nodeId, String intf, boolean includeNodeQueries) {
        if (nodeId == null || intf == null) {
            throw new IllegalArgumentException("Cannot take null parameters.");
        }

        // create a temporary list of queries to return
        ArrayList returnList = new ArrayList();

        // get the full list of all possible queries
        PrefabGraph[] queries = this.getQueries();

        // get all the data sources supported by this interface (and possibly
        // node)
        List availDataSourceList = this.getDataSourceList(nodeId, intf, includeNodeQueries);

        // for each query, see if all the required data sources are available
        // in the available data source list, if so, add that query to the
        // returnList
        for (int i = 0; i < queries.length; i++) {
            List requiredList = Arrays.asList(queries[i].getColumns());

            if (availDataSourceList.containsAll(requiredList)) {
                returnList.add(queries[i]);
            }
        }

        // put the queries in returnList into an array
        PrefabGraph[] availQueries = (PrefabGraph[]) returnList.toArray(new PrefabGraph[returnList.size()]);

        return availQueries;
    }

    public String[] getDataSources(int nodeId) {
        return this.getDataSources(String.valueOf(nodeId));
    }

    public String[] getDataSources(String nodeId) {
        List dataSourceList = this.getDataSourceList(String.valueOf(nodeId));
        String[] dataSources = (String[]) dataSourceList.toArray(new String[dataSourceList.size()]);

        return dataSources;
    }

    public String[] getDataSources(int nodeId, String intf, boolean includeNodeQueries) {
        return this.getDataSources(String.valueOf(nodeId), intf, includeNodeQueries);
    }

    public String[] getDataSources(String nodeId, String intf, boolean includeNodeQueries) {
        List dataSourceList = this.getDataSourceList(String.valueOf(nodeId), intf, includeNodeQueries);
        String[] dataSources = (String[]) dataSourceList.toArray(new String[dataSourceList.size()]);

        return dataSources;
    }

    public List getDataSourceList(int nodeId) {
        return this.getDataSourceList(String.valueOf(nodeId));
    }

    public List getDataSourceList(String nodeId) {
        if (nodeId == null) {
            throw new IllegalArgumentException("Cannot take null parameters.");
        }

        ArrayList dataSources = new ArrayList();
        File nodeDir = new File(this.rrdDirectory, nodeId);
        int suffixLength = RrdFileConstants.RRD_SUFFIX.length();

        // get the node data sources
        File[] nodeFiles = nodeDir.listFiles(RrdFileConstants.RRD_FILENAME_FILTER);

        for (int i = 0; i < nodeFiles.length; i++) {
            String fileName = nodeFiles[i].getName();
            String dsName = fileName.substring(0, fileName.length() - suffixLength);

            dataSources.add(dsName);