def evolve(self, raw_event, states):
next_shot = 0
for state in states:
while True:
now = time.time()
if next_shot < now:
raw_event['state'] = state
next_shot = now + random.randint(5, 10)
break
else:
yield
raw_event['start'] = int(time.time() * 1000)
self.publisher.publish(Event(**raw_event))
def _generate_data(self):
self.names = ['name-' + str(n) for n in xrange(10)]
self.entities = ['host-' + str(n) for n in xrange(10)]
tags = [['key-' + str(k), 'value-' + str(v)] for k in xrange(5)
for v in xrange(3)]
self.tags = [dict(tags[0::3]), dict(tags[1::3]), dict(tags[2::3])]
starts = [100, 200, 300, 400, 500]
for state in EventState.alls:
for i in xrange(len(self.names)):
name, entity = self.names[i], self.entities[i]
for ts in self.tags:
for start in starts:
Event(name=name, entity=entity, tags=ts,
start=start).save()
def gen_event(self, state, timestamp):
return Event(name='test_ttr',
entity='localhost',
tags={'cluster': 'test'},
start=timestamp,
state=state)
def _get_event(self, name, entity='ojr-test', state='critical', tags={}):
default = {'cluster': 'jy', 'role': 'DataNode', 'ip': '10.11.12.13'}
default.update(tags)
return Event(name=name, entity=entity, state=state, tags=default)