class brcdSdSensor(PollingSensor):
def __init__(self, sensor_service, config=None, poll_interval=5):
super(brcdSdSensor, self).__init__(sensor_service=sensor_service,
config=config,
poll_interval=poll_interval)
self._logger = self.sensor_service.get_logger(
name=self.__class__.__name__)
self._stop = False
self._bsd = None
def setup(self):
self._bsd = Bsd(self._config, self._logger)
self._last_errors = None
def poll(self):
try:
errors = self._bsd.get_errors()
last_errors = self._get_last_errors()
self._set_last_errors(errors)
self._process_changes(errors, last_errors)
if last_errors and not errors:
payload = {
"status": "all_clear",
"errors": "ALL CLEAR",
"error_level": "na"
}
self.sensor_service.dispatch(trigger="vadc.bsd_failure_event",
payload=payload)
except Exception as e:
payload = {
"status":
"sensor_fail",
"errors":
"BSD Sensor: {}: {}".format(self._config["brcd_sd_host"], e),
"error_level":
"na"
}
self.sensor_service.dispatch(trigger="vadc.bsd_failure_event",
payload=payload)
def _process_changes(self, errors, last_errors):
for instance in errors:
if "traffic_health" in errors[instance]:
error_level = errors[instance]["traffic_health"]["error_level"]
else:
error_level = "unknown"
if instance in last_errors:
if errors[instance] != last_errors[instance]:
# This has changed
payload = {
"status": "updated",
"error_level": error_level,
"errors": json.dumps(errors[instance],
encoding="utf-8")
}
self.sensor_service.dispatch(
trigger="vadc.bsd_failure_event", payload=payload)
else:
# No change
pass
else:
# New error
payload = {
"status": "new",
"error_level": error_level,
"errors": json.dumps(errors[instance], encoding="utf-8")
}
self.sensor_service.dispatch(trigger="vadc.bsd_failure_event",
payload=payload)
for instance in last_errors:
if instance not in errors:
# Recovered
payload = {
"status": "resolved",
"error_level": "ok",
"errors": json.dumps(last_errors[instance],
encoding="utf-8")
}
self.sensor_service.dispatch(trigger="vadc.bsd_failure_event",
payload=payload)
def _get_last_errors(self):
if not self._last_errors and hasattr(self._sensor_service,
'get_value'):
last_errors = self._sensor_service.get_value(name='last_errors')
if last_errors is not None:
self._last_errors = json.loads(last_errors, encoding="utf-8")
else:
self._last_errors = {}
return self._last_errors
def _set_last_errors(self, last_errors):
self._last_errors = last_errors
if hasattr(self._sensor_service, 'set_value'):
self._sensor_service.set_value(name='last_errors',
value=json.dumps(last_errors,
encoding="utf-8"))
def cleanup(self):
# This is called when the st2 system goes down. You can perform cleanup operations like
# closing the connections to external system here.
pass
def add_trigger(self, trigger):
# This method is called when trigger is created
pass
def update_trigger(self, trigger):
# This method is called when trigger is updated
pass
def remove_trigger(self, trigger):
# This method is called when trigger is deleted
pass
def run(self, stringify):
bsd = Bsd(self.config, self.logger)
result = bsd.get_errors(stringify)
return (True, result)
