def create_cassandra_keyspace(config):
sys = SystemManager(config.get('DEFAULT', 'cassandra_seeds'))
if config.get('DEFAULT', 'cassandra_keyspace') in sys.list_keyspaces():
print "found keyspace %r, not creating" % (config.get('DEFAULT', 'cassandra_keyspace'),)
return
print "creating %r" % (config.get('DEFAULT', 'cassandra_keyspace'),)
sys.create_keyspace(config.get('DEFAULT', 'cassandra_keyspace'), replication_factor=1)
def get_keyspaces(self,machine_id): """Returns all keyspaces in form of list """ print "->>>in get keyspace function" sys = SystemManager(machine_id) keyspace_list = sys.list_keyspaces() sys.close() return keyspace_list
def keyspace_contains(self,machine_id,name): """Returns true if keyspace with given name is on specified machine_id """ print "->>>in contain keyspace function" sys = SystemManager(machine_id) keyspace_list = sys.list_keyspaces() sys.close() for i in keyspace_list: if (i == name): return True return False
def keyspace_columnfamily_list(self,machine_id): """Returns dictionary of all keyspace with their column family on specified machine_id """ sys = SystemManager(machine_id) keyspace.complete_list.clear() keyspace_list = sys.list_keyspaces() for key in keyspace_list: x=[] result = sys.get_keyspace_column_families(key, use_dict_for_col_metadata=True) for i in result: x.append(i) keyspace.complete_list[key] = x sys.close() return keyspace.complete_list
def cassandra_changed():
import pycassa
# relation-get:
# port: "9160"
# private-address: 10.0.3.147
host = hookenv.relation_get('private-address')
port = hookenv.relation_get('port')
if host and port:
casshost = '%s:%d' % (host, int(port))
log("Connecting to Cassandra host: %s" % casshost)
sys = pycassa.system_manager.SystemManager(casshost)
# Create 'reddit' keyspace
keyspaces = sys.list_keyspaces()
if CASSANDRA_KEYSPACE not in keyspaces:
sys.create_keyspace(
CASSANDRA_KEYSPACE,
pycassa.system_manager.SIMPLE_STRATEGY,
{'replication_factor': '1'}
)
log("Created '%s' keyspace" % CASSANDRA_KEYSPACE)
else:
log("'%s' keyspace already exists." % CASSANDRA_KEYSPACE)
column_families = sys.get_keyspace_column_families(
CASSANDRA_KEYSPACE
).keys()
if CASSANDRA_COLUMN not in column_families:
sys.create_column_family(
CASSANDRA_KEYSPACE,
CASSANDRA_COLUMN,
column_type='Standard',
default_validation_class=pycassa.BYTES_TYPE
)
log("Created '%s' column" % CASSANDRA_COLUMN)
else:
log("'%s' column already exists" % CASSANDRA_COLUMN)
add_to_ini(values={
'cassandra_seeds': casshost
})
make_ini()
else:
log("cassandra not ready")
return
def keyspace_columnfamily_list(self, machine_id):
"""Returns dictionary of all keyspace with their column family on specified machine_id """
sys = SystemManager(machine_id)
keyspace.complete_list.clear()
keyspace_list = sys.list_keyspaces()
for key in keyspace_list:
x = []
result = sys.get_keyspace_column_families(
key, use_dict_for_col_metadata=True)
for i in result:
x.append(i)
keyspace.complete_list[key] = x
sys.close()
return keyspace.complete_list
def keyspace_get_list(self,machine_id): """Returns all keyspaces in form of list """ keyspace.error = "Unknown error occur please check your inputs" try: sys = SystemManager(machine_id) except Exception as e: print e return False try: keyspace_list = sys.list_keyspaces() except Exception as e: print e return False sys.close() return keyspace_list
def keyspace_contains(self,machine_id,keyspace_name,column_family_name = ''): """Returns true if keyspace with given name and/or column family is on specified machine_id """ sys = SystemManager(machine_id) keyspace.complete_list.clear() keyspace_list = sys.list_keyspaces() for key in keyspace_list: if (key == keyspace_name): if(column_family_name): column_family_list = sys.get_keyspace_column_families(key, use_dict_for_col_metadata=True) for cf in column_family_list: if(cf == column_family_name): return True return False else: return True return False
def keyspace_get_list(self,machine_id): """Returns all keyspaces in form of list """ sys = SystemManager(machine_id) keyspace_list = sys.list_keyspaces() sys.close() return keyspace_list
def keyspace_get_list(self, machine_id):
"""Returns all keyspaces in form of list """
sys = SystemManager(machine_id)
keyspace_list = sys.list_keyspaces()
sys.close()
return keyspace_list
