pymysql的連線池實現
阿新 • • 發佈:2018-12-30
在使用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