# Copyright 2018 TrilioData Inc.
# All Rights Reserved.

from oslo_db import concurrency
from oslo_log import log as logging

import dmapi.conf
CONF = dmapi.conf.CONF

_BACKEND_MAPPING = {'sqlalchemy': 'dmapi.db.sqlalchemy.api'}

IMPL = concurrency.TpoolDbapiWrapper(CONF, backend_mapping=_BACKEND_MAPPING)

LOG = logging.getLogger(__name__)
MAX_INT = 0x7FFFFFFF

def instance_get_by_uuid(context, uuid, columns_to_join=None):
    """Get an instance or raise if it does not exist."""

    return IMPL.instance_get_by_uuid(context, uuid, columns_to_join)


def create_context_manager(connection):
    """Return a context manager for a cell database connection."""
    return IMPL.create_context_manager(connection=connection)