解决peewee的MySQL长连接问题

最新推荐文章于 2025-03-04 15:36:50 发布

摩力克

最新推荐文章于 2025-03-04 15:36:50 发布

阅读量3k

点赞数 3

分类专栏： python

本文链接：https://blog.csdn.net/midion9/article/details/100562651

版权

MySQL的默认长链接只能保持8小时，超过后就会自动断开。

在peewee中如何维持长连接呢？

解决方法比较晦涩，需要自定义一个支持重试的类，然后自定义一种RetryMySQLDatabase混入类

from peewee import *
from peewee import __exception_wrapper__


class RetryOperationalError(object):
    def execute_sql(self, sql, params=None, commit=True):
        try:
            cursor = super(RetryOperationalError, self).execute_sql(sql, params, commit)
        except OperationalError:
            if not self.is_closed():
                self.close()
            with __exception_wrapper__:
                cursor = self.cursor()
                cursor.execute(sql, params or ())
                if commit and not self.in_transaction():
                    self.commit()