openstack/deb-python-kafka
Code Issues Proposed changes

Metadata with_partitions() (#787)

Browse Source 
add method ClusterMetadata.with_partitions
  also fixup ClusterMetadata __str__
This commit is contained in:
Dana Powers
2016-08-01 18:14:59 -07:00
committed by GitHub
parent 99891adfa0
commit 40e2faa9ab
1 changed files with 20 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
kafka/cluster.py
View File

@@ -326,6 +326,25 @@ class ClusterMetadata(object):
self._groups[group] = node_id self._groups[group] = node_id
return True return True
def with_partitions(self, partitions_to_add):
"""Returns a copy of cluster metadata with partitions added"""
new_metadata = ClusterMetadata(**self.config)
new_metadata._brokers = copy.deepcopy(self._brokers)
new_metadata._partitions = copy.deepcopy(self._partitions)
new_metadata._broker_partitions = copy.deepcopy(self._broker_partitions)
new_metadata._groups = copy.deepcopy(self._groups)
new_metadata.internal_topics = copy.deepcopy(self.internal_topics)
new_metadata.unauthorized_topics = copy.deepcopy(self.unauthorized_topics)
for partition in partitions_to_add:
new_metadata._partitions[partition.topic][partition.partition] = partition
if partition.leader is not None and partition.leader != -1:
new_metadata._broker_partitions[partition.leader].add(
TopicPartition(partition.topic, partition.partition))
return new_metadata
def __str__(self): def __str__(self):
return 'Cluster(brokers: %d, topics: %d, groups: %d)' % \ return 'ClusterMetadata(brokers: %d, topics: %d, groups: %d)' % \
(len(self._brokers), len(self._partitions), len(self._groups)) (len(self._brokers), len(self._partitions), len(self._groups))