如何优雅地实现 Python 客户端 SQL 查询超时？

如何优雅地实现 python 客户端 sql 查询超时

在使用 python 进行 sql 查询时，确保查询不会无限执行至关重要。本篇文章将探讨如何在 python 客户端设置超时，以防止查询超过设定阈值。

解决方案

一种优雅的解决方案是利用数据库驱动的超时参数。例如，使用 pymysql 驱动，可以通过设置以下参数来实现超时：

立即学习“Python免费学习笔记（深入）”；

connection_timeout=60  # 连接超时（以秒为单位） read_timeout=60  # 读取超时（以秒为单位） write_timeout=60  # 写入超时（以秒为单位）

这些参数指定了查询执行允许的最大时间。如果超过指定时间，则会引发异常。

其他一些 orm 和驱动也提供类似的超时参数：

• django orm： default_timeout
• peewee： mysql_db.connect(timeout=60)
• sqlalchemy： connect_args={"connect_timeout": 60}

注意：

• 修改 mysql 服务器无效：问题要求只能在客户端设置超时，因此修改 mysql 服务器不起作用。
• 避免在客户端循环处理异常：虽然在客户端可以通过不断捕获超时的异常来实现自定义超时，但这不是一个优雅的解决方案。在客户端不断处理异常会影响性能和可读性。