/
alerts.py
337 lines (262 loc) · 10.3 KB
/
alerts.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
329
330
331
332
333
334
335
336
337
#!/usr/bin/env python
"""
@package ion.agents.alarms.alarms
@file ion/agents/alarms/alarms.py
@author Edward Hunter
@brief Alarm objects to control construction of valid alarm expressions.
"""
__author__ = 'Edward Hunter'
__license__ = 'Apache 2.0'
# Pyon imports
from pyon.public import IonObject, log
# Standard imports.
import time
# gevent.
import gevent
# Alarm types and events.
from interface.objects import StreamAlertType, AggregateStatusType
# Events.
from pyon.event.event import EventPublisher
# Resource agent.
from pyon.agent.agent import ResourceAgentState
class BaseAlert(object):
"""
"""
def __init__(self, name=None, stream_name=None, message=None, alert_type=None,
value_id=None, resource_id=None, origin_type=None, aggregate_type=None):
assert isinstance(name, str)
assert isinstance(stream_name, str)
assert alert_type in StreamAlertType._str_map.keys()
if alert_type == StreamAlertType.ALL_CLEAR:
message == 'Alert is cleared.'
else:
assert isinstance(message, str)
if aggregate_type:
assert aggregate_type in AggregateStatusType._str_map.keys()
if value_id: assert isinstance(value_id, str)
assert isinstance(resource_id, str)
assert isinstance(origin_type, str)
self._name = name
self._stream_name = stream_name
self._message = message
self._alert_type = alert_type
self._aggregate_type = aggregate_type
self._value_id = value_id
self._resource_id = resource_id
self._origin_type = origin_type
self._status = None
self._prev_status = None
self._current_value = None
def get_status(self):
"""
"""
status = {
'name' : self._name,
'stream_name' : self._stream_name,
'message' : self._message,
'alert_type' : self._alert_type,
'aggregate_type' : self._aggregate_type,
'value_id' : self._value_id,
'alert_class' : self.__class__.__name__,
'value' : self._current_value,
'status' : self._status
}
return status
def make_event_data(self):
"""
"""
event_data = {
'name' : self._name,
'stream_name' : self._stream_name,
'message' : self._message,
'value' : self._current_value,
'event_type' : 'StreamAlertEvent',
'origin' : self._resource_id,
'origin_type' : self._origin_type
}
if self._status:
event_data['sub_type'] = 'ALL_CLEAR'
elif self._alert_type == StreamAlertType.WARNING:
event_data['sub_type'] = 'WARNING'
elif self._alert_type == StreamAlertType.ALERT:
event_data['sub_type'] = 'ALERT'
return event_data
def publish_alert(self):
"""
"""
event_data = self.make_event_data()
#print '########## publishing: ' + event_data['sub_type']
#print '########## publishing etc: ' + str(event_data)
pub = EventPublisher()
pub.publish_event(**event_data)
def stop(self):
pass
class IntervalAlert(BaseAlert):
"""
"""
rel_ops = ['<', '<=']
def __init__(self, name=None, stream_name=None, message=None, alert_type=None,
value_id=None, resource_id=None, origin_type=None, aggregate_type=None,
lower_bound=None, lower_rel_op=None, upper_bound=None,
upper_rel_op=None, **kwargs):
super(IntervalAlert, self).__init__(name, stream_name, message,
alert_type, value_id, resource_id, origin_type, aggregate_type)
assert isinstance(value_id, str)
self._value_id = value_id
self._lower_bound = None
self._upper_bound = None
self._upper_rel_op = None
self._lower_rel_op = None
assert (isinstance(lower_bound, (int, float)) \
or isinstance(upper_bound, (int, float)))
if isinstance(lower_bound, (int, float)):
assert lower_rel_op in IntervalAlert.rel_ops
self._lower_rel_op = lower_rel_op
self._lower_bound = lower_bound
if isinstance(upper_bound, (int, float)):
assert upper_rel_op in IntervalAlert.rel_ops
self._upper_rel_op = upper_rel_op
self._upper_bound= upper_bound
def get_status(self):
status = super(IntervalAlert, self).get_status()
status['lower_bound'] = self._lower_bound
status['upper_bound'] = self._upper_bound
status['lower_rel_op'] = self._lower_rel_op
status['upper_rel_op'] = self._upper_rel_op
return status
def eval_alert(self, stream_name=None, value=None, value_id=None, **kwargs):
if stream_name != self._stream_name or value_id != self._value_id \
or not value:
return
self._current_value = value
self._prev_status = self._status
if self._lower_bound and self._upper_bound:
if self._lower_rel_op == '<=':
if self._upper_rel_op == '<=':
self._status = (self._lower_bound <= self._current_value <= self._upper_bound)
else:
self._status = (self._lower_bound <= self._current_value < self._upper_bound)
else:
if self._upper_rel_op == '<=':
self._status = (self._lower_bound < self._current_value <= self._upper_bound)
else:
self._status = (self._lower_bound < self._current_value < self._upper_bound)
elif self._lower_bound:
if self._lower_rel_op == '<=':
self._status = (self._lower_bound <= self._current_value)
else:
self._status = (self._lower_bound < self._current_value)
elif self._upper_bound:
if self._upper_rel_op == '<=':
self._status = (self._current_value <= self._upper_bound)
else:
self._status = (self._current_value < self._upper_bound)
if self._prev_status != self._status:
self.publish_alert()
class RSNEventAlert(BaseAlert):
"""
"""
# value_id represents the name of the monitorable in an RSNAlert
#
def __init__(self, name=None, stream_name=None, message=None, alert_type=None,
value_id=None, resource_id=None, origin_type=None, aggregate_type=None,
**kwargs):
super(RSNEventAlert, self).__init__(name, '', message,
alert_type, value_id, resource_id, origin_type, aggregate_type)
assert isinstance(value_id, str)
self._value_id = value_id
# {
# "group": "power",
# "name" : "low_voltage_warning",
# "value_id" : "input_voltage",
# "value" : "1.2",
# "alert_type" : "warning",
# "url": "http://localhost:8000",
# "timestamp": 3573569514.295556,
# "ref_id": "44.78",
# "platform_id": "TODO_some_platform_id_of_type_UPS",
# "message": "low battery (synthetic event generated from simulator)"
# }
self._name = name
self._stream_name = stream_name
self._message = message
self._alert_type = alert_type
self._aggregate_type = aggregate_type
self._value_id = value_id
self._resource_id = resource_id
self._origin_type = origin_type
self._status = None
self._prev_status = None
self._current_value = None
def get_status(self):
status = super(RSNEventAlert, self).get_status()
return status
def eval_alert(self, rsn_alert=None):
# x is an RSN event struct TBD
assert isinstance(rsn_alert, dict)
#print 'x: %s',x.keys()
#print 'x: %s',x
self._current_value = rsn_alert['value']
self._prev_status = self._status
self._message = rsn_alert['message']
if rsn_alert['alert_type'] is "warning":
self._alert_type = StreamAlertType.WARNING
self._status = False
elif rsn_alert['alert_type'] is "error":
self._alert_type = StreamAlertType.ALERT
self._status = False
else:
self._alert_type = StreamAlertType.ALL_CLEAR
self._status = True
if self._prev_status != self._status:
self.publish_alert()
return
class UserExpressionAlert(BaseAlert):
"""
"""
pass
class DeltaAlert(BaseAlert):
"""
"""
pass
class LateDataAlert(BaseAlert):
"""
"""
def __init__(self, name=None, stream_name=None, message=None, alert_type=None,
value_id=None, resource_id=None, origin_type=None, aggregate_type=None,
time_delta=None, get_state=None, **kwargs):
super(LateDataAlert, self).__init__(name, stream_name, message,
alert_type, value_id, resource_id, origin_type, aggregate_type)
assert isinstance(time_delta, (int, float))
assert get_state
assert callable(get_state)
self._time_delta = time_delta
self._get_state = get_state
self._gl = gevent.spawn(self._check_data)
def get_status(self):
status = super(LateDataAlert, self).get_status()
status['time_delta'] = self._time_delta
return status
def eval_alert(self, stream_name=None, **kwargs):
if stream_name != self._stream_name:
return
self._current_value = time.time()
if not self._status:
self._status = True
self.publish_alert()
def _check_data(self):
"""
"""
while True:
prev_value = self._current_value
prev_status = self._status
gevent.sleep(self._time_delta)
if self._get_state() == ResourceAgentState.STREAMING:
if self._current_value == prev_value and self._status:
self._status = False
self.publish_alert()
def stop(self):
if self._gl:
self._gl.kill()
self._gl.join()
self._gl = None