如何给 Python 的 MySQLDB 模块增加 Timeout 超时功能

Posted 2023-05-10

使用Python操作mysql数据库的时候常使用MySQLdb这个模块。

今天在开发的过程发现MySQLdb.connect有些参数没法设置。通过这个页面我们可以看到在connect的时候，可以设置的option和client_flags和MySQL
c api相比差不少。

一个很重要的参数
MYSQL_OPT_READ_TIMEOUT没法设置，这个参数如果不设置，极致状况MySQL处于hang住，自动切换IP漂移，客户端无法重连到新MySQL。

给MySQLdb加Option很简单，只要修改_mysql.c这个把Python对象映射到MySQL操作的文件，添加参数，再加一段mysql_option即可。

下面是修改后的git diff 文件

diff --git a/_mysql.c b/_mysql.c
index d42cc54..61a9b34 100644
--- a/_mysql.c
+++ b/_mysql.c
@@ -489,9 +489,10 @@ _mysql_ConnectionObject_Initialize( "named_pipe", "init_command", "read_default_file", "read_default_group", "client_flag", "ssl",
- "local_infile",
+ "local_infile", "read_timeout",
NULL ;
int connect_timeout = 0;
+ int read_timeout = 0;
int compress = -1, named_pipe = -1, local_infile = -1;
char *init_command=NULL,
*read_default_file=NULL,
@@ -500,7 +501,7 @@ _mysql_ConnectionObject_Initialize(
self->converter = NULL;
self->open = 0;
check_server_init(-1);
- if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ssssisOiiisssiOi:connect",
+ if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ssssisOiiisssiOii:connect",
kwlist,
&host, &user, &passwd, &db,
&port, &unix_socket, &conv,
@@ -509,7 +510,8 @@ _mysql_ConnectionObject_Initialize(
&init_command, &read_default_file,
&read_default_group,
&client_flag, &ssl,
- &local_infile /* DO NOT PATCH FOR RECONNECT, IDIOTS
+ &local_infile, &read_timeout
+ /* DO NOT PATCH FOR RECONNECT, IDIOTS
IF YOU DO THIS, I WILL NOT SUPPORT YOUR PACKAGES. */
)) return -1;
@@ -540,6 +542,12 @@ _mysql_ConnectionObject_Initialize(
mysql_options(&(self->connection), MYSQL_OPT_CONNECT_TIMEOUT,
(char *)&timeout);

+
+ if (read_timeout)
+ unsigned int timeout = read_timeout;
+ mysql_options(&(self->connection), MYSQL_OPT_READ_TIMEOUT, (char *)&timeout);
+
+ if (compress != -1)
mysql_options(&(self->connection), MYSQL_OPT_COMPRESS, 0);
client_flag |= CLIENT_COMPRESS;

代码修改完毕，python
setup.py install 即可，如果出现mysql_config找不到的问题。你还要修改setup_posix.py文件。

OperationalError: (2013, \'Lost connection to MySQL server during query\')
>/home/hoterran/Projects/dbaas/trunk/dbtest.py(18)()
>mydb.execute_sql(conn, sql)
(Pdb)
--Return--
> /home/hoterran/Projects/dbaas/trunk/dbtest.py(18)()->None
> mydb.execute_sql(conn, sql)
(Pdb)
OperationalError: (2013, \'Lost connection to MySQL server during query\')
> (1)()->None

ps: 在_mysql.c找到一句很搞的话

/* DO NOT PATCH FOR RECONNECT, IDIOTS
IF YOU DO THIS, I WILL NOT SUPPORT YOUR PACKAGES. */

转载仅供参考，版权属于原作者。祝你愉快，满意请采纳哦 参考技术A 使用Python操作MySQL数据库的时候常使用MySQLdb这个模块。
今天在开发的过程发现MySQLdb.connect有些参数没法设置。通过这个页面我们可以看到在connect的时候，可以设置的option和client_flags和MySQL c api相比差不少。
一个很重要的参数 MYSQL_OPT_READ_TIMEOUT没法设置，这个参数如果不设置，极致状况MySQL处于hang住，自动切换IP漂移，客户端无法重连到新MySQL。
给MySQLdb加Option很简单，只要修改_mysql.c这个把Python对象映射到MySQL操作的文件，添加参数，再加一段mysql_option即可。本回答被提问者和网友采纳

Python MySQLdb 模块

MySQLdb 是 Python 连接 MySQL 的一个模块，常见用法如下：

[[email protected] ~]$ yum install -y MySQL-python    # 安装 MySQLdb 模块

In [1]: import MySQLdb

In [2]: conn = MySQLdb.connect(host=‘127.0.0.1‘, user=‘root‘, passwd=‘123456‘)    # connect()用于连接MySQL数据库，结果返回一个连接对象
                                                                                  # 常用的连接参数有：host 、user 、passwd 、db 、port
In [3]: cur = conn.cursor()               # 创建一个可以执行SQL语句的游标

In [4]: cur.execute(‘show databases‘)     # 使用游标来执行SQL语句，8L表示结果有8行
Out[4]: 8L

In [5]: cur.fetchone()
Out[5]: (‘information_schema‘,)           # fetchone()用于查看一条结果

In [6]: cur.fetchmany(3)
Out[6]: ((‘mysql‘,), (‘performance_schema‘,), (‘test‘,))    # fetchmany()用于查看多条结果

In [7]: cur.fetchall()
Out[7]: ((‘test1‘,), (‘test2‘,), (‘test3‘,), (‘wordpress‘,))    # fetchall()用于查看所有结果

In [7]: conn.close()    # 关闭连接