celery中的时区陷阱

Django中使用celery执行异步任务非常方便,通过apply_async可以控制定时任务的执行时间。一般使用场景中,比较多的是任务在一定时间之后执行,这时只要指定countdown参数就可以了,eta参数能够指定任务执行的绝对时间,由于在项目中使用很少,因此一些特性之前没注意到。

eta默认为UTC时间

假设在celery 4.x 版本中定义如下任务

@app.task(queue='celery')
def hello():
    print '==========='
    print 'hello world'
    print '==========='

调用hello.apply_async(eta=datetime.datetime.now())时任务不会立即执行,实际情况是过8小时之后才会执行。查了celery源码,发现eta参数在消费者也就是web server发消息时基本会直接设置在amqp消息中,默认为UTC时间。

if countdown:  # convert countdown to ETA
    self._verify_seconds(countdown, 'countdown')
    now = now or self.app.now()
    eta = now + timedelta(seconds=countdown)
if isinstance(expires, numbers.Real):
    self._verify_seconds(expires, 'expires')
    now = now or self.app.now()
    expires = now + timedelta(seconds=expires)
eta = eta and eta.isoformat()

当worker收到消息后,会根据本地的时区信息将UTC转换为本地时间进行执行
其中to_system_tz就是转换为系统也即本地时间

if req.utc:
    eta = to_timestamp(to_system_tz(req.eta))
else:
    eta = to_timestamp(req.eta, app.timezone)

countdown 为什么没问题

由上面代码可以看出,countdown也会转换为eta存储在消息中,而countdown之所以没问题,是因为基本时间即now为UTC时间。

celery时区相关配置

celery中和时区相关的配置主要有[两个]

  • CELERY_ENABLE_UTC 开启后消息的时间将会按照UTC时区转化,默认开启
  • ** CELERY_TIMEZONE** 默认为UTC,可以是任何[pytz]支持的时区

设置本地时区并不能解决上面的问题,因为eta就没有进行过时区转换,完全是按照用户传进来的时间进行计算

解决方法

可以采用下面的代码进行解决,时区为Asia/Shanghai

import datetime
from django.utils.timezone import get_current_timezone

n = datetime.datetime.now()
tz = get_current_timezone()
n = tz.localize(n)
hello.apply_async(eta=n)

需要注意的是通过n.replace(tzinfo=tz)进行替换的话,总是会和期望的时间相差6分钟,而这又是另外一个问题。pytz中上海时间比北京时间提前了6分钟。。。

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 概念解释 时区(Timezone) 为什么会将地球分为不同时区呢?因为地球总是自西向东自转,东边总比西边先看到太阳...
    黑炭妈阅读 6,939评论 0 5
  • Celery 进阶 前面已经对Celery的简单使用和配置做了介绍,本章将会展示Celery的更多细节 在自己的应...
    0ooops阅读 5,896评论 0 7
  • 一. celery 简介 Celery 是一个专注于实时处理和任务调度的分布式任务队列, 同时提供操作和维护分布式...
    眼睛好酸阅读 10,582评论 1 11
  • 转https://blog.csdn.net/kk123a/article/details/74549117 一....
    你常不走的路阅读 18,469评论 2 14
  • 睁开眼睛,已经七点多了,太阳还没有完全出来,天边已经泛红,看样又是一个大晴天。 ...
    静起风云阅读 178评论 1 0