PyBitmessage-2024-12-08/src/multiqueue.py

"""
A queue with multiple internal subqueues.
Elements are added into a random subqueue, and retrieval rotates
"""

from collections import deque

from six.moves import queue

try:
    import helper_random
except ImportError:
    from . import helper_random


class MultiQueue(queue.Queue):
    """A base queue class"""
    # pylint: disable=redefined-builtin,attribute-defined-outside-init
    defaultQueueCount = 10

    def __init__(self, maxsize=0, count=0):
        if not count:
            self.queueCount = MultiQueue.defaultQueueCount
        else:
            self.queueCount = count
        queue.Queue.__init__(self, maxsize)

    # Initialize the queue representation
    def _init(self, maxsize):
        self.iter = 0
        self.queues = []
        for _ in range(self.queueCount):
            self.queues.append(deque())

    def _qsize(self, len=len):
        return len(self.queues[self.iter])

    # Put a new item in the queue
    def _put(self, item):
        # self.queue.append(item)
        self.queues[helper_random.randomrandrange(self.queueCount)].append(
            (item))

    # Get an item from the queue
    def _get(self):
        return self.queues[self.iter].popleft()

    def iterate(self):
        """Increment the iteration counter"""
        self.iter = (self.iter + 1) % self.queueCount

    def totalSize(self):
        """Return the total number of items in all the queues"""
        return sum(len(x) for x in self.queues)