github/sprockets.clients.cassandra
1
0
Fork 0
mirror of https://github.com/sprockets/sprockets.clients.cassandra.git synced 2024-11-24 11:19:54 +00:00
Code Issues Projects Releases Packages Wiki Activity

Merge pull request #4 from sprockets/allow-different-ports

Browse source 
Allow different ports to be used
This commit is contained in:
Chris McGuire 2015-06-01 11:34:23 -04:00
commit 752d489437
1 changed files with 8 additions and 5 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
sprockets/clients/cassandra/__init__.py
View file

@ -20,7 +20,8 @@ except:
version_info = (0, 0, 1) version_info = (0, 0, 1)
__version__ = '.'.join(str(v) for v in version_info) __version__ = '.'.join(str(v) for v in version_info)
DEFAULT_URI = 'cassandra://localhost' DEFAULT_URI = 'cassandra://localhost:9042'
DEFAULT_PORT = 9042
class CassandraConnection(object): class CassandraConnection(object):
@ -48,13 +49,13 @@ class CassandraConnection(object):
def __init__(self, ioloop=None): def __init__(self, ioloop=None):
self._config = self._get_cassandra_config() self._config = self._get_cassandra_config()
self._cluster = Cluster(self._config['contact_points']) self._cluster = Cluster(contact_points=self._config['contact_points'],
port=self._config['port'])
self._session = self._cluster.connect() self._session = self._cluster.connect()
self._ioloop = IOLoop.current() self._ioloop = IOLoop.current()
def _get_cassandra_config(self): def _get_cassandra_config(self):
"""Retrieve a dict containing Cassandra client config params.""" """Retrieve a dict containing Cassandra client config params."""
config = {}
parts = urlsplit(os.environ.get('CASSANDRA_URI', DEFAULT_URI)) parts = urlsplit(os.environ.get('CASSANDRA_URI', DEFAULT_URI))
if parts.scheme != 'cassandra': if parts.scheme != 'cassandra':
raise RuntimeError( raise RuntimeError(
@ -64,8 +65,10 @@ class CassandraConnection(object):
if not ip_addresses: if not ip_addresses:
raise RuntimeError('Unable to find Cassandra in DNS!') raise RuntimeError('Unable to find Cassandra in DNS!')
config['contact_points'] = ip_addresses return {
return config 'contact_points': ip_addresses,
'port': parts.port or DEFAULT_PORT,
}
def set_keyspace(self, keyspace): def set_keyspace(self, keyspace):
"""Set the keyspace used by the connection.""" """Set the keyspace used by the connection."""