如何给Python的MySQL模块加功能

本文涉及的产品
RDS MySQL Serverless 基础系列,0.5-2RCU 50GB
云数据库 RDS MySQL,集群系列 2核4GB
推荐场景:
搭建个人博客
云数据库 RDS MySQL,高可用系列 2核4GB
简介:

使用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 文件

 

  1. diff --git a/_mysql.c b/_mysql.c  
  2. index d42cc54..61a9b34 100644  
  3. --- a/_mysql.c  
  4. +++ b/_mysql.c  
  5. @@ -489,9 +489,10 @@ _mysql_ConnectionObject_Initialize(  
  6. "named_pipe""init_command",  
  7. "read_default_file""read_default_group",  
  8. "client_flag""ssl",  
  9. -                                 "local_infile",  
  10. +                                 "local_infile""read_timeout",  
  11. NULL } ;  
  12. int connect_timeout = 0;  
  13. +       int read_timeout = 0;  
  14. int compress = -1, named_pipe = -1, local_infile = -1;  
  15. char *init_command=NULL,  
  16. *read_default_file=NULL,  
  17. @@ -500,7 +501,7 @@ _mysql_ConnectionObject_Initialize(  
  18. self->converter = NULL;  
  19. self->open = 0;  
  20. check_server_init(-1);  
  21. -       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ssssisOiiisssiOi:connect",  
  22. +       if (!PyArg_ParseTupleAndKeywords(args, kwargs, "|ssssisOiiisssiOii:connect",  
  23. kwlist,  
  24. &host, &user, &passwd, &db,  
  25. &port, &unix_socket, &conv,  
  26. @@ -509,7 +510,8 @@ _mysql_ConnectionObject_Initialize(  
  27. &init_command, &read_default_file,  
  28. &read_default_group,  
  29. &client_flag, &ssl,  
  30. -                                        &local_infile /* DO NOT PATCH FOR RECONNECT, IDIOTS  
  31. +                                        &local_infile, &read_timeout  
  32. +                                        /* DO NOT PATCH FOR RECONNECT, IDIOTS  
  33. IF YOU DO THIS, I WILL NOT SUPPORT YOUR PACKAGES. */  
  34. ))  
  35. return -1;  
  36. @@ -540,6 +542,12 @@ _mysql_ConnectionObject_Initialize(  
  37. mysql_options(&(self->connection), MYSQL_OPT_CONNECT_TIMEOUT,  
  38. (char *)&timeout);  
  39. }  
  40. +  
  41. +        if (read_timeout) {  
  42. +                unsigned int timeout = read_timeout;  
  43. +                mysql_options(&(self->connection), MYSQL_OPT_READ_TIMEOUT, (char *)&timeout);  
  44. +        }  
  45. +  
  46. if (compress != -1) {  
  47. mysql_options(&(self->connection), MYSQL_OPT_COMPRESS, 0);  
  48. client_flag |= CLIENT_COMPRESS;

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

 

  1. hoterran@hoterran-laptop:~/Projects/MySQL-python-1.2.3$ git diff setup_posix.py 
  2. diff --git a/setup_posix.py b/setup_posix.py 
  3. index 86432f5..f4f08f1 100644 
  4. --- a/setup_posix.py 
  5. +++ b/setup_posix.py 
  6. @@ -23,7 +23,7 @@ def mysql_config(what): 
  7. if ret/256 > 1
  8. raise EnvironmentError("%s not found" % (mysql_config.path,)) 
  9. return data 
  10. -mysql_config.path = "mysql_config" 
  11. +mysql_config.path = "/usr/local/mysql/bin/mysql_config" 
  12. def get_config(): 
  13. import os, sys

 

  编译通过,我们来试试添加的read_timeout这个参数。

 

conn = MySQLdb.connect(host = DB_SERVER,user = DB_USERNAME,passwd = DB_PASSWORD,db = DB_NAME, port=int(DB_PORT), client_flag = 2, read_timeout = 10)

 

  然后执行语句前,你试着把mysql用gdb hang住10s后,python就会异常抛错

 

  1. OperationalError: (2013'Lost connection to MySQL server during query'
  2. >/home/hoterran/Projects/dbaas/trunk/dbtest.py(18)() 
  3. >mydb.execute_sql(conn, sql) 
  4. (Pdb) 
  5. --Return-- 
  6. > /home/hoterran/Projects/dbaas/trunk/dbtest.py(18)()->None 
  7. > mydb.execute_sql(conn, sql) 
  8. (Pdb) 
  9. OperationalError: (2013'Lost connection to MySQL server during query'
  10. > (1)()->None









本文转自 小强测试帮 51CTO博客,原文链接:http://blog.51cto.com/xqtesting/836469,如需转载请自行联系原作者
相关实践学习
如何快速连接云数据库RDS MySQL
本场景介绍如何通过阿里云数据管理服务DMS快速连接云数据库RDS MySQL,然后进行数据表的CRUD操作。
全面了解阿里云能为你做什么
阿里云在全球各地部署高效节能的绿色数据中心,利用清洁计算为万物互联的新世界提供源源不断的能源动力,目前开服的区域包括中国(华北、华东、华南、香港)、新加坡、美国(美东、美西)、欧洲、中东、澳大利亚、日本。目前阿里云的产品涵盖弹性计算、数据库、存储与CDN、分析与搜索、云通信、网络、管理与监控、应用服务、互联网中间件、移动服务、视频服务等。通过本课程,来了解阿里云能够为你的业务带来哪些帮助     相关的阿里云产品:云服务器ECS 云服务器 ECS(Elastic Compute Service)是一种弹性可伸缩的计算服务,助您降低 IT 成本,提升运维效率,使您更专注于核心业务创新。产品详情: https://www.aliyun.com/product/ecs
目录
打赏
0
0
0
0
235
分享
相关文章
MySQL细节优化:关闭大小写敏感功能的方法。
通过这种方法,你就可以成功关闭 MySQL 的大小写敏感功能,让你的数据库操作更加便捷。
51 19
[oeasy]python071_我可以自己做一个模块吗_自定义模块_引入模块_import_diy
本文介绍了 Python 中模块的导入与自定义模块的创建。首先,我们回忆了模块的概念,即封装好功能的部件,并通过导入 `__hello__` 模块实现了输出 "hello world!" 的功能。接着,尝试创建并编辑自己的模块 `my_file.py`,引入 `time` 模块以获取当前时间,并在其中添加自定义输出。
30 4
[oeasy]python070_如何导入模块_导入模块的作用_hello_dunder_双下划线
本文介绍了如何在Python中导入模块及其作用,重点讲解了`__hello__`模块的导入与使用。通过`import`命令可以将外部模块引入当前环境,增强代码功能。例如,导入`__hello__`模块后可输出“Hello world!”。此外,还演示了如何使用`help()`和`dir()`函数查询模块信息,并展示了导入多个模块的方法。最后,通过一个实例,介绍了如何利用`jieba`、`WordCloud`和`matplotlib`模块生成词云图。总结来说,模块是封装好的功能部件,能够简化编程任务并提高效率。未来将探讨如何创建自定义模块。
40 8
Python中使用MySQL模糊查询的方法
本文介绍了两种使用Python进行MySQL模糊查询的方法:一是使用`pymysql`库,二是使用`mysql-connector-python`库。通过这两种方法,可以连接MySQL数据库并执行模糊查询。具体步骤包括安装库、配置数据库连接参数、编写SQL查询语句以及处理查询结果。文中详细展示了代码示例,并提供了注意事项,如替换数据库连接信息、正确使用通配符和关闭数据库连接等。确保在实际应用中注意SQL注入风险,使用参数化查询以保障安全性。
通义灵码 2.0 智能编码功能评测:Deepseek 加持下的 Python 开发体验
通义灵码 2.0 智能编码功能评测:Deepseek 加持下的 Python 开发体验
114 11
|
3月前
|
[oeasy]python057_如何删除print函数_dunder_builtins_系统内建模块
本文介绍了如何删除Python中的`print`函数,并探讨了系统内建模块`__builtins__`的作用。主要内容包括: 1. **回忆上次内容**:上次提到使用下划线避免命名冲突。 2. **双下划线变量**:解释了双下划线(如`__name__`、`__doc__`、`__builtins__`)是系统定义的标识符,具有特殊含义。
45 3
Python 语言结合 Flask 框架来实现一个基础的代购商品管理、用户下单等功能的简易系统
这是一个使用 Python 和 Flask 框架实现的简易代购系统示例,涵盖商品管理、用户注册登录、订单创建及查看等功能。通过 SQLAlchemy 进行数据库操作,支持添加商品、展示详情、库存管理等。用户可注册登录并下单,系统会检查库存并记录订单。此代码仅为参考,实际应用需进一步完善,如增强安全性、集成支付接口、优化界面等。
|
4月前
|
Python Internet 模块
Python Internet 模块。
147 74

热门文章

最新文章

AI助理

你好,我是AI助理

可以解答问题、推荐解决方案等