pymysql的连接池实现

在使用pyMySQL作为MySQL驱动时,在多线程模型下,如果我们没有为每个线程创建一个单独的连接的话,就会遇到下列错误

创新互联专注于北林网站建设服务及定制,我们拥有丰富的企业做网站经验。 热诚为您提供北林营销型网站建设,北林网站制作、北林网页设计、北林网站官网定制、成都小程序开发服务,打造北林网络公司原创品牌,更为您提供北林网站排名全网营销落地服务。

pymysql.err.InternalError: Packet sequence number wrong - got 0 expected 1

这是因为pymysql的threadsafety级别为 1: Threads may share the module, but not connections.

但是为每个线程创建单独的连接的话,如果任务量比较大同时使用线程池的场景下,会造成client和MySQL server之间的频繁“连接-登录-退出”,这显然是不优雅的。所以我们需要一个“连接池”,以达到client和MySQL之间只需维持较少的连接,即可完成任务的效果。

笔者实现了一个基于pymysql的连接池模块pymysql_pool,包含两个类

  • pymysql_pool.Connection类,继承自pymysql.connections.Connection类,在保证使用方法一致的前提下,增加了对连接池的维护

  • pymysql_pool.ConnectionPool类,代表连接池,其中包含指定数目的pymysql_pool.Connection对象。这里的pymysql_pool.Connection对象相比pymysql.connections.Connection对象,增加了一个属性来标明该连接所属的连接池,也正是通过该属性来区分一个普通连接和一个池中的连接,进而对连接对象的close()方法和Context Manager Protocol进行不同的处理

项目详情请移步github


本文题目:pymysql的连接池实现
网页地址:http://pwwzsj.com/article/gjesjp.html