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分钟。。。