extending.rst 29 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908
  1. .. _guide-extending:
  2. ==========================
  3. Extensions and Bootsteps
  4. ==========================
  5. .. contents::
  6. :local:
  7. :depth: 2
  8. .. _extending-custom-consumers:
  9. Custom Message Consumers
  10. ========================
  11. You may want to embed custom Kombu consumers to manually process your messages.
  12. For that purpose a special :class:`~celery.bootstep.ConsumerStep` bootstep class
  13. exists, where you only need to define the ``get_consumers`` method, which must
  14. return a list of :class:`kombu.Consumer` objects to start
  15. whenever the connection is established:
  16. .. code-block:: python
  17. from celery import Celery
  18. from celery import bootsteps
  19. from kombu import Consumer, Exchange, Queue
  20. my_queue = Queue('custom', Exchange('custom'), 'routing_key')
  21. app = Celery(broker='amqp://')
  22. class MyConsumerStep(bootsteps.ConsumerStep):
  23. def get_consumers(self, channel):
  24. return [Consumer(channel,
  25. queues=[my_queue],
  26. callbacks=[self.handle_message],
  27. accept=['json'])]
  28. def handle_message(self, body, message):
  29. print('Received message: {0!r}'.format(body))
  30. message.ack()
  31. app.steps['consumer'].add(MyConsumerStep)
  32. def send_me_a_message(self, who='world!', producer=None):
  33. with app.producer_or_acquire(producer) as producer:
  34. producer.send(
  35. {'hello': who},
  36. serializer='json',
  37. exchange=my_queue.exchange,
  38. routing_key='routing_key',
  39. declare=[my_queue],
  40. retry=True,
  41. )
  42. if __name__ == '__main__':
  43. send_me_a_message('celery')
  44. .. note::
  45. Kombu Consumers can take use of two different message callback dispatching
  46. mechanisms. The first one is the ``callbacks`` argument which accepts
  47. a list of callbacks with a ``(body, message)`` signature,
  48. the second one is the ``on_message`` argument which takes a single
  49. callback with a ``(message, )`` signature. The latter will not
  50. automatically decode and deserialize the payload which is useful
  51. in many cases:
  52. .. code-block:: python
  53. def get_consumers(self, channel):
  54. return [Consumer(channel, queues=[my_queue],
  55. on_message=self.on_message)]
  56. def on_message(self, message):
  57. payload = message.decode()
  58. print(
  59. 'Received message: {0!r} {props!r} rawlen={s}'.format(
  60. payload, props=message.properties, s=len(message.body),
  61. ))
  62. message.ack()
  63. .. _extending-blueprints:
  64. Blueprints
  65. ==========
  66. Bootsteps is a technique to add functionality to the workers.
  67. A bootstep is a custom class that defines hooks to do custom actions
  68. at different stages in the worker. Every bootstep belongs to a blueprint,
  69. and the worker currently defines two blueprints: **Worker**, and **Consumer**
  70. ----------------------------------------------------------
  71. **Figure A:** Bootsteps in the Worker and Consumer blueprints. Starting
  72. from the bottom up the first step in the worker blueprint
  73. is the Timer, and the last step is to start the Consumer blueprint,
  74. which then establishes the broker connection and starts
  75. consuming messages.
  76. .. figure:: ../images/worker_graph_full.png
  77. ----------------------------------------------------------
  78. .. _extending-worker_blueprint:
  79. Worker
  80. ======
  81. The Worker is the first blueprint to start, and with it starts major components like
  82. the event loop, processing pool, the timer, and also optional components
  83. like the autoscaler. When the worker is fully started it will continue
  84. to the Consumer blueprint.
  85. The :class:`~celery.worker.WorkController` is the core worker implementation,
  86. and contains several methods and attributes that you can use in your bootstep.
  87. .. _extending-worker_blueprint-attributes:
  88. Attributes
  89. ----------
  90. .. _extending-worker-app:
  91. .. attribute:: app
  92. The current app instance.
  93. .. _extending-worker-hostname:
  94. .. attribute:: hostname
  95. The workers node name (e.g. `worker1@example.com`)
  96. .. _extending-worker-blueprint:
  97. .. attribute:: blueprint
  98. This is the worker :class:`~celery.bootsteps.Blueprint`.
  99. .. _extending-worker-hub:
  100. .. attribute:: hub
  101. Event loop object (:class:`~kombu.async.Hub`). You can use
  102. this to register callbacks in the event loop.
  103. This is only supported by async I/O enabled transports (amqp, redis),
  104. in which case the `worker.use_eventloop` attribute should be set.
  105. Your worker bootstep must require the Hub bootstep to use this:
  106. .. code-block:: python
  107. class WorkerStep(bootsteps.StartStopStep):
  108. requires = ('celery.worker.components:Hub', )
  109. .. _extending-worker-pool:
  110. .. attribute:: pool
  111. The current process/eventlet/gevent/thread pool.
  112. See :class:`celery.concurrency.base.BasePool`.
  113. Your worker bootstep must require the Pool bootstep to use this:
  114. .. code-block:: python
  115. class WorkerStep(bootsteps.StartStopStep):
  116. requires = ('celery.worker.components:Pool', )
  117. .. _extending-worker-timer:
  118. .. attribute:: timer
  119. :class:`~kombu.async.timer.Timer` used to schedule functions.
  120. Your worker bootstep must require the Timer bootstep to use this:
  121. .. code-block:: python
  122. class WorkerStep(bootsteps.StartStopStep):
  123. requires = ('celery.worker.components:Timer', )
  124. .. _extending-worker-statedb:
  125. .. attribute:: statedb
  126. :class:`Database <celery.worker.state.Persistent>`` to persist state between
  127. worker restarts.
  128. This is only defined if the ``statedb`` argument is enabled.
  129. Your worker bootstep must require the Statedb bootstep to use this:
  130. .. code-block:: python
  131. class WorkerStep(bootsteps.StartStopStep):
  132. requires = ('celery.worker.components:Statedb', )
  133. .. _extending-worker-autoscaler:
  134. .. attribute:: autoscaler
  135. :class:`~celery.worker.autoscaler.Autoscaler` used to automatically grow
  136. and shrink the number of processes in the pool.
  137. This is only defined if the ``autoscale`` argument is enabled.
  138. Your worker bootstep must require the `Autoscaler` bootstep to use this:
  139. .. code-block:: python
  140. class WorkerStep(bootsteps.StartStopStep):
  141. requires = ('celery.worker.autoscaler:Autoscaler', )
  142. .. _extending-worker-autoreloader:
  143. .. attribute:: autoreloader
  144. :class:`~celery.worker.autoreloder.Autoreloader` used to automatically
  145. reload use code when the filesystem changes.
  146. This is only defined if the ``autoreload`` argument is enabled.
  147. Your worker bootstep must require the `Autoreloader` bootstep to use this;
  148. .. code-block:: python
  149. class WorkerStep(bootsteps.StartStopStep):
  150. requires = ('celery.worker.autoreloader:Autoreloader', )
  151. Example worker bootstep
  152. -----------------------
  153. An example Worker bootstep could be:
  154. .. code-block:: python
  155. from celery import bootsteps
  156. class ExampleWorkerStep(bootsteps.StartStopStep):
  157. requires = ('Pool', )
  158. def __init__(self, worker, **kwargs):
  159. print('Called when the WorkController instance is constructed')
  160. print('Arguments to WorkController: {0!r}'.format(kwargs))
  161. def create(self, worker):
  162. # this method can be used to delegate the action methods
  163. # to another object that implements ``start`` and ``stop``.
  164. return self
  165. def start(self, worker):
  166. print('Called when the worker is started.')
  167. def stop(self, worker):
  168. print("Called when the worker shuts down.")
  169. def terminate(self, worker):
  170. print("Called when the worker terminates")
  171. Every method is passed the current ``WorkController`` instance as the first
  172. argument.
  173. Another example could use the timer to wake up at regular intervals:
  174. .. code-block:: python
  175. from celery import bootsteps
  176. class DeadlockDetection(bootsteps.StartStopStep):
  177. requires = ('Timer', )
  178. def __init__(self, worker, deadlock_timeout=3600):
  179. self.timeout = deadlock_timeout
  180. self.requests = []
  181. self.tref = None
  182. def start(self, worker):
  183. # run every 30 seconds.
  184. self.tref = worker.timer.call_repeatedly(
  185. 30.0, self.detect, (worker, ), priority=10,
  186. )
  187. def stop(self, worker):
  188. if self.tref:
  189. self.tref.cancel()
  190. self.tref = None
  191. def detect(self, worker):
  192. # update active requests
  193. for req in self.worker.active_requests:
  194. if req.time_start and time() - req.time_start > self.timeout:
  195. raise SystemExit()
  196. .. _extending-consumer_blueprint:
  197. Consumer
  198. ========
  199. The Consumer blueprint establishes a connection to the broker, and
  200. is restarted every time this connection is lost. Consumer bootsteps
  201. include the worker heartbeat, the remote control command consumer, and
  202. importantly, the task consumer.
  203. When you create consumer bootsteps you must take into account that it must
  204. be possible to restart your blueprint. An additional 'shutdown' method is
  205. defined for consumer bootsteps, this method is called when the worker is
  206. shutdown.
  207. .. _extending-consumer-attributes:
  208. Attributes
  209. ----------
  210. .. _extending-consumer-app:
  211. .. attribute:: app
  212. The current app instance.
  213. .. _extending-consumer-controller:
  214. .. attribute:: controller
  215. The parent :class:`~@WorkController` object that created this consumer.
  216. .. _extending-consumer-hostname:
  217. .. attribute:: hostname
  218. The workers node name (e.g. `worker1@example.com`)
  219. .. _extending-consumer-blueprint:
  220. .. attribute:: blueprint
  221. This is the worker :class:`~celery.bootsteps.Blueprint`.
  222. .. _extending-consumer-hub:
  223. .. attribute:: hub
  224. Event loop object (:class:`~kombu.async.Hub`). You can use
  225. this to register callbacks in the event loop.
  226. This is only supported by async I/O enabled transports (amqp, redis),
  227. in which case the `worker.use_eventloop` attribute should be set.
  228. Your worker bootstep must require the Hub bootstep to use this:
  229. .. code-block:: python
  230. class WorkerStep(bootsteps.StartStopStep):
  231. requires = ('celery.worker:Hub', )
  232. .. _extending-consumer-connection:
  233. .. attribute:: connection
  234. The current broker connection (:class:`kombu.Connection`).
  235. A consumer bootstep must require the 'Connection' bootstep
  236. to use this:
  237. .. code-block:: python
  238. class Step(bootsteps.StartStopStep):
  239. requires = ('celery.worker.consumer:Connection', )
  240. .. _extending-consumer-event_dispatcher:
  241. .. attribute:: event_dispatcher
  242. A :class:`@events.Dispatcher` object that can be used to send events.
  243. A consumer bootstep must require the `Events` bootstep to use this.
  244. .. code-block:: python
  245. class Step(bootsteps.StartStopStep):
  246. requires = ('celery.worker.consumer:Events', )
  247. .. _extending-consumer-gossip:
  248. .. attribute:: gossip
  249. Worker to worker broadcast communication
  250. (:class:`~celery.worker.consumer.Gossip`).
  251. A consumer bootstep must require the `Gossip` bootstep to use this.
  252. .. code-block:: python
  253. class RatelimitStep(bootsteps.StartStopStep):
  254. """Rate limit tasks based on the number of workers in the
  255. cluster."""
  256. requires = ('celery.worker.consumer:Gossip',)
  257. def start(self, c):
  258. self.c = c
  259. self.c.gossip.on.node_join.add(self.on_cluster_size_change)
  260. self.c.gossip.on.node_leave.add(self.on_cluster_size_change)
  261. self.c.gossip.on.node_lost.add(self.on_node_lost)
  262. self.tasks = [
  263. self.app.tasks['proj.tasks.add']
  264. self.app.tasks['proj.tasks.mul']
  265. ]
  266. self.last_size = None
  267. def on_cluster_size_change(self, worker):
  268. cluster_size = len(self.c.gossip.state.alive_workers())
  269. if cluster_size != self.last_size:
  270. for task in self.tasks:
  271. task.rate_limit = 1.0 / cluster_size
  272. self.c.reset_rate_limits()
  273. self.last_size = cluster_size
  274. def on_node_lost(self, worker):
  275. # may have processed heartbeat too late, so wake up soon
  276. # in order to see if the worker recovered.
  277. self.c.timer.call_after(10.0, self.on_cluster_size_change)
  278. **Callbacks**
  279. - ``<set> gossip.on.node_join``
  280. Called whenever a new node joins the cluster, providing a
  281. :class:`~celery.events.state.Worker` instance.
  282. - ``<set> gossip.on.node_leave``
  283. Called whenever a new node leaves the cluster (shuts down),
  284. providing a :class:`~celery.events.state.Worker` instance.
  285. - ``<set> gossip.on.node_lost``
  286. Called whenever heartbeat was missed for a worker instance in the
  287. cluster (heartbeat not received or processed in time),
  288. providing a :class:`~celery.events.state.Worker` instance.
  289. This does not necessarily mean the worker is actually offline, so use a time
  290. out mechanism if the default heartbeat timeout is not sufficient.
  291. .. _extending-consumer-pool:
  292. .. attribute:: pool
  293. The current process/eventlet/gevent/thread pool.
  294. See :class:`celery.concurrency.base.BasePool`.
  295. .. _extending-consumer-timer:
  296. .. attribute:: timer
  297. :class:`Timer <celery.utils.timer2.Schedule` used to schedule functions.
  298. .. _extending-consumer-heart:
  299. .. attribute:: heart
  300. Responsible for sending worker event heartbeats
  301. (:class:`~celery.worker.heartbeat.Heart`).
  302. Your consumer bootstep must require the `Heart` bootstep to use this:
  303. .. code-block:: python
  304. class Step(bootsteps.StartStopStep):
  305. requires = ('celery.worker.consumer:Heart', )
  306. .. _extending-consumer-task_consumer:
  307. .. attribute:: task_consumer
  308. The :class:`kombu.Consumer` object used to consume task messages.
  309. Your consumer bootstep must require the `Tasks` bootstep to use this:
  310. .. code-block:: python
  311. class Step(bootsteps.StartStopStep):
  312. requires = ('celery.worker.consumer:Tasks',)
  313. .. _extending-consumer-strategies:
  314. .. attribute:: strategies
  315. Every registered task type has an entry in this mapping,
  316. where the value is used to execute an incoming message of this task type
  317. (the task execution strategy). This mapping is generated by the Tasks
  318. bootstep when the consumer starts::
  319. for name, task in app.tasks.items():
  320. strategies[name] = task.start_strategy(app, consumer)
  321. task.__trace__ = celery.app.trace.build_tracer(
  322. name, task, loader, hostname
  323. )
  324. Your consumer bootstep must require the `Tasks` bootstep to use this:
  325. .. code-block:: python
  326. class Step(bootsteps.StartStopStep):
  327. requires = ('celery.worker.consumer:Tasks',)
  328. .. _extending-consumer-task_buckets:
  329. .. attribute:: task_buckets
  330. A :class:`~collections.defaultdict` used to lookup the rate limit for
  331. a task by type.
  332. Entries in this dict may be None (for no limit) or a
  333. :class:`~kombu.utils.limits.TokenBucket` instance implementing
  334. ``consume(tokens)`` and ``expected_time(tokens)``.
  335. TokenBucket implements the `token bucket algorithm`_, but any algorithm
  336. may be used as long as it conforms to the same interface and defines the
  337. two methods above.
  338. .. _`token bucket algorithm`: http://en.wikipedia.org/wiki/Token_bucket
  339. .. _extending_consumer-qos:
  340. .. attribute:: qos
  341. The :class:`~kombu.common.QoS` object can be used to change the
  342. task channels current prefetch_count value, e.g::
  343. # increment at next cycle
  344. consumer.qos.increment_eventually(1)
  345. # decrement at next cycle
  346. consumer.qos.decrement_eventually(1)
  347. consumer.qos.set(10)
  348. Methods
  349. -------
  350. .. method:: consumer.reset_rate_limits()
  351. Updates the ``task_buckets`` mapping for all registered task types.
  352. .. method:: consumer.bucket_for_task(type, Bucket=TokenBucket)
  353. Creates rate limit bucket for a task using its ``task.rate_limit``
  354. attribute.
  355. .. method:: consumer.add_task_queue(name, exchange=None, exchange_type=None,
  356. routing_key=None, \*\*options):
  357. Adds new queue to consume from. This will persist on connection restart.
  358. .. method:: consumer.cancel_task_queue(name)
  359. Stop consuming from queue by name. This will persist on connection
  360. restart.
  361. .. method:: apply_eta_task(request)
  362. Schedule eta task to execute based on the ``request.eta`` attribute.
  363. (:class:`~celery.worker.job.Request`)
  364. .. _extending-bootsteps:
  365. Installing Bootsteps
  366. ====================
  367. ``app.steps['worker']`` and ``app.steps['consumer']`` can be modified
  368. to add new bootsteps::
  369. >>> app = Celery()
  370. >>> app.steps['worker'].add(MyWorkerStep) # < add class, do not instantiate
  371. >>> app.steps['consumer'].add(MyConsumerStep)
  372. >>> app.steps['consumer'].update([StepA, StepB])
  373. >>> app.steps['consumer']
  374. {step:proj.StepB{()}, step:proj.MyConsumerStep{()}, step:proj.StepA{()}
  375. The order of steps is not important here as the order is decided by the
  376. resulting dependency graph (``Step.requires``).
  377. To illustrate how you can install bootsteps and how they work, this is an example step that
  378. prints some useless debugging information.
  379. It can be added both as a worker and consumer bootstep:
  380. .. code-block:: python
  381. from celery import Celery
  382. from celery import bootsteps
  383. class InfoStep(bootsteps.Step):
  384. def __init__(self, parent, **kwargs):
  385. # here we can prepare the Worker/Consumer object
  386. # in any way we want, set attribute defaults and so on.
  387. print('{0!r} is in init'.format(parent))
  388. def start(self, parent):
  389. # our step is started together with all other Worker/Consumer
  390. # bootsteps.
  391. print('{0!r} is starting'.format(parent))
  392. def stop(self, parent):
  393. # the Consumer calls stop every time the consumer is restarted
  394. # (i.e. connection is lost) and also at shutdown. The Worker
  395. # will call stop at shutdown only.
  396. print('{0!r} is stopping'.format(parent))
  397. def shutdown(self, parent):
  398. # shutdown is called by the Consumer at shutdown, it's not
  399. # called by Worker.
  400. print('{0!r} is shutting down'.format(parent))
  401. app = Celery(broker='amqp://')
  402. app.steps['worker'].add(InfoStep)
  403. app.steps['consumer'].add(InfoStep)
  404. Starting the worker with this step installed will give us the following
  405. logs::
  406. <Worker: w@example.com (initializing)> is in init
  407. <Consumer: w@example.com (initializing)> is in init
  408. [2013-05-29 16:18:20,544: WARNING/MainProcess]
  409. <Worker: w@example.com (running)> is starting
  410. [2013-05-29 16:18:21,577: WARNING/MainProcess]
  411. <Consumer: w@example.com (running)> is starting
  412. <Consumer: w@example.com (closing)> is stopping
  413. <Worker: w@example.com (closing)> is stopping
  414. <Consumer: w@example.com (terminating)> is shutting down
  415. The ``print`` statements will be redirected to the logging subsystem after
  416. the worker has been initialized, so the "is starting" lines are timestamped.
  417. You may notice that this does no longer happen at shutdown, this is because
  418. the ``stop`` and ``shutdown`` methods are called inside a *signal handler*,
  419. and it's not safe to use logging inside such a handler.
  420. Logging with the Python logging module is not :term:`reentrant`,
  421. which means that you cannot interrupt the function and
  422. call it again later. It's important that the ``stop`` and ``shutdown`` methods
  423. you write is also :term:`reentrant`.
  424. Starting the worker with ``--loglevel=debug`` will show us more
  425. information about the boot process::
  426. [2013-05-29 16:18:20,509: DEBUG/MainProcess] | Worker: Preparing bootsteps.
  427. [2013-05-29 16:18:20,511: DEBUG/MainProcess] | Worker: Building graph...
  428. <celery.apps.worker.Worker object at 0x101ad8410> is in init
  429. [2013-05-29 16:18:20,511: DEBUG/MainProcess] | Worker: New boot order:
  430. {Hub, Queues (intra), Pool, Autoreloader, Timer, StateDB,
  431. Autoscaler, InfoStep, Beat, Consumer}
  432. [2013-05-29 16:18:20,514: DEBUG/MainProcess] | Consumer: Preparing bootsteps.
  433. [2013-05-29 16:18:20,514: DEBUG/MainProcess] | Consumer: Building graph...
  434. <celery.worker.consumer.Consumer object at 0x101c2d8d0> is in init
  435. [2013-05-29 16:18:20,515: DEBUG/MainProcess] | Consumer: New boot order:
  436. {Connection, Mingle, Events, Gossip, InfoStep, Agent,
  437. Heart, Control, Tasks, event loop}
  438. [2013-05-29 16:18:20,522: DEBUG/MainProcess] | Worker: Starting Hub
  439. [2013-05-29 16:18:20,522: DEBUG/MainProcess] ^-- substep ok
  440. [2013-05-29 16:18:20,522: DEBUG/MainProcess] | Worker: Starting Pool
  441. [2013-05-29 16:18:20,542: DEBUG/MainProcess] ^-- substep ok
  442. [2013-05-29 16:18:20,543: DEBUG/MainProcess] | Worker: Starting InfoStep
  443. [2013-05-29 16:18:20,544: WARNING/MainProcess]
  444. <celery.apps.worker.Worker object at 0x101ad8410> is starting
  445. [2013-05-29 16:18:20,544: DEBUG/MainProcess] ^-- substep ok
  446. [2013-05-29 16:18:20,544: DEBUG/MainProcess] | Worker: Starting Consumer
  447. [2013-05-29 16:18:20,544: DEBUG/MainProcess] | Consumer: Starting Connection
  448. [2013-05-29 16:18:20,559: INFO/MainProcess] Connected to amqp://guest@127.0.0.1:5672//
  449. [2013-05-29 16:18:20,560: DEBUG/MainProcess] ^-- substep ok
  450. [2013-05-29 16:18:20,560: DEBUG/MainProcess] | Consumer: Starting Mingle
  451. [2013-05-29 16:18:20,560: INFO/MainProcess] mingle: searching for neighbors
  452. [2013-05-29 16:18:21,570: INFO/MainProcess] mingle: no one here
  453. [2013-05-29 16:18:21,570: DEBUG/MainProcess] ^-- substep ok
  454. [2013-05-29 16:18:21,571: DEBUG/MainProcess] | Consumer: Starting Events
  455. [2013-05-29 16:18:21,572: DEBUG/MainProcess] ^-- substep ok
  456. [2013-05-29 16:18:21,572: DEBUG/MainProcess] | Consumer: Starting Gossip
  457. [2013-05-29 16:18:21,577: DEBUG/MainProcess] ^-- substep ok
  458. [2013-05-29 16:18:21,577: DEBUG/MainProcess] | Consumer: Starting InfoStep
  459. [2013-05-29 16:18:21,577: WARNING/MainProcess]
  460. <celery.worker.consumer.Consumer object at 0x101c2d8d0> is starting
  461. [2013-05-29 16:18:21,578: DEBUG/MainProcess] ^-- substep ok
  462. [2013-05-29 16:18:21,578: DEBUG/MainProcess] | Consumer: Starting Heart
  463. [2013-05-29 16:18:21,579: DEBUG/MainProcess] ^-- substep ok
  464. [2013-05-29 16:18:21,579: DEBUG/MainProcess] | Consumer: Starting Control
  465. [2013-05-29 16:18:21,583: DEBUG/MainProcess] ^-- substep ok
  466. [2013-05-29 16:18:21,583: DEBUG/MainProcess] | Consumer: Starting Tasks
  467. [2013-05-29 16:18:21,606: DEBUG/MainProcess] basic.qos: prefetch_count->80
  468. [2013-05-29 16:18:21,606: DEBUG/MainProcess] ^-- substep ok
  469. [2013-05-29 16:18:21,606: DEBUG/MainProcess] | Consumer: Starting event loop
  470. [2013-05-29 16:18:21,608: WARNING/MainProcess] celery@example.com ready.
  471. .. _extending-programs:
  472. Command-line programs
  473. =====================
  474. .. _extending-commandoptions:
  475. Adding new command-line options
  476. -------------------------------
  477. .. _extending-command-options:
  478. Command-specific options
  479. ~~~~~~~~~~~~~~~~~~~~~~~~
  480. You can add additional command-line options to the ``worker``, ``beat`` and
  481. ``events`` commands by modifying the :attr:`~@user_options` attribute of the
  482. application instance.
  483. Celery commands uses the :mod:`optparse` module to parse command-line
  484. arguments, and so you have to use :mod:`optparse` specific option instances created
  485. using :func:`optparse.make_option`. Please see the :mod:`optparse`
  486. documentation to read about the fields supported.
  487. Example adding a custom option to the :program:`celery worker` command:
  488. .. code-block:: python
  489. from celery import Celery
  490. from celery.bin import Option # <-- alias to optparse.make_option
  491. app = Celery(broker='amqp://')
  492. app.user_options['worker'].add(
  493. Option('--enable-my-option', action='store_true', default=False,
  494. help='Enable custom option.'),
  495. )
  496. All bootsteps will now receive this argument as a keyword argument to
  497. ``Bootstep.__init__``:
  498. .. code-block:: python
  499. from celery import bootsteps
  500. class MyBootstep(bootsteps.Step):
  501. def __init__(self, worker, enable_my_option=False, **options):
  502. if enable_my_option:
  503. party()
  504. app.steps['worker'].add(MyBootstep)
  505. .. _extending-preload_options:
  506. Preload options
  507. ~~~~~~~~~~~~~~~
  508. The :program:`celery` umbrella command supports the concept of 'preload
  509. options', which are special options passed to all subcommands and parsed
  510. outside of the main parsing step.
  511. The list of default preload options can be found in the API reference:
  512. :mod:`celery.bin.base`.
  513. You can add new preload options too, e.g. to specify a configuration template:
  514. .. code-block:: python
  515. from celery import Celery
  516. from celery import signals
  517. from celery.bin import Option
  518. app = Celery()
  519. app.user_options['preload'].add(
  520. Option('-Z', '--template', default='default',
  521. help='Configuration template to use.'),
  522. )
  523. @signals.user_preload_options.connect
  524. def on_preload_parsed(options, **kwargs):
  525. use_template(options['template'])
  526. .. _extending-subcommands:
  527. Adding new :program:`celery` sub-commands
  528. -----------------------------------------
  529. New commands can be added to the :program:`celery` umbrella command by using
  530. `setuptools entry-points`_.
  531. .. _`setuptools entry-points`:
  532. http://reinout.vanrees.org/weblog/2010/01/06/zest-releaser-entry-points.html
  533. Entry-points is special metadata that can be added to your packages ``setup.py`` program,
  534. and then after installation, read from the system using the :mod:`pkg_resources` module.
  535. Celery recognizes ``celery.commands`` entry-points to install additional
  536. subcommands, where the value of the entry-point must point to a valid subclass
  537. of :class:`celery.bin.base.Command`. There is limited documentation,
  538. unfortunately, but you can find inspiration from the various commands in the
  539. :mod:`celery.bin` package.
  540. This is how the Flower_ monitoring extension adds the :program:`celery flower` command,
  541. by adding an entry-point in :file:`setup.py`:
  542. .. code-block:: python
  543. setup(
  544. name='flower',
  545. entry_points={
  546. 'celery.commands': [
  547. 'flower = flower.command:FlowerCommand',
  548. ],
  549. }
  550. )
  551. .. _Flower: http://pypi.python.org/pypi/flower
  552. The command definition is in two parts separated by the equal sign, where the
  553. first part is the name of the sub-command (flower), then the second part is
  554. the fully qualified symbol path to the class that implements the command:
  555. .. code-block:: text
  556. flower.command:FlowerCommand
  557. The module path and the name of the attribute should be separated by colon
  558. as above.
  559. In the module :file:`flower/command.py`, the command class is defined
  560. something like this:
  561. .. code-block:: python
  562. from celery.bin.base import Command, Option
  563. class FlowerCommand(Command):
  564. def get_options(self):
  565. return (
  566. Option('--port', default=8888, type='int',
  567. help='Webserver port',
  568. ),
  569. Option('--debug', action='store_true'),
  570. )
  571. def run(self, port=None, debug=False, **kwargs):
  572. print('Running our command')
  573. Worker API
  574. ==========
  575. :class:`~kombu.async.Hub` - The workers async event loop.
  576. ---------------------------------------------------------
  577. :supported transports: amqp, redis
  578. .. versionadded:: 3.0
  579. The worker uses asynchronous I/O when the amqp or redis broker transports are
  580. used. The eventual goal is for all transports to use the eventloop, but that
  581. will take some time so other transports still use a threading-based solution.
  582. .. method:: hub.add(fd, callback, flags)
  583. .. method:: hub.add_reader(fd, callback, \*args)
  584. Add callback to be called when ``fd`` is readable.
  585. The callback will stay registered until explictly removed using
  586. :meth:`hub.remove(fd) <hub.remove>`, or the fd is automatically discarded
  587. because it's no longer valid.
  588. Note that only one callback can be registered for any given fd at a time,
  589. so calling ``add`` a second time will remove any callback that
  590. was previously registered for that fd.
  591. A file descriptor is any file-like object that supports the ``fileno``
  592. method, or it can be the file descriptor number (int).
  593. .. method:: hub.add_writer(fd, callback, \*args)
  594. Add callback to be called when ``fd`` is writable.
  595. See also notes for :meth:`hub.add_reader` above.
  596. .. method:: hub.remove(fd)
  597. Remove all callbacks for ``fd`` from the loop.
  598. Timer - Scheduling events
  599. -------------------------
  600. .. method:: timer.call_after(secs, callback, args=(), kwargs=(),
  601. priority=0)
  602. .. method:: timer.call_repeatedly(secs, callback, args=(), kwargs=(),
  603. priority=0)
  604. .. method:: timer.call_at(eta, callback, args=(), kwargs=(),
  605. priority=0)