diff options
Diffstat (limited to 'kafka/partitioner/base.py')
| -rw-r--r-- | kafka/partitioner/base.py | 13 |
1 files changed, 7 insertions, 6 deletions
diff --git a/kafka/partitioner/base.py b/kafka/partitioner/base.py index 00f7be3..0e36253 100644 --- a/kafka/partitioner/base.py +++ b/kafka/partitioner/base.py @@ -5,22 +5,23 @@ class Partitioner(object): """ Base class for a partitioner """ - def __init__(self, partitions): + def __init__(self, partitions=None): """ Initialize the partitioner Arguments: - partitions: A list of available partitions (during startup) + partitions: A list of available partitions (during startup) OPTIONAL. """ self.partitions = partitions - def partition(self, key, partitions=None): + def __call__(self, key, all_partitions=None, available_partitions=None): """ - Takes a string key and num_partitions as argument and returns + Takes a string key, num_partitions and available_partitions as argument and returns a partition to be used for the message Arguments: - key: the key to use for partitioning - partitions: (optional) a list of partitions. + key: the key to use for partitioning. + all_partitions: a list of the topic's partitions. + available_partitions: a list of the broker's currently avaliable partitions(optional). """ raise NotImplementedError('partition function has to be implemented') |