devicetracker.java

监控大型网络的软件。能够自动发现拓扑结构

/*
 *  DeviceTracker plugin for eventEngine
 *  Hans Jørgen Hoel ([e-mail])
 *
 */

package no.ntnu.nav.eventengine.handlerplugins.DeviceTracker;

import no.ntnu.nav.eventengine.*;
import no.ntnu.nav.eventengine.deviceplugins.Box.*;
import no.ntnu.nav.eventengine.deviceplugins.Netel.*;

import no.ntnu.nav.Database.*;
import no.ntnu.nav.ConfigParser.*;
import no.ntnu.nav.logger.*;

import java.util.*;
import java.sql.ResultSet;
import java.sql.ResultSetMetaData;
import java.sql.SQLException;


public class DeviceTracker implements EventHandler
{
	private static final boolean DEBUG_OUT = true;

	public String[] handleEventTypes()
	{
		return new String[] { "deviceActive", "deviceState", "deviceNotice" };
	}

	public void handle(DeviceDB ddb, Event e, ConfigParser cp)
	{
        Log.setDefaultSubsystem("DEVICE_TRACKER_HANDLER");
        if(DEBUG_OUT)
    		Log.d("HANDLE","DeviceTracker plugin handling event: " + e);

        String eventtype = e.getEventtypeid();
        String alerttype = e.getVar("alerttype");

		// Create alert
		Alert a = ddb.alertFactory(e);
		a.addEvent(e);
        if(alerttype != null) {
        	a.setAlerttype(alerttype);
        }
        // Don't post alert to alertq
        a.setPostAlertq(false);
    	
        if (DEBUG_OUT)
    		Log.d("HANDLE","Added alert: " + a);

        if (eventtype.equals("deviceActive")) {
            // deviceActive is the main "lifecycle of a device" event
            if (e.getState() == Event.STATE_START) {
                // Sent when a device is made active
                a.copyHistoryVar(e, "serial");
                // Set device.active = True
                try {
                    Database.update("UPDATE device SET active = 'TRUE' WHERE deviceid = " + e.getDeviceid());
                    Database.commit();
                } catch (SQLException exp) {
                    Log.e("HANDLE","deviceState-start: Unable to set device.active=True for device " + e.getDeviceid());
                }
            } else if (e.getState() == Event.STATE_END) {
                // Sent when a device reaches it's end of life
                // Set device.active = False
                try {
                    Database.update("UPDATE device SET active = 'FALSE' WHERE deviceid = " + e.getDeviceid());
                    Database.commit();
                } catch (SQLException exp) {
                    Log.e("HANDLE","deviceState-end: Unable to set device.active=False for device " + e.getDeviceid());
                }
            }
        } else if (eventtype.equals("deviceState")) {
            // deviceState event (stateful)
            if (alerttype.equals("deviceInIPOperation")) {
                // If there is already a deviceInIPOperation event for this 
                // device which hasn't ended yet, set the end_time for that 
                // event to the start_time for this one (minus one minute, 
                // to avoid confusion)
                if (e.getState() == Event.STATE_START) {
                    try {
                        ResultSet rs = Database.query("SELECT alerthist.alerthistid FROM alerthist,alerttype WHERE alerthist.alerttypeid = alerttype.alerttypeid AND alerttype.alerttype = 'deviceInIPOperation' AND alerthist.deviceid = " + e.getDeviceid() + " AND alerthist.eventtypeid = 'deviceState' AND alerthist.end_time = 'infinity'");
                        ResultSetMetaData rsmd = rs.getMetaData();
                        if (rs.next()) {
                            // There should be only one result from this query as it
                            // is run every time a new deviceInIPOperation-start event
                            // is posted 
                            try {
                                // subtract one minute, ugly
                                Date end_time = new Date(e.getTime().getTime() - 1000*60);
                                // close event
                                Database.update("UPDATE alerthist SET end_time = '" + end_time + "' WHERE alerthistid = " + rs.getString("alerthistid"));
                                Database.commit(); 
                            } catch (SQLException exp) {
                                Log.e("HANDLE","Unable to update end_time of old deviceState (deviceInIPOperation) event: " + exp.getMessage());
                            }
                        }
                    } catch (SQLException exp) {
                        Log.e("HANDLE","Error while looking for old deviceState (deviceInIPOperation) event: " + exp.getMessage());
                    }
                    a.copyHistoryVars(e, new String[] { "username", "sysname", "moduleid", "roomid", "orgid", "catid" });
                } else if (e.getState() == Event.STATE_END) {
                    // An excplicit "out of operation" event
                    a.copyHistoryVar(e, "username");
                }
            } else if (alerttype.equals("deviceRma")) {
                a.copyHistoryVar(e, "username");
                if (e.getState() == Event.STATE_START) {
                    a.copyHistoryVars(e, new String[] { "rmanumber", "comment" });
                }
            } else {
                Log.e("HANDLE","Unknown alerttype '" + alerttype + "' for deviceState event.");
            }
        } else if (eventtype.equals("deviceNotice")) {
            // deviceNotice event (stateless)
            if (alerttype.equals("deviceError")) {
                a.copyHistoryVars(e, new String[] { "username", "comment", "unittype", "roomid", "locationid" });
            } else if (alerttype.equals("deviceHwUpgrade")) {
                a.copyHistoryVar(e, "description");
            } else if (alerttype.equals("deviceSwUpgrade")) {
                a.copyHistoryVars(e, new String[] { "oldversion", "newversion" });
            } else {
                Log.e("HANDLE","Unknown alerttype '" + alerttype + "' for deviceNotice event.");
            }
        }

		// Post the alert
		try {
			ddb.postAlert(a);
		} catch (PostAlertException exp) {
			Log.w("HANDLE","While posting alert, PostAlertException: " + exp.getMessage());
		}
		
	}
}