本文整理汇总了Python中celery.datastructures.LocalCache.clear方法的典型用法代码示例。如果您正苦于以下问题：Python LocalCache.clear方法的具体用法？Python LocalCache.clear怎么用？Python LocalCache.clear使用的例子？那么恭喜您, 这里精选的方法代码示例或许可以为您提供帮助。您也可以进一步了解该方法所在类celery.datastructures.LocalCache的用法示例。
在下文中一共展示了LocalCache.clear方法的2个代码示例，这些例子默认根据受欢迎程度排序。您可以为喜欢或者感觉有用的代码点赞，您的评价将有助于系统推荐出更棒的Python代码示例。
示例1: State
# 需要导入模块: from celery.datastructures import LocalCache [as 别名]
# 或者: from celery.datastructures.LocalCache import clear [as 别名]
class State(object):
"""Records clusters state."""
event_count = 0
task_count = 0
def __init__(self, callback=None,
max_workers_in_memory=5000, max_tasks_in_memory=10000):
self.workers = LocalCache(max_workers_in_memory)
self.tasks = LocalCache(max_tasks_in_memory)
self.event_callback = callback
self.group_handlers = {"worker": self.worker_event,
"task": self.task_event}
self._mutex = Lock()
def freeze_while(self, fun, *args, **kwargs):
clear_after = kwargs.pop("clear_after", False)
self._mutex.acquire()
try:
return fun(*args, **kwargs)
finally:
if clear_after:
self._clear()
self._mutex.release()
def clear_tasks(self, ready=True):
self._mutex.acquire()
try:
return self._clear_tasks(ready)
finally:
self._mutex.release()
def _clear_tasks(self, ready=True):
if ready:
self.tasks = dict((uuid, task)
for uuid, task in self.tasks.items()
if task.state not in states.READY_STATES)
else:
self.tasks.clear()
def _clear(self, ready=True):
self.workers.clear()
self._clear_tasks(ready)
self.event_count = 0
self.task_count = 0
def clear(self, ready=True):
self._mutex.acquire()
try:
return self._clear(ready)
finally:
self._mutex.release()
def get_or_create_worker(self, hostname, **kwargs):
"""Get or create worker by hostname."""
try:
worker = self.workers[hostname]
worker.update(kwargs)
except KeyError:
worker = self.workers[hostname] = Worker(
hostname=hostname, **kwargs)
return worker
def get_or_create_task(self, uuid):
"""Get or create task by uuid."""
try:
return self.tasks[uuid]
except KeyError:
task = self.tasks[uuid] = Task(uuid=uuid)
return task
def worker_event(self, type, fields):
"""Process worker event."""
hostname = fields.pop("hostname", None)
if hostname:
worker = self.get_or_create_worker(hostname)
handler = getattr(worker, "on_%s" % type, None)
if handler:
handler(**fields)
def task_event(self, type, fields):
"""Process task event."""
uuid = fields.pop("uuid")
hostname = fields.pop("hostname")
worker = self.get_or_create_worker(hostname)
task = self.get_or_create_task(uuid)
handler = getattr(task, "on_%s" % type, None)
if type == "received":
self.task_count += 1
if handler:
handler(**fields)
task.worker = worker
def event(self, event):
self._mutex.acquire()
try:
return self._dispatch_event(event)
finally:
self._mutex.release()
def _dispatch_event(self, event):
#.........这里部分代码省略.........
示例2: State
# 需要导入模块: from celery.datastructures import LocalCache [as 别名]
# 或者: from celery.datastructures.LocalCache import clear [as 别名]
class State(object):
"""Records clusters state."""
event_count = 0
task_count = 0
_buffering = False
buffer = deque()
frozen = False
def __init__(self, callback=None,
max_workers_in_memory=5000, max_tasks_in_memory=10000):
self.workers = LocalCache(max_workers_in_memory)
self.tasks = LocalCache(max_tasks_in_memory)
self.event_callback = callback
self.group_handlers = {"worker": self.worker_event,
"task": self.task_event}
self._resource = RLock()
def freeze(self, buffer=True):
"""Stop recording the event stream.
:keyword buffer: If true, any events received while frozen
will be buffered, you can use ``thaw(replay=True)`` to apply
this buffer. :meth:`thaw` will clear the buffer and resume
recording the stream.
"""
self._buffering = buffer
self.frozen = True
def _replay(self):
while self.buffer:
try:
event = self.buffer.popleft()
except IndexError:
pass
self._dispatch_event(event)
def thaw(self, replay=True):
"""Resume recording of the event stream.
:keyword replay: Will replay buffered events received while
the stream was frozen.
This will always clear the buffer, deleting any events collected
while the stream was frozen.
"""
self._buffering = False
try:
if replay:
self._replay()
else:
self.buffer.clear()
finally:
self.frozen = False
def freeze_while(self, fun, *args, **kwargs):
self.freeze()
try:
return fun(*args, **kwargs)
finally:
self.thaw(replay=True)
def clear_tasks(self, ready=True):
if ready:
self.tasks = dict((uuid, task)
for uuid, task in self.tasks.items()
if task.state not in states.READY_STATES)
else:
self.tasks.clear()
def clear(self, ready=True):
try:
self.workers.clear()
self.clear_tasks(ready)
self.event_count = 0
self.task_count = 0
finally:
pass
def get_or_create_worker(self, hostname, **kwargs):
"""Get or create worker by hostname."""
try:
worker = self.workers[hostname]
worker.update(kwargs)
except KeyError:
worker = self.workers[hostname] = Worker(
hostname=hostname, **kwargs)
return worker
def get_or_create_task(self, uuid):
"""Get or create task by uuid."""
try:
return self.tasks[uuid]
except KeyError:
task = self.tasks[uuid] = Task(uuid=uuid)
return task
def worker_event(self, type, fields):
"""Process worker event."""
#.........这里部分代码省略.........