summaryrefslogtreecommitdiff
path: root/kafka/partitioner
diff options
context:
space:
mode:
Diffstat (limited to 'kafka/partitioner')
-rw-r--r--kafka/partitioner/__init__.py2
-rw-r--r--kafka/partitioner/base.py2
-rw-r--r--kafka/partitioner/default.py2
-rw-r--r--kafka/partitioner/roundrobin.py3
4 files changed, 9 insertions, 0 deletions
diff --git a/kafka/partitioner/__init__.py b/kafka/partitioner/__init__.py
index 5b6ac2d..9ce6ade 100644
--- a/kafka/partitioner/__init__.py
+++ b/kafka/partitioner/__init__.py
@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
from .roundrobin import RoundRobinPartitioner
from .hashed import HashedPartitioner, Murmur2Partitioner, LegacyPartitioner
diff --git a/kafka/partitioner/base.py b/kafka/partitioner/base.py
index 857f634..00f7be3 100644
--- a/kafka/partitioner/base.py
+++ b/kafka/partitioner/base.py
@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
class Partitioner(object):
"""
diff --git a/kafka/partitioner/default.py b/kafka/partitioner/default.py
index 358efeb..79205b6 100644
--- a/kafka/partitioner/default.py
+++ b/kafka/partitioner/default.py
@@ -1,3 +1,5 @@
+from __future__ import absolute_import
+
import random
from .hashed import murmur2
diff --git a/kafka/partitioner/roundrobin.py b/kafka/partitioner/roundrobin.py
index 6439e53..d244353 100644
--- a/kafka/partitioner/roundrobin.py
+++ b/kafka/partitioner/roundrobin.py
@@ -1,7 +1,10 @@
+from __future__ import absolute_import
+
from itertools import cycle
from .base import Partitioner
+
class RoundRobinPartitioner(Partitioner):
"""
Implements a round robin partitioner which sends data to partitions