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