806a418ede
* set the maximum line length to 100 * removed unnecessary ignores * updated codes for pep8 * added apache 2.0 license for missed files Change-Id: If9398a8fdd094e50d6a07cb73f7389558dbe0a0d Signed-off-by: Amir Mofakhar <amofakhar@op5.com>
37 lines
1.2 KiB
Python
37 lines
1.2 KiB
Python
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
|
# not use this file except in compliance with the License. You may obtain
|
|
# a copy of the License at
|
|
#
|
|
# http://www.apache.org/licenses/LICENSE-2.0
|
|
#
|
|
# Unless required by applicable law or agreed to in writing, software
|
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
|
# License for the specific language governing permissions and limitations
|
|
# under the License.
|
|
|
|
from itertools import cycle
|
|
|
|
from .base import Partitioner
|
|
|
|
|
|
class RoundRobinPartitioner(Partitioner):
|
|
"""
|
|
Implements a round robin partitioner which sends data to partitions
|
|
in a round robin fashion
|
|
"""
|
|
def __init__(self, partitions):
|
|
super(RoundRobinPartitioner, self).__init__(partitions)
|
|
self.iterpart = cycle(partitions)
|
|
|
|
def _set_partitions(self, partitions):
|
|
self.partitions = partitions
|
|
self.iterpart = cycle(partitions)
|
|
|
|
def partition(self, key, partitions=None):
|
|
# Refresh the partition list if necessary
|
|
if partitions and self.partitions != partitions:
|
|
self._set_partitions(partitions)
|
|
|
|
return next(self.iterpart)
|