python如何判断与数据库断连

在Python中判断与数据库断连有多种方法。捕获数据库连接异常、使用心跳机制、轮询数据库状态、利用数据库驱动的连接池功能是其中常见的方法。下面我们将详细介绍其中一种：捕获数据库连接异常。

一、捕获数据库连接异常

在与数据库进行交互时，异常捕获是一种常见且有效的方法。通过捕获数据库操作中的异常，可以判断当前连接是否有效。如果捕获到特定的数据库连接异常，则表示与数据库的连接已经断开。

import mysql.connector
from mysql.connector import Error
try:
    connection = mysql.connector.connect(
        host='localhost',
        database='test_db',
        user='test_user',
        password='test_password'
    )
    if connection.is_connected():
        print("Connection is established")
        cursor = connection.cursor()
        cursor.execute("SELECT * FROM test_table;")
        records = cursor.fetchall()
        print(records)
except Error as e:
    print(f"Error: {e}")
finally:
    if connection.is_connected():
        cursor.close()
        connection.close()
        print("MySQL connection is closed")

在上面的代码中，通过捕获 mysql.connector.Error 异常来判断数据库连接是否断开，并在 finally 块中关闭连接。

二、使用心跳机制

心跳机制是一种通过定期发送测试查询来检查数据库连接状态的方法。可以通过在应用程序中定期发送简单的查询（例如 SELECT 1）来检测连接是否仍然有效。如果查询失败，则表示连接已断开。

import time
import mysql.connector
from mysql.connector import Error
def check_connection(connection):
    try:
        cursor = connection.cursor()
        cursor.execute("SELECT 1")
        cursor.close()
        return True
    except Error:
        return False
try:
    connection = mysql.connector.connect(
        host='localhost',
        database='test_db',
        user='test_user',
        password='test_password'
    )
    if connection.is_connected():
        print("Connection is established")
    while True:
        if check_connection(connection):
            print("Connection is still alive")
        else:
            print("Connection is lost")
            break
        time.sleep(10)
except Error as e:
    print(f"Error: {e}")
finally:
    if connection.is_connected():
        connection.close()
        print("MySQL connection is closed")

三、轮询数据库状态

轮询数据库状态是一种通过定期检查数据库连接对象的状态来确定连接是否有效的方法。这可以通过在应用程序中设置定时器或调度任务来实现，定期检查连接对象的状态。

import time
import mysql.connector
from mysql.connector import Error
def poll_connection_status(connection):
    while True:
        try:
            cursor = connection.cursor()
            cursor.execute("SELECT 1")
            cursor.close()
            print("Connection is still alive")
        except Error:
            print("Connection is lost")
            break
        time.sleep(10)
try:
    connection = mysql.connector.connect(
        host='localhost',
        database='test_db',
        user='test_user',
        password='test_password'
    )
    if connection.is_connected():
        print("Connection is established")
        poll_connection_status(connection)
except Error as e:
    print(f"Error: {e}")
finally:
    if connection.is_connected():
        connection.close()
        print("MySQL connection is closed")

四、利用数据库驱动的连接池功能

许多数据库驱动程序提供了连接池功能，可以通过配置连接池来自动管理数据库连接。连接池通常会自动检测和回收失效的连接，并提供新的连接。

from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from sqlalchemy.exc import OperationalError
engine = create_engine('mysql+mysqlconnector://test_user:test_password@localhost/test_db', pool_size=5, max_overflow=10)
Session = sessionmaker(bind=engine)
session = Session()
def check_connection():
    try:
        session.execute('SELECT 1')
        print("Connection is still alive")
    except OperationalError:
        print("Connection is lost")
while True:
    check_connection()
    time.sleep(10)

总结

在Python中判断与数据库断连的方法有多种，捕获数据库连接异常是最简单直接的方法，使用心跳机制可以定期检查连接状态，轮询数据库状态则通过定时器或调度任务来实现，而利用数据库驱动的连接池功能则可以自动管理连接。这些方法可以根据实际需求进行选择和结合使用，以确保与数据库的连接稳定可靠。