from config import settings

es_client = None
pg_client = None


async def connect():
    if settings.db_established:
        from .es_client import get_es_client
        global es_client
        es_client = get_es_client()
        await es_client.connect()
    if settings.db_postgres_enable:
        from .postgres_client import get_pg_client
        global pg_client
        pg_client = get_pg_client()
        await pg_client.connect()


async def disconnect():
    global es_client, pg_client
    if es_client is not None:
        await es_client.disconnect()
        es_client = None
    if pg_client is not None:
        await pg_client.disconnect()