def __init__(self, config, registry, trigger):
global _theEManager_
_theEManager_ = self
self.eventqueue = []
self.triggered = []
self.history = EventHistory(config)
self.registry = registry
self.trigger = trigger
self.eventmap = {} #eid -> ref
self.alertmap = {} #lid -> ref
self.halflifes= {} #eid -> int
if self.trigger():
Thread(target=self.watchList).start()
Thread(target=self.watchQueue).start()
class EventManager():
""" """
def __init__(self, config, registry, trigger):
global _theEManager_
_theEManager_ = self
self.eventqueue = []
self.triggered = []
self.history = EventHistory(config)
self.registry = registry
self.trigger = trigger
self.eventmap = {} #eid -> ref
self.alertmap = {} #lid -> ref
self.halflifes= {} #eid -> int
if self.trigger():
Thread(target=self.watchList).start()
Thread(target=self.watchQueue).start()
def getInstance(self):
"""Returns the EventManager, make sure it was initialized first."""
return _theEManager_
def loadEvent(self, event):
eid = self.registry.loadEvent(event.name, event._getPID())
event.ID = eid
self.eventmap[eid] = event
logging.debug("saved event: %s, id=%s"%(event.name,eid))
def loadEvents(self, eventlist):
for event in eventlist:
eid = self.registry.loadEvent(event.name, event._getPID())
event.ID = eid
self.eventmap[eid] = event
def unloadEvent(self, event):
if self.registry.unloadEvent(event.ID):
if self.eventmap.pop(event.ID):
event.ID = UNKNOWN #FIXME: Just use None
return True
return False
def loadAlert(self, alert):
lid = self.registry.loadAlert(alert.name, alert.aid)
alert.ID = lid
self.alertmap[lid] = alert
def loadAlerts(self, alertlist):
for alert in alertlist:
lid = self.registry.loadAlert(alert.name, alert.aid)
alert.ID = lid
self.alertmap[lid] = alert
def unloadAlert(self, alert):
if self.registry.unloadAlert(alert.ID):
if self.alertmap.pop(alert.ID):
alert.ID = UNKNOWN #FIXME: Just use None
return True
return False
def triggerEvent(self, eid):
self.eventqueue.append(eid)
self.history.triggered(eid)
logging.debug("Event(%s) triggered!"%eid)
def detriggerEvent(self, eid):
"""To de-trigger we need to remove it from both the list
of events that have yet to be processed and also the ones
that have been added to the already triggered list.
"""
try:self.eventmap.pop(eid, None)
except: pass
try: self.triggered.remove(eid)
except: pass
#### THE FOLLOWING NEEDS TO BE REALLY FRIGGIN FAST! ####
def runsubscribers(self, eventobj, lids):
""" Throws all the alert's run methods into threads to signal
the event.
"""
logging.debug("running subscribers") #XXX: remove me
for lid in lids:
try: Thread(target=self.alertmap[lid].run, args=(eventobj,)).start()
except Exception as e:
logging.exception(e)
logging.debug("Alerts: %s"%str(self.alertmap))
logging.debug("running subscribers dead") #XXX: remove me
def watchQueue(self):
""" Watches the queue and if there is an event in there, runs all of
the subscribers as quickly as possible.
"""
logging.debug("EventQueue watcher thread started")
while self.trigger():
try:
if len(self.eventqueue) == 0: raise queue.Empty
id = self.eventqueue.pop()
Thread(target=self.runsubscribers,
args=(self.eventmap[id],
self.registry.subscribedTo(id),)).start()
if self.eventmap[id].halflife > 0:
self.triggered.append(self.eventmap[id])
self.halflifes[id] = self.eventmap[id].halflife
except queue.Empty: pass
except Exception as e: logging.exception(e)
if len(self.eventqueue) != 0: continue
# sleep for a very small amount
try: time.sleep(random.uniform(MIN_SLEEP, MAX_SLEEP))
except: pass
logging.debug("EventQueue watcher thread dead")
def watchList(self):
""" Watches the list and removes halflifes from all the events! """
logging.debug("Event list watcher thread started")
while self.trigger():
t = time.time()
for k,v in self.halflifes.items():
if v-1 > 0: self.halflifes[k] = v-1
else:
self.halflifes.pop(k)
detriggerEvent( k )
try: time.sleep(1.0-(time.time()-t))
except:pass #took too long, TODO: what should we do in this case?
logging.debug("Event list watcher thread dead")
