celery worker闲置24小时后,数据库连接错误

时间:2019-05-08 07:38:25

标签: mysql python-3.x django-celery confluent-kafka

我有一个基于Django的Web应用程序,我在其中使用Kafka处理一些订单。现在,我使用Celery Workers为每个主题分配一个Kafka使用者。每个Kafka使用者都以Kafka任务的形式分配给Kafka主题。但是,大约一天后,当我提交任务时,出现以下错误:

    _mysql.connection.query(self, query)
_mysql_exceptions.OperationalError: (2006, 'MySQL server has gone away')

The above exception was the direct cause of the following exception:

下面是我的task.py文件的外观:

@shared_task
def init_kafka_consumer(topic):
    try:
        if topic is None:
            raise Exception("Topic is none, unable to initialize kafka consumer")
        logger.info("Spawning new task to subscribe to topic")
        params = []
        params.append(topic)
        background_thread = Thread(target=sunscribe_consumer, args=params)
        background_thread.start()
    except Exception :
        logger.exception("An exception occurred while reading message from kafka")

def sunscribe_consumer(topic) :
    try:
        if topic is None:
            raise Exception("Topic is none, unable to initialize kafka consumer")
        conf = {'bootstrap.servers': "localhost:9092", 'group.id': 'test', 'session.timeout.ms': 6000,
                'auto.offset.reset': 'earliest'}
        c = Consumer(conf)
        logger.info("Subscribing consumer to topic "+str(topic[0]))
        c.subscribe(topic)
        # Read messages from Kafka
        try:
            while True:
                msg = c.poll(timeout=1.0)
                if msg is None:
                    continue
                if msg.error():
                    raise KafkaException(msg.error())
                else:
                    try:
                        objs = serializers.deserialize("json", msg.value())
                        for obj in objs:
                            order = obj.object
                            order = BuyOrder.objects.get(id=order.id) #Getting an error while accessing DB
                            if order.is_pushed_to_kafka :
                                return
                            order.is_pushed_to_kafka = True
                            order.save()
                            from web3 import HTTPProvider, Web3, exceptions
                            w3 = Web3(HTTPProvider(INFURA_MAIN_NET_ETH_URL))
                            processBuyerPayout(order,w3)
                    except Exception :
                        logger.exception("An exception occurred while de-serializing message")
        except Exception :
            logger.exception("An exception occurred while reading message from kafka")
        finally:
            c.close()
    except Exception :
        logger.exception("An exception occurred while reading message from kafka")

无论如何,一旦收到任务,我是否可以检查数据库连接是否存在,如果没有,我可以重新建立连接吗?

1 个答案:

答案 0 :(得分:0)

根据https://github.com/celery/django-celery-results/issues/58#issuecomment-418413369 和上面放置此代码的注释:

from django.db import close_old_connections
close_old_connections()

关闭旧连接并在任务内部打开新连接应该会有所帮助。

相关问题