视图流.io:实现队列tas

2024-05-29 06:45:50 发布

您现在位置:Python中文网/ 问答频道 /正文

我想用ViewFlow library实现以下用例:

问题

由用户启动的特定流的进程在执行芹菜作业之前必须在队列中等待。每个用户都有这些进程的队列。根据计划或手动触发,允许队列中的下一个进程继续。在

示例

流中的节点进入命名队列。对于每个队列,应用程序中的其他逻辑决定何时允许下一个任务继续。将选择队列中的下一个任务,并调用其激活的done()方法。在

示例流可能如下所示:

class MyFlow(Flow):

    start = flow.Start(...).Next(queue_wait)
    queue_wait = QueueWait("myQueue").Next(job)
    job = celery.Job(...).Next(end)
    end = flow.End()

问题

实现排队的最佳方法是什么?在上面的例子中,我不知道“QueueWait”应该是什么。在

我已经通读了docs和viewflow代码,但是我还不清楚是否可以使用内置的节点和激活类来完成,比如函数函数,或者如果需要使用自定义类进行扩展。在


Tags: 方法函数用户示例节点队列queue进程
1条回答
网友
1楼 · 发布于 2024-05-29 06:45:50

经过多次试验,我找到了一个可行且简单的解决方案:

from viewflow.flow import base
from viewflow.flow.func import FuncActivation
from viewflow.activation import STATUS


class Queue(base.NextNodeMixin,
            base.UndoViewMixin,
            base.CancelViewMixin,
            base.DetailsViewMixin,
            base.Event):

    """
    Node that halts the flow and waits in a queue. To process the next waiting task
    call the dequeue method, optionally specifying the task owner.

    Example placing a job in a queue::

        class MyFlow(Flow):
            wait = Queue().Next(this.job)
            job = celery.Job(send_stuff).Next(this.end)
            end = flow.End()

        somewhere in the application code:
        MyFlow.wait.dequeue()
        or:
        MyFlow.wait.dequeue(process__myprocess__owner=user)

    Queues are logically separated by the task_type, so new queues defined in a
    subclass by overriding task_type attribute.
    """

    task_type = 'QUEUE'
    activation_cls = FuncActivation

    def __init__(self, **kwargs):
        super(Queue, self).__init__(**kwargs)

    def dequeue(self, **kwargs):
        """
        Process the next task in the queue by created date/time. kwargs is
        used to add task filter arguments, thereby effectively splitting the queue
        into subqueues. This could be used to implement per-user queues.

        Returns True if task was found and dequeued, False otherwise
        """
        filter_kwargs = {'flow_task_type': self.task_type, 'status': STATUS.NEW}
        if kwargs is not None:
            filter_kwargs.update(kwargs)

        task = self.flow_cls.task_cls.objects.filter(**filter_kwargs).order_by('created').first()
        if task is not None:
            lock = self.flow_cls.lock_impl(self.flow_cls.instance)
            with lock(self.flow_cls, task.process_id):
                task = self.flow_cls.task_cls._default_manager.get(pk=task.pk)
                activation = self.activation_cls()
                activation.initialize(self, task)
                activation.prepare()
                activation.done()
            return True

        return False

我试图使其尽可能通用,并支持多个命名队列和子队列的定义,如按用户队列。在

相关问题 更多 >

    热门问题