a gazillion things to implement cosmos db healthcheck
This commit is contained in:
parent
7af3b12c81
commit
91c79357f8
11 changed files with 421 additions and 15 deletions
|
|
@ -1,10 +1,32 @@
|
|||
import logging
|
||||
from anaxi.config import CosmosDBDatabaseConfig
|
||||
from anaxi.constants import PATHS
|
||||
from anaxi.cosmos_tools import create_cosmos_client_from_config
|
||||
from anaxi.logging import get_anaxi_logger
|
||||
|
||||
logger = logging.getLogger()
|
||||
logger = get_anaxi_logger(__name__)
|
||||
|
||||
|
||||
def run_cosmos_db_healthcheck_process(cosmos_db_id: str) -> None:
|
||||
|
||||
logger.info(
|
||||
f"If I had a cosmos db connector, I would really try to hit this database: {cosmos_db_id}"
|
||||
logger.info("Reading Cosmos DB config file...")
|
||||
cosmos_db_configs = CosmosDBDatabaseConfig.from_yaml(
|
||||
PATHS.cosmos_db_config_file_path
|
||||
)
|
||||
logger.info(f"Found file with {len(cosmos_db_configs)} entries.")
|
||||
try:
|
||||
relevant_cosmos_db_config = cosmos_db_configs[cosmos_db_id]
|
||||
except KeyError as e:
|
||||
logger.error(
|
||||
f"Couldn't find a config entry for database with id: {cosmos_db_id}"
|
||||
)
|
||||
raise e
|
||||
|
||||
logger.info("Creating client...")
|
||||
cosmos_client = create_cosmos_client_from_config(relevant_cosmos_db_config)
|
||||
logger.info("Client created.")
|
||||
|
||||
logger.info("Throwing a SELECT 1;")
|
||||
response = cosmos_client.query_databases(query="SELECT 1")
|
||||
logger.info(f"Response: {response.next()}")
|
||||
|
||||
return
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue