openstack/deb-python-cassandra-driver
Code Issues Proposed changes

Avoid unneeded rebuild of token replica map

Browse Source
This commit is contained in:
Tyler Hobbs
2014-05-27 13:49:39 -05:00
parent cc12ed9505
commit b5fbb206d9
2 changed files with 13 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
CHANGELOG.rst
View File

@@ -16,6 +16,9 @@ Bug Fixes
* Avoid submitting tasks to the ThreadPoolExecutor after shutdown. With * Avoid submitting tasks to the ThreadPoolExecutor after shutdown. With
retries enabled, this could cause Cluster.shutdown() to hang under retries enabled, this could cause Cluster.shutdown() to hang under
some circumstances. some circumstances.
* Fix unintended rebuild of token replica map when keyspaces are
discovered (on startup), added, or updated and TokenAwarePolicy is not
in use.
1.1.2 1.1.2
===== =====

16
cassandra/metadata.py
View File

@@ -181,11 +181,11 @@ class Metadata(object):
def _keyspace_added(self, ksname): def _keyspace_added(self, ksname):
if self.token_map: if self.token_map:
self.token_map.rebuild_keyspace(ksname) self.token_map.rebuild_keyspace(ksname, build_if_absent=False)
def _keyspace_updated(self, ksname): def _keyspace_updated(self, ksname):
if self.token_map: if self.token_map:
self.token_map.rebuild_keyspace(ksname) self.token_map.rebuild_keyspace(ksname, build_if_absent=False)
def _keyspace_removed(self, ksname): def _keyspace_removed(self, ksname):
if self.token_map: if self.token_map:
@@ -950,10 +950,14 @@ class TokenMap(object):
self.tokens_to_hosts_by_ks = {} self.tokens_to_hosts_by_ks = {}
self._metadata = metadata self._metadata = metadata
self._rebuild_lock = RLock()
def rebuild_keyspace(self, keyspace): def rebuild_keyspace(self, keyspace, build_if_absent=False):
self.tokens_to_hosts_by_ks[keyspace] = \ with self._rebuild_lock:
self.replica_map_for_keyspace(self._metadata.keyspaces[keyspace]) current = self.tokens_to_hosts_by_ks.get(keyspace, None)
if (build_if_absent and current is None) or (not build_if_absent and current is not None):
replica_map = self.replica_map_for_keyspace(self._metadata.keyspaces[keyspace])
self.tokens_to_hosts_by_ks[keyspace] = replica_map
def replica_map_for_keyspace(self, ks_metadata): def replica_map_for_keyspace(self, ks_metadata):
strategy = ks_metadata.replication_strategy strategy = ks_metadata.replication_strategy
@@ -972,7 +976,7 @@ class TokenMap(object):
""" """
tokens_to_hosts = self.tokens_to_hosts_by_ks.get(keyspace, None) tokens_to_hosts = self.tokens_to_hosts_by_ks.get(keyspace, None)
if tokens_to_hosts is None: if tokens_to_hosts is None:
self.rebuild_keyspace(keyspace) self.rebuild_keyspace(keyspace, build_if_absent=True)
tokens_to_hosts = self.tokens_to_hosts_by_ks.get(keyspace, None) tokens_to_hosts = self.tokens_to_hosts_by_ks.get(keyspace, None)
if tokens_to_hosts is None: if tokens_to_hosts is None:
return [] return []