इस तरह की चीज़ के लिए एक कनेक्शन पूल अच्छा काम करता है। मैंने इसके साथ उत्पादन में काम नहीं किया है (मुख्य रूप से Django या SQLAlchemy का उपयोग करके), लेकिन psycopg2.pool
कुछ अलग कार्यान्वयन शामिल हैं (SimpleConnectionPool
या PersistentConnectionPool
) जो शायद आपकी आवश्यकता के अनुरूप होगा। सामान्यतया, एक पूल न केवल एक साझा संसाधन के रूप में कनेक्शन को प्रबंधित करने में मदद करता है, बल्कि जरूरत पड़ने पर कनेक्शन का परीक्षण और पुन:आरंभ करने में भी मदद करता है।
from psycopg2 import pool
conn_pool = pool.PersistentConnectionPool(minconn, maxconn, **dbopts)
def work_method():
conn = conn_pool.getconn()
with conn.cursor() as stmt:
stmt.execute(sql)
conn_pool.putconn(conn)
putconn
अत्यंत महत्वपूर्ण है, ताकि एक अपवाद पूल को यह सोचकर न छोड़े कि कनेक्शन अभी भी उपयोग में है। संदर्भ प्रबंधक के रूप में इसे संभालना अच्छा होगा:
import contextlib
@contextlib.contextmanager
def get_db_connection():
conn = conn_pool.getconn()
yield conn
conn_pool.putconn(conn)
def work_method():
with get_db_connection() as conn:
with conn.cursor() as stmt:
stmt.execute(sql)
आशा है कि यह मदद करता है।