modulemonhandler.java

Network Administration Visualized 网络管理可视化源码

package no.ntnu.nav.getDeviceData.dataplugins.ModuleMon;

import java.util.*;
import java.sql.*;

import no.ntnu.nav.logger.*;
import no.ntnu.nav.Database.*;
import no.ntnu.nav.util.*;
import no.ntnu.nav.event.*;
import no.ntnu.nav.getDeviceData.Netbox;
import no.ntnu.nav.getDeviceData.dataplugins.*;

/**
 * DataHandler plugin for getDeviceData; provides an interface for storing
 * info about which modules of a switch/router is responding to requests.
 *
 * @see ModuleMonContainer
 */

public class ModuleMonHandler implements DataHandler {

	private static Map moduleMap;
	private static Map modidMap;
	private static MultiMap modules;
	
	private static MultiMap queryIfindices;
	private static Map moduleToIfindex;

	private static Set modulesDown = new HashSet();
	

	/**
	 * Fetch initial data from swport table.
	 */
	public synchronized void init(Map persistentStorage, Map changedDeviceids) {
		/*
		boolean onlyUpdate = true;
		for (Iterator it = changedDeviceids.values().iterator(); it.hasNext() && onlyUpdate;) {
			if (((Integer)it.next()).intValue() != DataHandler.DEVICE_UPDATED) onlyUpdate = false;
		}

		if (persistentStorage.containsKey("initDone") && (changedDeviceids.isEmpty() || onlyUpdate)) return;
		persistentStorage.put("initDone", null);
		*/
		if (persistentStorage.containsKey("initDone")) return;
		persistentStorage.put("initDone", null);

		Log.setDefaultSubsystem("ModuleMonHandler");

		/*
		synchronized (modulesDown) {
			int oldcnt = modidMap != null ? modidMap.size() : 0;

			try {
				// We need to the mapping from netboxid:ifindex -> module and the modules belonging to each netbox
				Map moduleMapL = Collections.synchronizedMap(new HashMap());
				Map modidMapL = Collections.synchronizedMap(new HashMap());
				MultiMap modulesL = new HashMultiMap();
				MultiMap queryIfindicesL = new HashMultiMap();
				Map moduleToIfindexL = Collections.synchronizedMap(new HashMap());
				Set queryDupe = new HashSet();
				Set nullPortSet = new HashSet();
				ResultSet rs = Database.query("SELECT deviceid,netboxid,ifindex,moduleid,module,swportid,port FROM module LEFT JOIN swport USING(moduleid) ORDER BY RANDOM()");
				while (rs.next()) {
					String netboxid = rs.getString("netboxid");
					moduleMapL.put(rs.getString("netboxid")+":"+rs.getString("module"), rs.getString("moduleid"));
					modidMapL.put(rs.getString("moduleid"), rs.getString("deviceid"));
					modulesL.put(rs.getString("netboxid"), rs.getString("moduleid"));

					if (rs.getString("swportid") != null) {
						String k = rs.getString("netboxid")+":"+rs.getString("moduleid");
						if (rs.getString("port") == null && !queryDupe.contains(k)) {
							//System.err.println("add null: " + k);
							nullPortSet.add(k);
						} else {
							if (nullPortSet.remove(k)) {
								//System.err.println("remove dupe: " + k);
								queryDupe.remove(k);
							}
						}
						if (queryDupe.add(k)) {
							//System.err.println("add final: " + rs.getString("ifindex") + ", " + rs.getString("port"));
							queryIfindicesL.put(rs.getString("netboxid"), new String[] { rs.getString("ifindex"), rs.getString("module") });
							
							Map mm;
							if ( (mm=(Map)moduleToIfindexL.get(netboxid)) == null) moduleToIfindexL.put(netboxid, mm = new HashMap());
							mm.put(rs.getString("module"), rs.getString("ifindex"));
						}
					}
				}

				rs = Database.query("SELECT netboxid,moduleid FROM module WHERE up='n'");
				while (rs.next()) {
					String key = rs.getString("netboxid")+":"+rs.getString("moduleid");
					modulesDown.add(key);
				}

				modidMap = modidMapL;
				moduleMap = moduleMapL;
				modules = modulesL;
				queryIfindices = queryIfindicesL;
				moduleToIfindex = moduleToIfindexL;
				Log.d("INIT", "Fetched " + modidMap.size() + " modules (" + (modidMap.size()-oldcnt) + " new)");
				if ((modidMap.size()-oldcnt) == 0) Log.w("INIT", "No new modules, changed: " + changedDeviceids);

			} catch (SQLException e) {
				Log.e("INIT", "SQLException: " + e.getMessage());
				e.printStackTrace(System.err);
			}
		}
		*/

	}

