from aioch import Client

from core.config import settings
from .ckdb import CKDrive


async def connect_to_ck(pool_size=100):
    for i in range(pool_size):
        client = Client(**settings.CK_CONFIG)
        CKDrive.ClientPool[i] = client


async def close_ck_connection():
    await CKDrive.client.disconnect()