百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 技术文章 > 正文

Python异步 操作MySQL 连接池,打破数据库瓶颈!提升性能

nanshan 2024-12-30 05:59 24 浏览 0 评论

引言:

AsyncMysql 类实现的功能是通过异步方式连接、查询、操作 MySQL 数据库,利用 aiomysql 库提供的异步支持来提升数据库操作的效率,尤其适用于需要高并发访问数据库的场景。

准备(mysql数据库信息):

D盘或者其他路径 创建一个文件Database.yaml内容如下:

host: 127.0.0.1
port: 3306
user: sa
password: 密码1
database: mysql

封装代码如下:

import asyncio  # 导入 asyncio 库,用于异步编程
import aiomysql  # 导入 aiomysql 库,用于异步 MySQL 操作
from Public_Config.PublicConfig import ConfigYaml  # 导入配置文件模块,获取 MySQL 配置信息

class AsyncMysql:

    def __init__(self):
        # 获取 MySQL 配置信息
        mysql = self.load_config(r'D:\Database.yaml')  # 从配置文件中读取 MySQL 配置
        self.host = mysql.get('host')  # 主机地址
        self.port = int(mysql.get('port'))  # 端口,转为整数
        self.user = mysql.get('user')  # 用户名
        self.password = mysql.get('password')  # 密码
        self.database = mysql('database')  # 数据库名
        self.pool = None  # 初始化连接池为 None,稍后创建连接池
       
    def load_config(self, config_file):   #todo 加载配置文件的方法
        with open(config_file, 'r') as file:  #todo 以只读模式打开配置文件
            return yaml.safe_load(file)  #todo 使用 yaml.safe_load 解析 YAML 文件内容

    async def create_pool(self):
        """ 创建 MySQL 连接池 """
        try:
            # 创建 MySQL 连接池,设置最小连接数 minsize 和最大连接数 maxsize
            self.pool = await aiomysql.create_pool(
                host=self.host,  # 主机
                port=self.port,  # 端口
                user=self.user,  # 用户名
                password=self.password,  # 密码
                db=self.database,  # 数据库名
                autocommit=True,  # 自动提交事务
                minsize=2,  # 设置最小连接数
                maxsize=10  # 设置最大连接数
            )
        except Exception as e:
            # 如果创建连接池失败,捕获异常并输出错误信息
            print(f"Failed to create MySQL connection pool: {str(e)}")

    async def execute_query(self, query):
        """ 执行增删改操作 """
        try:
            # 从连接池中获取一个连接对象
            async with self.pool.acquire() as conn:
                # 使用连接对象创建游标,用于执行 SQL 查询
                async with conn.cursor() as cursor:
                    await cursor.execute(query)  # 执行 SQL 查询
                    return True  # 返回执行成功标志
        except Exception as e:
            # 如果执行 SQL 失败,捕获异常并输出错误信息
            print(f"Failed to execute query: {query}")
            print(f"Error: {str(e)}")
            return False  # 执行失败时返回 False

    async def execute_select(self, query):
        """ 执行查询操作 """
        try:
            # 从连接池中获取一个连接对象
            async with self.pool.acquire() as conn:
                # 使用连接对象创建游标,用于执行 SQL 查询
                async with conn.cursor() as cursor:
                    await cursor.execute(query)  # 执行 SQL 查询
                    results = await cursor.fetchall()  # 获取所有查询结果
                    return results  # 返回查询结果
        except Exception as e:
            # 如果执行查询失败,捕获异常并输出错误信息
            print(f"Failed to execute select query: {query}")
            print(f"Error: {str(e)}")
            return None  # 执行失败时返回 None

    async def close_pool(self):
        """ 关闭 MySQL 连接池 """
        if self.pool:
            self.pool.close()  # 关闭连接池
            await self.pool.wait_closed()  # 等待连接池完全关闭
            print("MySQL connection pool closed.")  # 输出连接池已关闭的信息

# 测试异步操作
async def main():
    mysql = AsyncMysql()  # 创建 AsyncMysql 实例
    await mysql.create_pool()  # 异步创建连接池

    # 定义 SQL 查询语句
    query1 = "SELECT * FROM USERLOGINLOG LIMIT 5"  # 查询 USERLOGINLOG 表的前 5 条记录
    query2 = "SELECT COUNT(*) FROM USERLOGINLOG"  # 查询 USERLOGINLOG 表的记录总数
    
    # 使用 asyncio.gather 同时执行多个查询操作
    results = await asyncio.gather(
        mysql.execute_select(query1),  # 执行第一个查询
        mysql.execute_select(query2)   # 执行第二个查询
    )

    # 打印查询结果
    print("Query 1 results:", results[0])  # 输出第一个查询的结果
    print("Query 2 results:", results[1])  # 输出第二个查询的结果

    # 关闭连接池
    await mysql.close_pool()  # 异步关闭连接池

# 运行异步任务
if __name__ == '__main__':
    asyncio.run(main())  # 启动并运行 main 异步任务

功能总结

