def cassandra_init(**kwargs): """ Initialize a clean Cassandra connection. """ if cql_cluster is not None: cql_cluster.shutdown() if cql_session is not None: cql_session.shutdown() connection.setup()
def sync(cls): try: cassandra_host = connections['default'].settings_dict['HOST'].split(',') keyspace = connections['default'].settings_dict['NAME'] user = connections['default'].settings_dict['USER'] password = connections['default'].settings_dict['PASSWORD'] auth_provider = PlainTextAuthProvider(username=user, password=password) if cql_cluster is not None: cql_cluster.shutdown() if cql_session is not None: cql_session.shutdown() connection.setup(cassandra_host, keyspace, auth_provider=auth_provider) sync_table(cls) except NoHostAvailable: pass
def cassandra_init(sender=None, signal=None): """ Issue reported in ADDNOW-483: Based on suggestions in http://stackoverflow.com/questions/24785299/python-cassandra-driver-operationtimeout-on-every-query-in-celery-task. """ try: cassandra_host = connections['default'].settings_dict['HOST'].split(',') keyspace = connections['default'].settings_dict['NAME'] user = connections['default'].settings_dict['USER'] password = connections['default'].settings_dict['PASSWORD'] auth_provider = PlainTextAuthProvider(username=user, password=password) if cql_cluster is not None: cql_cluster.shutdown() if cql_session is not None: cql_session.shutdown() connection.setup(cassandra_host, keyspace, auth_provider=auth_provider) except NoHostAvailable: pass
def celery_init(*args, **kwargs): """ Initialize a clean Cassandra connection. """ try: if cql_cluster is not None: cql_cluster.shutdown() if cql_session is not None: cql_session.shutdown() connection.setup( hosts=settings.STREAM_CASSANDRA_HOSTS, consistency=settings.STREAM_CASSANDRA_CONSISTENCY_LEVEL, default_keyspace=settings.STREAM_DEFAULT_KEYSPACE, **settings.CASSANDRA_DRIVER_KWARGS) except Exception: # noqa logger.exception('celery_cassandra_signal_failure') sys.exit(1) from lego.apps.stats import analytics_client analytics_client.default_client = None analytics_client.setup_analytics()