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

你可能感兴趣的:(Python)