python使用单例模式创建MySQL链接

最新推荐文章于 2024-04-27 08:50:14 发布

lislislislislis

最新推荐文章于 2024-04-27 08:50:14 发布

阅读量827

点赞数

CC 4.0 BY-SA版权

分类专栏： python 文章标签： python MySQL

本文链接：https://blog.youkuaiyun.com/lislislislislis/article/details/89065090

python 专栏收录该内容

56 篇文章

订阅专栏

本文介绍了如何在Python中利用单例模式创建并管理MySQL数据库连接，确保在整个应用中只存在一个数据库连接实例，提高资源利用率和代码的稳定性。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

from functools import wraps
import mysql.connector
from sshtunnel import SSHTunnelForwarder


def singleton(cls):
    instances = {}

    @wraps(cls)
    def get_instance(*args, **kw):
        if cls not in instances:
            instances[cls] = cls(*args, **kw)
        return instances[cls]
    return get_instance


# 数据库连接实例
@singleton
class MySQLSingle(object):
    def __init__(self, conn='', server=''):
        self.conn = conn
        self.server = server

    def get_conn(self, host_jump, port_jump, ssh_pk_jump, user_name_jump, host_mysql, port_mysql, user_name_mysql, password_mysql, database):
        try:
            self.server = SSHTunnelForwarder(
                (host_jump, int(port_jump)),  # 跳板机的配置
                ssh_pkey=ssh_pk_jump,
                ssh_username=user_name_jump,
                remote_bind_address=(host_mysql, int(port_mysql)))  # 数据库服务器的配置
            self.server.start()
            self.conn = mysql.connector.connect(host='127.0.0.1', port=self.server.local_bind_port, user=user_name_mysql,
                                           password=password_mysql, database=database)
        except Exception as e:
            print('File to connect database: %s' % e)
        return self.conn, self.server

使用方法

mysql_single = MySQLSingle()
conn, server = mysql_single.get_conn(host_jump, port_jump, ssh_pk_jump, user_name_jump, host_mysql, port_mysql,
                                     user_name_mysql, password_mysql, database)
# do something...
conn.close()
server.close()