async def wait_for_flow_control(self, stream_id):
"""
Blocks until the flow control window for a given stream is opened.
"""
evt = Event()
self.flow_control_events[stream_id] = evt
await evt.wait()
def _get_hub_class(self, hub_type, sensor, sensor_name, capabilities):
stop_evt = Event()
@attach(sensor, name=sensor_name, capabilities=capabilities)
class TestHub(hub_type):
async def sensor_change(self):
pass
async def run(self):
pass
await stop_evt.wait()
return TestHub, stop_evt
def __init__(self, worker_nums, worker_timeout, getter_queue, putter_queue,
task_module_path, logger):
self.worker_nums = worker_nums
self.worker_timeout = worker_timeout
self.getter_queue, self.putter_queue = getter_queue, putter_queue
self.task_module_path = task_module_path
self.workers = {}
self.idle_workers = []
self.busy_workers = {}
self.watch_tasks = {}
self.idle_available = Event()
self.wait_for_idle = False
self.logger = logger
self.alive = True
self.logger.debug('+++new worker pool instance++++')
return
def __init__(self,
queues,
logger,
getter_queue,
putter_queue,
qos=None,
amqp_url='amqp://*****:*****@localhost:5672//'):
# queues = ['q1', 'q2', 'q3', ...]
self.logger = logger
self.queues = set([i.upper() for i in queues])
self.amqp_url = amqp_url
self.getter_queue, self.putter_queue = getter_queue, putter_queue
self.parse_amqp_url()
self.status = ConnectionStatus.INITAL
self.reconnect_done_event = Event()
self.qos = qos if qos is not None else len(self.queues)
self.logger.debug('connection initial~~~~')
return
def __init__(self):
self.outgoing = Queue()
self.incoming = Queue()
self.closure = None
self.closing = Event()
def __init__(self):
self._event = Event()
self._result = None
self._exception = None
def __init__(self):
self._event = Event()
self._data = None
self._exception = None
async def wait_for_flow_control(self, stream_id):
evt = Event()
self.flow_control_events[stream_id] = evt
await evt.wait()
