1. 程式人生 > >pymysql的連線池實現

pymysql的連線池實現

在使用pymysql作為MySQL驅動時,在多執行緒模型下,如果我們沒有為每個執行緒建立一個單獨的連線的話,就會遇到下列錯誤

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

但是為每個執行緒建立單獨的連線的話,如果任務量比較大同時使用執行緒池的場景下,會造成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

    進行不同的處理