blob: 0e36253ef53928838e799cb6dc4810b6565ed35d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
|
from __future__ import absolute_import
class Partitioner(object):
"""
Base class for a partitioner
"""
def __init__(self, partitions=None):
"""
Initialize the partitioner
Arguments:
partitions: A list of available partitions (during startup) OPTIONAL.
"""
self.partitions = partitions
def __call__(self, key, all_partitions=None, available_partitions=None):
"""
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.
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')
|
