celery/celery/worker/consumer/heart.py

from __future__ import absolute_import, unicode_literals

from celery import bootsteps

from celery.worker import heartbeat

from .events import Events

__all__ = ['Heart']


class Heart(bootsteps.StartStopStep):

    requires = (Events,)

    def __init__(self, c,
                 without_heartbeat=False, heartbeat_interval=None, **kwargs):
        self.enabled = not without_heartbeat
        self.heartbeat_interval = heartbeat_interval
        c.heart = None

    def start(self, c):
        c.heart = heartbeat.Heart(
            c.timer, c.event_dispatcher, self.heartbeat_interval,
        )
        c.heart.start()

    def stop(self, c):
        c.heart = c.heart and c.heart.stop()
    shutdown = stop