celerybeat 6.0 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241
  1. #!/bin/bash -e
  2. # =========================================================
  3. # celerybeat - Starts the Celery periodic task scheduler.
  4. # =========================================================
  5. #
  6. # :Usage: /etc/init.d/celerybeat {start|stop|force-reload|restart|try-restart|status}
  7. #
  8. # :Configuration file: /etc/default/celerybeat or /etc/default/celeryd
  9. #
  10. # EXAMPLE CONFIGURATION
  11. # =====================
  12. #
  13. # this is an example configuration for a Python project:
  14. #
  15. # /etc/default/celeryd:
  16. #
  17. # # Where to chdir at start.
  18. # CELERYD_CHDIR="/opt/Myproject/"
  19. #
  20. # # Extra arguments to celeryd
  21. # CELERYD_OPTS="--time-limit=300"
  22. #
  23. # # Extra arguments to celerybeat
  24. # CELERYBEAT_OPTS="--schedule=/var/run/celerybeat-schedule"
  25. #
  26. # # Name of the celery config module.#
  27. # CELERY_CONFIG_MODULE="celeryconfig"
  28. #
  29. # EXAMPLE DJANGO CONFIGURATION
  30. # ============================
  31. #
  32. # # Where the Django project is.
  33. # CELERYD_CHDIR="/opt/Project/"
  34. #
  35. # # Name of the projects settings module.
  36. # export DJANGO_SETTINGS_MODULE="MyProject.settings"
  37. #
  38. # # Path to celeryd
  39. # CELERYD="/opt/Project/manage.py celeryd"
  40. #
  41. # # Path to celerybeat
  42. # CELERYBEAT="/opt/Project/manage.py"
  43. #
  44. # # Extra arguments to celerybeat
  45. # CELERYBEAT_OPTS="celerybeat --schedule=/var/run/celerybeat-schedule"
  46. #
  47. # AVAILABLE OPTIONS
  48. # =================
  49. #
  50. # * CELERYBEAT_OPTS
  51. # Additional arguments to celerybeat, see `celerybeat --help` for a
  52. # list.
  53. #
  54. # * CELERYBEAT_PID_FILE
  55. # Full path to the pidfile. Default is /var/run/celeryd.pid.
  56. #
  57. # * CELERYBEAT_LOG_FILE
  58. # Full path to the celeryd logfile. Default is /var/log/celeryd.log
  59. #
  60. # * CELERYBEAT_LOG_LEVEL
  61. # Log level to use for celeryd. Default is INFO.
  62. #
  63. # * CELERYBEAT
  64. # Path to the celeryd program. Default is `celeryd`.
  65. # You can point this to an virtualenv, or even use manage.py for django.
  66. #
  67. # * CELERYBEAT_USER
  68. # User to run celeryd as. Default is current user.
  69. #
  70. # * CELERYBEAT_GROUP
  71. # Group to run celeryd as. Default is current user.
  72. #
  73. # * VIRTUALENV
  74. # Full path to the virtualenv environment to activate. Default is none.
  75. ### BEGIN INIT INFO
  76. # Provides: celerybeat
  77. # Required-Start: $network $local_fs $remote_fs
  78. # Required-Stop: $network $local_fs $remote_fs
  79. # Default-Start: 2 3 4 5
  80. # Default-Stop: 0 1 6
  81. # Short-Description: celery periodic task scheduler
  82. ### END INIT INFO
  83. set -e
  84. DEFAULT_PID_FILE="/var/run/celerybeat.pid"
  85. DEFAULT_LOG_FILE="/var/log/celerybeat.log"
  86. DEFAULT_LOG_LEVEL="INFO"
  87. DEFAULT_CELERYBEAT="celerybeat"
  88. # /etc/init.d/ssh: start and stop the celery task worker daemon.
  89. if test -f /etc/default/celeryd; then
  90. . /etc/default/celeryd
  91. fi
  92. CELERYBEAT=${CELERYBEAT:-$DEFAULT_CELERYBEAT}
  93. CELERYBEAT_PID_FILE=${CELERYBEAT_PID_FILE:-${CELERYBEAT_PIDFILE:-$DEFAULT_PID_FILE}}
  94. CELERYBEAT_LOG_FILE=${CELERYBEAT_LOG_FILE:-${CELERYBEAT_LOGFILE:-$DEFAULT_LOG_FILE}}
  95. CELERYBEAT_LOG_LEVEL=${CELERYBEAT_LOG_LEVEL:-${CELERYBEAT_LOGLEVEL:-$DEFAULT_LOG_LEVEL}}
  96. export CELERY_LOADER
  97. . /lib/lsb/init-functions
  98. CELERYBEAT_OPTS="$CELERYBEAT_OPTS -f $CELERYBEAT_LOG_FILE -l $CELERYBEAT_LOG_LEVEL"
  99. if [ -n "$2" ]; then
  100. CELERYBEAT_OPTS="$CELERYBEAT_OPTS $2"
  101. fi
  102. # Issue #434
  103. CELERYBEAT_OPTS="$CELERYBEAT_OPTS --pidfile="
  104. # Extra start-stop-daemon options, like user/group.
  105. if [ -n "$CELERYBEAT_USER" ]; then
  106. DAEMON_OPTS="$DAEMON_OPTS --chuid $CELERYBEAT_USER"
  107. fi
  108. if [ -n "$CELERYBEAT_GROUP" ]; then
  109. DAEMON_OPTS="$DAEMON_OPTS --group $CELERYBEAT_GROUP"
  110. fi
  111. CELERYBEAT_CHDIR=${CELERYBEAT_CHDIR:-$CELERYD_CHDIR}
  112. if [ -n "$CELERYBEAT_CHDIR" ]; then
  113. DAEMON_OPTS="$DAEMON_OPTS --chdir $CELERYBEAT_CHDIR"
  114. fi
  115. # Are we running from init?
  116. run_by_init() {
  117. ([ "$previous" ] && [ "$runlevel" ]) || [ "$runlevel" = S ]
  118. }
  119. check_dev_null() {
  120. if [ ! -c /dev/null ]; then
  121. if [ "$1" = log_end_msg ]; then
  122. log_end_msg 1 || true
  123. fi
  124. if ! run_by_init; then
  125. log_action_msg "/dev/null is not a character device!"
  126. fi
  127. exit 1
  128. fi
  129. }
  130. export PATH="${PATH:+$PATH:}/usr/sbin:/sbin"
  131. stop_worker () {
  132. cmd="start-stop-daemon --stop \
  133. --quiet \
  134. $* \
  135. --pidfile $CELERYBEAT_PID_FILE"
  136. if $cmd; then
  137. log_end_msg 0
  138. else
  139. log_end_msg 1
  140. fi
  141. }
  142. start_worker () {
  143. cmd="start-stop-daemon --start $DAEMON_OPTS \
  144. --quiet \
  145. --oknodo \
  146. --background \
  147. --make-pidfile \
  148. $* \
  149. --pidfile $CELERYBEAT_PID_FILE
  150. --exec $CELERYBEAT -- $CELERYBEAT_OPTS"
  151. if [ -n "$VIRTUALENV" ]; then
  152. source $VIRTUALENV/bin/activate
  153. fi
  154. if $cmd; then
  155. log_end_msg 0
  156. else
  157. log_end_msg 1
  158. fi
  159. }
  160. case "$1" in
  161. start)
  162. check_dev_null
  163. log_daemon_msg "Starting celery periodic task scheduler" "celerybeat"
  164. start_worker
  165. ;;
  166. stop)
  167. log_daemon_msg "Stopping celery periodic task scheduler" "celerybeat"
  168. stop_worker --oknodo
  169. ;;
  170. reload|force-reload)
  171. echo "Use start+stop"
  172. ;;
  173. restart)
  174. log_daemon_msg "Restarting celery periodic task scheduler" "celerybeat"
  175. stop_worker --oknodo --retry 30
  176. check_dev_null log_end_msg
  177. start_worker
  178. ;;
  179. try-restart)
  180. log_daemon_msg "Restarting celery periodic task scheduler" "celerybeat"
  181. set +e
  182. stop_worker --retry 30
  183. RET="$?"
  184. set -e
  185. case $RET in
  186. 0)
  187. # old daemon stopped
  188. check_dev_null log_end_msg
  189. start_worker
  190. ;;
  191. 1)
  192. # daemon not running
  193. log_progress_msg "(not running)"
  194. log_end_msg 0
  195. ;;
  196. *)
  197. # failed to stop
  198. log_progress_msg "(failed to stop)"
  199. log_end_msg 1
  200. ;;
  201. esac
  202. ;;
  203. status)
  204. status_of_proc -p $CELERYBEAT_PID_FILE $CELERYBEAT celerybeat && exit 0 || exit $?
  205. ;;
  206. *)
  207. log_action_msg "Usage: /etc/init.d/celerybeat {start|stop|force-reload|restart|try-restart|status}"
  208. exit 1
  209. esac
  210. exit 0