该AsyncMysql 类主要用于处理 MySQL 数据库的连接池管理和异步数据库操作,具有以下特点:

  1. 连接池管理:通过异步创建和关闭数据库连接池,提高连接的复用性和操作效率。
  2. 异步操作:使用 aiomysql 和 asyncio 实现非阻塞的数据库操作,可以同时执行多个查询或操作,不会阻塞主程序,适用于高并发环境。
  3. 增删改查操作支持:支持执行数据库的增删改查操作,且能处理 SQL 查询结果。
  4. 自动提交事务:每次执行 SQL 操作时,自动提交事务,避免需要手动调用 commit()。

应用场景

这个类适用于以下几种情况:

  • 高并发场景:比如需要频繁访问数据库的 Web 应用或服务,异步操作可以有效提高数据库操作的并发性,提升性能。
  • 大规模数据查询和操作:尤其是在需要大量数据库查询或更新操作的场景,使用连接池可以减少连接的建立和销毁开销,提升整体效率。
  • 异步 Web 框架的数据库访问:比如在使用 FastAPI、Sanic 等异步框架时,通过 aiomysql 可以方便地执行数据库操作,保持程序的高效运行。

相关推荐

Let’s Encrypt免费搭建HTTPS网站

HTTPS(全称:HyperTextTransferProtocoloverSecureSocketLayer),是以安全为目标的HTTP通道,简单讲是HTTP的安全版。即HTTP下加入...

使用Nginx配置TCP负载均衡(nginx tcp负载)

假设Kubernetes集群已经配置好,我们将基于CentOS为Nginx创建一个虚拟机。以下是实验种设置的详细信息:Nginx(CenOS8Minimal)-192.168.1.50Kube...

Nginx负载均衡及支持HTTPS与申请免费SSL证书

背景有两台minio文件服务器已做好集群配置,一台是192.168.56.41:9000;另一台是192.168.56.42:9000。应用程序通过Nginx负载均衡调用这两台minio服务,减轻单点...

HTTPS配置实战(https配置文件)

原因现在网站使用HTTPS是规范操作之一,前些日子买了腾讯云服务,同时申请了域名http://www.asap2me.top/,目前该域名只支持HTTP,想升级为HTTPS。关于HTTPS的链接过程大...

只有IP地址没有域名实现HTTPS访问方法

一般来说,要实现HTTPS,得有个注册好的域名才行。但有时候呢,咱只有服务器的IP地址,没注册域名,这种特殊情况下,也能照样实现HTTPS安全访问,按下面这些步骤来就行:第一步,先确认公网...

超详解:HTTPS及配置Django+HTTPS开发环境

众所周知HTTP协议是以TCP协议为基石诞生的一个用于传输Web内容的一个网络协议,在“网络分层模型”中属于“应用层协议”的一种。在这里我们并不研究该协议标准本身,而是从安全角度去探究使用该协议传输数...

Godaddy购买SSL之后Nginx配置流程以及各种错误的解决

完整流程:参考地址:https://sg.godaddy.com/zh/help/nginx-generate-csrs-certificate-signing-requests-3601生成NGI...

Nginx从安装到高可用,一篇搞定(nginx安装与配置详解)

一、Nginx安装1、去官网http://nginx.org/下载对应的nginx包,推荐使用稳定版本2、上传nginx到linux系统3、安装依赖环境(1)安装gcc环境yuminstallgc...

阿里云免费证书申请,配置安装,使用tomcat,支持http/https访问

参数说明商品类型默认已选择云盾证书服务(无需修改)。云盾证书服务类型SSL证书服务的类型。默认已选择云盾SSL证书(无需修改),表示付费版SSL证书。如果您需要免费领取或付费扩容DV单域名证书【免费试...

你试过两步实现Nginx的规范配置吗?极速生成Nginx配置小工具

NGINX是一款轻量级的Web服务器,最强大的功能之一是能够有效地提供HTML和媒体文件等静态内容。NGINX使用异步事件驱动模型,在负载下提供可预测的性能。是当下最受欢迎的高性能的Web...

从零开始搭建HTTPS服务(搭建https网站)

搭建HTTPS服务的最初目的是为了开发微信小程序,因为wx.request只允许发起HTTPS请求,并且还必须和指定的域名进行网络通信。要从零开始搭建一个HTTPS的服务需要下面4...

群晖NAS使用官网域名和自己的域名配置SSL实现HTTPS访问

安全第一步,群晖NAS使用官网域名和自己的域名配置SSL实现HTTPS访问【新手导向】NAS本质还是一个可以随时随地访问的个人数据存储中心,我们在外网访问的时候,特别是在公网IP下,其实会面临着很多安...

让网站快速升级HTTPS协议提高安全性

为什么用HTTPS网络安全越来越受到重视,很多互联网服务网站,都已经升级改造为https协议。https协议下数据包是ssl/tcl加密的,而http包是明文传输。如果请求一旦被拦截,数据就会泄露产生...

用Https方式访问Harbor-1.9版本(https访问流程)

我上周在头条号写过一篇原创文章《Docker-Harbor&Docker-kitematic史上最详细双系统配置手册》,这篇算是它的姊妹篇吧。这篇文章也将用到我在头条写的另一篇原创文章的...

如何启用 HTTPS 并配置免费的 SSL 证书

在Linux服务器上启用HTTPS并配置免费的SSL证书(以Let'sEncrypt为例)可以通过以下步骤完成:---###**一、准备工作**1.**确保域名已解析**...

取消回复欢迎 发表评论: