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()
