From bd3255760b24bf9f3772415afd2e87c7a12a91e6 Mon Sep 17 00:00:00 2001 From: Maxim Bublis Date: Mon, 2 Jul 2018 12:34:32 -0400 Subject: Use MySQL protocol-level ping. Utilizes MySQL protocol-level pings for disconnection detection. This is just a 5-byte packet followed by a 7-byte response. Affects MySQLdb, MySQL Connector and PyMySQL dialects. Change-Id: I672f75e3746878d88987a31750444dde0cf8eb9b Pull-request: https://github.com/zzzeek/sqlalchemy/pull/460 --- lib/sqlalchemy/dialects/mysql/pymysql.py | 8 ++++++++ 1 file changed, 8 insertions(+) (limited to 'lib/sqlalchemy/dialects/mysql/pymysql.py') diff --git a/lib/sqlalchemy/dialects/mysql/pymysql.py b/lib/sqlalchemy/dialects/mysql/pymysql.py index 4f1c792f9..5f176cef2 100644 --- a/lib/sqlalchemy/dialects/mysql/pymysql.py +++ b/lib/sqlalchemy/dialects/mysql/pymysql.py @@ -61,6 +61,14 @@ class MySQLDialect_pymysql(MySQLDialect_mysqldb): def dbapi(cls): return __import__('pymysql') + def is_disconnect(self, e, connection, cursor): + if super(MySQLDialect_pymysql, self).is_disconnect(e, connection, cursor): + return True + elif isinstance(e, self.dbapi.Error): + return "Already closed" in str(e) + else: + return False + if py3k: def _extract_error_code(self, exception): if isinstance(exception.args[0], Exception): -- cgit v1.2.1