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

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章