Spaces:

oyi77
/

1proxy

Paused

1proxy / app /database.py

Paijo

update app/database.py

ad09154 verified 9 months ago

1.73 kB

from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
from sqlalchemy.orm import DeclarativeBase
from sqlalchemy import create_engine
import os

DATABASE_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./data/1proxy.db")

# Auto-create data directory for SQLite if it doesn't exist
if "sqlite" in DATABASE_URL and "./data/" in DATABASE_URL:
    os.makedirs("./data", exist_ok=True)

# Configure connection pooling for production performance
engine_kwargs = {
    "echo": False,
    "future": True,
}

# SQLite doesn't support connection pooling, only add pool settings for PostgreSQL
if DATABASE_URL.startswith("postgresql"):
    engine_kwargs.update(
        {
            "pool_size": 20,  # Number of permanent connections
            "max_overflow": 30,  # Additional connections beyond pool_size
            "pool_pre_ping": True,  # Verify connections before using
            "pool_recycle": 3600,  # Recycle connections after 1 hour
            "pool_timeout": 30,  # Wait max 30s for connection from pool
        }
    )

engine = create_async_engine(DATABASE_URL, **engine_kwargs)

AsyncSessionLocal = async_sessionmaker(
    engine,
    class_=AsyncSession,
    expire_on_commit=False,
)


class Base(DeclarativeBase):
    pass


async def get_db():
    async with AsyncSessionLocal() as session:
        try:
            yield session
            await session.commit()
        except Exception:
            await session.rollback()
            raise
        finally:
            await session.close()


async def init_db():
    async with engine.begin() as conn:
        await conn.run_sync(Base.metadata.create_all)

	from sqlalchemy.ext.asyncio import create_async_engine, AsyncSession, async_sessionmaker
	from sqlalchemy.orm import DeclarativeBase
	from sqlalchemy import create_engine
	import os

	DATABASE_URL = os.getenv("DATABASE_URL", "sqlite+aiosqlite:///./data/1proxy.db")

	# Auto-create data directory for SQLite if it doesn't exist
	if "sqlite" in DATABASE_URL and "./data/" in DATABASE_URL:
	os.makedirs("./data", exist_ok=True)

	# Configure connection pooling for production performance
	engine_kwargs = {
	"echo": False,
	"future": True,
	}

	# SQLite doesn't support connection pooling, only add pool settings for PostgreSQL
	if DATABASE_URL.startswith("postgresql"):
	engine_kwargs.update(
	{
	"pool_size": 20, # Number of permanent connections
	"max_overflow": 30, # Additional connections beyond pool_size
	"pool_pre_ping": True, # Verify connections before using
	"pool_recycle": 3600, # Recycle connections after 1 hour
	"pool_timeout": 30, # Wait max 30s for connection from pool
	}
	)

	engine = create_async_engine(DATABASE_URL, **engine_kwargs)

	AsyncSessionLocal = async_sessionmaker(
	engine,
	class_=AsyncSession,
	expire_on_commit=False,
	)


	class Base(DeclarativeBase):
	pass


	async def get_db():
	async with AsyncSessionLocal() as session:
	try:
	yield session
	await session.commit()
	except Exception:
	await session.rollback()
	raise
	finally:
	await session.close()


	async def init_db():
	async with engine.begin() as conn:
	await conn.run_sync(Base.metadata.create_all)