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

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

java2013liu

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

阅读量1.7k

点赞数

Appium自动化专栏收录该内容

59 篇文章

订阅专栏

实际项目中，多个方法使用MySQL连接，若每次新建、关闭连接，高访问量时可能致服务器崩溃。单例模式可解决此问题，本文采用Python单例模式方法4，因其更pythonic，还给出了代码及使用方法。

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

在实际项目中，可能会在多个不同的方法中使用MySQL链接，如果每次都新建、关闭连接，当访问量高时可能会造服务器崩溃无法访问等问题，而单例模式可以很好的解决这个问题。

关于python的单例模式这篇文章有详细的例子，这里我们使用方法4，因为这种方法更加pythonic。

代码如下：

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()