49 lines
1.2 KiB
Python
49 lines
1.2 KiB
Python
import os
|
|
import logging
|
|
import psycopg2
|
|
from psycopg2 import pool
|
|
|
|
logger = logging.getLogger(__name__)
|
|
|
|
# Connection pool for reusing connections
|
|
_connection_pool = None
|
|
|
|
|
|
def get_connection_pool():
|
|
"""Create a connection pool"""
|
|
global _connection_pool
|
|
if _connection_pool is None:
|
|
try:
|
|
DATABASE_URL = os.environ.get("DATABASE_URL")
|
|
_connection_pool = psycopg2.pool.SimpleConnectionPool(
|
|
1, 5, DATABASE_URL)
|
|
|
|
logger.info("Database connection pool created")
|
|
except Exception as e:
|
|
logger.error(f"Failed to create connection pool: {e}")
|
|
return None
|
|
return _connection_pool
|
|
|
|
|
|
def get_db_connection():
|
|
"""Get a connection from the pool"""
|
|
try:
|
|
pool = get_connection_pool()
|
|
if pool:
|
|
conn = pool.getconn()
|
|
return conn
|
|
return None
|
|
except Exception as e:
|
|
logger.error(f"Database connection error: {e}")
|
|
return None
|
|
|
|
|
|
def release_db_connection(conn):
|
|
"""Return connection to the pool"""
|
|
try:
|
|
pool = get_connection_pool()
|
|
if pool and conn:
|
|
pool.putconn(conn)
|
|
except Exception as e:
|
|
logger.error(f"Error releasing connection: {e}")
|