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

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

nanshan 2024-12-30 05:59 27 浏览 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 可以方便地执行数据库操作,保持程序的高效运行。

相关推荐

0722-6.2.0-如何在RedHat7.2使用rpm安装CDH(无CM)

文档编写目的在前面的文档中,介绍了在有CM和无CM两种情况下使用rpm方式安装CDH5.10.0,本文档将介绍如何在无CM的情况下使用rpm方式安装CDH6.2.0,与之前安装C5进行对比。环境介绍:...

ARM64 平台基于 openEuler + iSula 环境部署 Kubernetes

为什么要在arm64平台上部署Kubernetes,而且还是鲲鹏920的架构。说来话长。。。此处省略5000字。介绍下系统信息;o架构:鲲鹏920(Kunpeng920)oOS:ope...

生产环境starrocks 3.1存算一体集群部署

集群规划FE:节点主要负责元数据管理、客户端连接管理、查询计划和查询调度。>3节点。BE:节点负责数据存储和SQL执行。>3节点。CN:无存储功能能的BE。环境准备CPU检查JDK...

在CentOS上添加swap虚拟内存并设置优先级

现如今很多云服务器都会自己配置好虚拟内存,当然也有很多没有配置虚拟内存的,虚拟内存可以让我们的低配服务器使用更多的内存,可以减少很多硬件成本,比如我们运行很多服务的时候,内存常常会满,当配置了虚拟内存...

国产深度(deepin)操作系统优化指南

1.升级内核随着deepin版本的更新,会自动升级系统内核,但是我们依旧可以通过命令行手动升级内核,以获取更好的性能和更多的硬件支持。具体操作:-添加PPAs使用以下命令添加PPAs:```...

postgresql-15.4 多节点主从(读写分离)

1、下载软件[root@TX-CN-PostgreSQL01-252software]#wgethttps://ftp.postgresql.org/pub/source/v15.4/postg...

Docker 容器 Java 服务内存与 GC 优化实施方案

一、设置Docker容器内存限制(生产环境建议)1.查看宿主机可用内存bashfree-h#示例输出(假设宿主机剩余16GB可用内存)#Mem:64G...

虚拟内存设置、解决linux内存不够问题

虚拟内存设置(解决linux内存不够情况)背景介绍  Memory指机器物理内存,读写速度低于CPU一个量级,但是高于磁盘不止一个量级。所以,程序和数据如果在内存的话,会有非常快的读写速度。但是,内存...

Elasticsearch性能调优(5):服务器配置选择

在选择elasticsearch服务器时,要尽可能地选择与当前业务量相匹配的服务器。如果服务器配置太低,则意味着需要更多的节点来满足需求,一个集群的节点太多时会增加集群管理的成本。如果服务器配置太高,...

Es如何落地

一、配置准备节点类型CPU内存硬盘网络机器数操作系统data节点16C64G2000G本地SSD所有es同一可用区3(ecs)Centos7master节点2C8G200G云SSD所有es同一可用区...

针对Linux内存管理知识学习总结

现在的服务器大部分都是运行在Linux上面的,所以,作为一个程序员有必要简单地了解一下系统是如何运行的。对于内存部分需要知道:地址映射内存管理的方式缺页异常先来看一些基本的知识,在进程看来,内存分为内...

MySQL进阶之性能优化

概述MySQL的性能优化,包括了服务器硬件优化、操作系统的优化、MySQL数据库配置优化、数据库表设计的优化、SQL语句优化等5个方面的优化。在进行优化之前,需要先掌握性能分析的思路和方法,找出问题,...

Linux Cgroups(Control Groups)原理

LinuxCgroups(ControlGroups)是内核提供的资源分配、限制和监控机制,通过层级化进程分组实现资源的精细化控制。以下从核心原理、操作示例和版本演进三方面详细分析:一、核心原理与...

linux 常用性能优化参数及理解

1.优化内核相关参数配置文件/etc/sysctl.conf配置方法直接将参数添加进文件每条一行.sysctl-a可以查看默认配置sysctl-p执行并检测是否有错误例如设置错了参数:[roo...

如何在 Linux 中使用 Sysctl 命令?

sysctl是一个用于配置和查询Linux内核参数的命令行工具。它通过与/proc/sys虚拟文件系统交互,允许用户在运行时动态修改内核参数。这些参数控制着系统的各种行为,包括网络设置、文件...

取消回复欢迎 发表评论: