hiderfong
49 lines
1.6 KiB
Python
49 lines
1.6 KiB
Python
import collections
|
|
import logging
|
|
import time
|
|
from functools import partial
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
|
|
class Inspector:
|
|
methods = ('stats', 'active_queues', 'registered', 'scheduled',
|
|
'active', 'reserved', 'revoked', 'conf')
|
|
|
|
def __init__(self, io_loop, capp, timeout):
|
|
self.io_loop = io_loop
|
|
self.capp = capp
|
|
self.timeout = timeout
|
|
self.workers = collections.defaultdict(dict)
|
|
|
|
def inspect(self, workername=None):
|
|
feutures = []
|
|
for method in self.methods:
|
|
feutures.append(self.io_loop.run_in_executor(None, partial(self._inspect, method, workername)))
|
|
return feutures
|
|
|
|
def _on_update(self, workername, method, response):
|
|
info = self.workers[workername]
|
|
info[method] = response
|
|
info['timestamp'] = time.time()
|
|
|
|
def _inspect(self, method, workername):
|
|
destination = [workername] if workername else None
|
|
inspect = self.capp.control.inspect(timeout=self.timeout, destination=destination)
|
|
|
|
logger.debug('Sending %s inspect command', method)
|
|
start = time.time()
|
|
result = (
|
|
getattr(inspect, method)()
|
|
if method != 'active'
|
|
else getattr(inspect, method)(safe=True)
|
|
)
|
|
logger.debug("Inspect command %s took %.2fs to complete", method, time.time() - start)
|
|
|
|
if result is None or 'error' in result:
|
|
logger.warning("Inspect method %s failed", method)
|
|
return
|
|
for worker, response in result.items():
|
|
if response is not None:
|
|
self.io_loop.add_callback(partial(self._on_update, worker, method, response))
|