	/**
	 * Return a DataContainer object used to return data to this
	 * DataHandler.
	 */
	public DataContainer dataContainerFactory() {
		//return new ModuleMonContainer(this, queryIfindices, moduleToIfindex);
		return new ModuleMonContainer(this);
	}
	
	/**
	 * Store the data in the DataContainer in the database.
	 */
	public void handleData(Netbox nb, DataContainer dc, Map changedDeviceids) {
		if (!(dc instanceof ModuleMonContainer)) return;
		ModuleMonContainer mmc = (ModuleMonContainer)dc;
		if (!mmc.isCommited()) return;

		Map modules = new HashMap();
		Map deviceMap = new HashMap();
		Map moduleMap = new HashMap();
		try {
			ResultSet rs = Database.query("SELECT module, up, deviceid, moduleid FROM module WHERE netboxid='"+nb.getNetboxid()+"'");
			while (rs.next()) {
				modules.put(rs.getString("module"), rs.getString("up"));
				deviceMap.put(rs.getString("module"), rs.getString("deviceid"));
				moduleMap.put(rs.getString("module"), rs.getString("moduleid"));
			}
		} catch (SQLException e) {
			Log.e("INIT", "SQLException: " + e.getMessage());
			e.printStackTrace(System.err);
		}

		int severity = 50;
		Log.d("MODULE_MON", "Modules up: " + mmc.getModulesUpSet());
		for (Iterator it = mmc.getModulesUp(); it.hasNext();) {
			String module = (String)it.next();
			String up = (String)modules.remove(module);
			// Apparently, this module hasn't been seen on this box before
			if (up == null) continue;

			if ("n".equals(""+up)) {
				// The module is coming up, send up event
				String deviceid = (String)deviceMap.get(module);
				String moduleid = (String)moduleMap.get(module);
				sendEvent(nb, deviceid, moduleid, Event.STATE_END, severity);
			}
		}
		
		/**
		 * Modules that were somehow marked as both up and down, are 
		 * considered up (let's give them the benefit of the doubt).  Remove
		 * any duplicates from the set of modules that are down.
		 */
		mmc.getModulesDownSet().removeAll(mmc.getModulesUpSet());
		
		Log.d("MODULE_MON", "Modules down: " + mmc.getModulesDownSet());
		for (Iterator it = mmc.getModulesDown(); it.hasNext();) {
			String module = (String)it.next();
			String up = (String)modules.remove(module);
			// Apparently, this module hasn't been seen on this box before
			if (up == null) continue;

			if ("y".equals(""+up)) {
				// The module is going down, send down event
				String deviceid = (String)deviceMap.get(module);
				String moduleid = (String)moduleMap.get(module);
				sendEvent(nb, deviceid, moduleid, Event.STATE_START, severity);
			}
		}

		if (mmc.isUnknownDown()) {
			// All remaining modules (those we've seen before, but are uncertain about now) are now considered down; send event
			Log.d("MODULE_MON", "Remaining modules: " + modules);
			if (!modules.isEmpty()) Log.d("MODULE_MON", "REPORT_DOWN", "Reporting modules down (uncertainly): " + modules);
			for (Iterator it = modules.entrySet().iterator(); it.hasNext();) {
				Map.Entry me = (Map.Entry)it.next();
				String module = (String)me.getKey();
				String up = (String)me.getValue();

				if ("y".equals(""+up)) {
					// Module is going down, send down event
					String deviceid = (String)deviceMap.get(module);
					String moduleid = (String)moduleMap.get(module);
					sendEvent(nb, deviceid, moduleid, Event.STATE_START, severity);
				}
				
			}
		} else if (!modules.isEmpty()) {
			Log.d("MODULE_MON", "Ignoring modules: " + modules);
		}
	}

	// Post the event
	private void sendEvent(Netbox nb, String deviceid, String moduleid, int state, int severity) {
		if (!EventQ.createAndPostEvent("moduleMon", "eventEngine", Integer.parseInt(deviceid), nb.getNetboxid(), Integer.parseInt(moduleid), "moduleState", state, -1, severity, null)) {
			Log.c("MODULE_MON", "SEND_EVENT", "Error sending moduleUp|Down event for " + nb + ", moduleid: " + moduleid);
		}
	}

}