欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

Django数据库连接丢失问题的解决方法

程序员文章站 2022-07-06 11:54:57
问题 在django中使用mysql偶尔会出现数据库连接丢失的情况,错误通常有如下两种 operationalerror: (2006, 'mysql se...

问题

在django中使用mysql偶尔会出现数据库连接丢失的情况,错误通常有如下两种

operationalerror: (2006, 'mysql server has gone away')
operationalerror: (2013, 'lost connection to mysql server during query')

查询mysql全局变量show global variables;可以看到wait_timeout,此变量表示连接空闲时间。如果客户端使用一个连接查询多次数据库,如果连续查询则没有问题,如果查询几次后停顿超过wait_timeout后再次查询就会出现数据库连接丢失。

复现

下面用django复现下次问题:

将mysql的wait_timeout设置为10秒,然后进入django shell模拟查询(以下错误信息只保留了部分)

in[1]:import time
in[2]:from django.contrib.auth.models import user
in[3]:list(user.objects.filter(id=1))
out[3]:[<user: admin>]
in[4]:time.sleep(15) # 模拟比较慢的代码(其中没有查询数据库的代码),或者空闲什么都不操作一段时间,此时间要比`wait_timeout`大一些
list(user.objects.filter(id=1))
traceback (most recent call last):

  file "<ipython-input-4-3574ae8220ee>", line 1, in <module>
    list(user.objects.filter(id=1))

  file "/usr/lib/python3.6/site-packages/pymysql/connections.py", line 1037, in _read_bytes
    cr.cr_server_lost, "lost connection to mysql server during query")
django.db.utils.operationalerror: (2013, 'lost connection to mysql server during query')

寻求

那么以上问题就基本说明了是空闲时间过长导致的错误。

django为了减少不必要的数据库连接、关闭,复用了数据库连接,当开始一个请求后建立一个连接池存放连接,之后此次请求都复用一个连接。那猜测就是django保存连接的比wait_timeout长了,如果保存时间短一些就可以重新建立连接避免此错误了。 没错,也已经说明了此问题,设置数据库 conn_max_age参数,示例:

databases = {
 "default": {
 'engine': 'django.db.backends.mysql',
 'name': '',
 'user': '',
 'password': '',
 'host': '',
 'conn_max_age': 9 # 比wait_timeout小一些
 }
}

当我们测试后却发现,事情并非想想中那么简单。为何错误依旧出现?这一切的背后, 是人性的扭曲还是道德的沦丧?敬请收看下节《突破》。

突破

对django源码中conn_max_age进行了一番搜索,顺藤摸瓜发现了django关闭失效连接的方法django.db.close_old_connections():

# register an event to reset transaction state and close connections past
# their lifetime.
def close_old_connections(**kwargs):
 for conn in connections.all():
  conn.close_if_unusable_or_obsolete()

signals.request_started.connect(close_old_connections)
signals.request_finished.connect(close_old_connections)

重点在最后两行,通过signal实现特定事件时执行此方法,两个特定事件顾名思义是请求开始和请求结束。而我们报错的是在一次请求中,所以此法通常无效,仅仅是实现每个请求关闭并重新建立连接。

解决

复现问题的django shell不要关闭,继续执行如下代码:

in[5]:from django.db import close_old_connections
in[6]:close_old_connections()
in[7]:list(user.objects.filter(id=1))
out[7]: [<user: admin>]

调用django.db.close_old_connections后再次查询就没有错误了。 那么我们要避免此错误就要执行每个数据库查询前调用django.db.close_old_connections方法。

一般情况不会出现此类问题,因为一个请求中不间断进行数据库查询,无需每个请求调用此方法,杞人忧天。

有时候一个请求中数据量较大,会查询数据库后进行一段时间其他(不涉及数据库)处理,比如先查询一些数据,然后将数据处理、生成excel、保存文件并生成url。已知此过长需要非常长时间,那么最终url保存数据库就最好先调用django.db.close_old_connections防止连接丢失

题外话 实际上②所述情况最好从根本上解决处理慢的问题,也可以换作异步处理,从根本上解决问题。

以上就是本文的全部内容,希望对大家的学习有所帮助,也希望大家多多支持。