PyBitmessage-2024-12-02/src/network/receivequeuethread.py

import Queue
import sys
import threading
import time

import addresses
from bmconfigparser import BMConfigParser
from debug import logger
from helper_threading import StoppableThread
from inventory import Inventory
from network.connectionpool import BMConnectionPool
from network.bmproto import BMProto
from queues import receiveDataQueue
import protocol
import state

class ReceiveQueueThread(threading.Thread, StoppableThread):
    def __init__(self):
        threading.Thread.__init__(self, name="ReceiveQueueThread")
        self.initStop()
        self.name = "ReceiveQueueThread"
        logger.info("init receive queue thread")

    def run(self):
        while not self._stopped and state.shutdown == 0:
            try:
                connection = receiveDataQueue.get(block=True, timeout=1)
                receiveDataQueue.task_done()
            except Queue.Empty:
                continue

            if self._stopped:
                break
            # cycle as long as there is data
            # methods should return False if there isn't enough data, or the connection is to be aborted
            try:
                while connection.process():
                    pass
            except AttributeError:
                # missing command
                logger.error("Unknown state %s, ignoring", connection.state)

    def stopThread(self):
        super(ReceiveQueueThread, self).stopThread()