Exemple #1
0
 def __init__(self, name):
     self.redis = StrictRedis(
         port=Config.redis_port(),
         host=Config.redis_host(),
     )
     self._name = name
Exemple #2
0
            node_id_attr,
            run_id,
            x_consumer_id,
        )
        self.runs.update_one(
            filter={
                'id': run_id,
                'X-Consumer-ID': x_consumer_id,
            },
            update={
                '$set': {
                    'done': True,
                    'subgraph_ids': subgraph_ids
                },
            },
        )


REDIS_URI = 'redis://{}:{}/1'.format(Config.redis_host(), Config.redis_port())
celery = Celery('main', broker=REDIS_URI)


@celery.task(name='find-subgraph')
def find_subgraphs(run_id):
    runner = Runner()
    runner.run(run_id)


if __name__ == '__main__':
    celery.start()