class StreamHelperOps():
def __init__(self):
self.stream_keys = ['all_observations']
self.db = Database(host=url.hostname, port=url.port)
def create_push_cg(self):
self.get_push_cg(create=True)
def get_push_cg(self, create=False):
cg = self.db.time_series('cg-push', self.stream_keys)
if create:
for stream in self.stream_keys:
self.db.xadd(stream, {'data': ''})
cg.create()
cg.set_id('$')
return cg
def create_pull_cg(self):
self.get_pull_cg(create=True)
def get_pull_cg(self, create=False):
cg = self.db.time_series('cg-pull', self.stream_keys)
if create:
for stream in self.stream_keys:
self.db.xadd(stream, {'data': ''})
cg.create()
cg.set_id('$')
return cg
def get_consumer_group(self, create=False):
db = Database(host=REDIS_HOST, port=REDIS_PORT)
stream_keys = ['all_observations']
cg = db.time_series('cg-obs', stream_keys)
if create:
for stream in stream_keys:
db.xadd(stream, {'data': ''})
if create:
cg.create()
cg.set_id('$')
return cg.all_observations
