mirror of
https://github.com/mariadb-corporation/mariadb-columnstore-engine.git
synced 2025-04-18 21:44:02 +03:00
49abbb4d7e
This task is a next logical part of MCOL-5470 that implements AppManager and API endpoint to check if CMAPI ready or not. - [add] cmapi_app.py with is-ready command implementation - [fix] add cmapi is-ready command to main typer app
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
"""Cmapi typer application.
|
|
|
|
Formally this module contains all subcommands for "mcs cmapi" cli command.
|
|
"""
|
|
import logging
|
|
from typing_extensions import Annotated
|
|
|
|
import requests
|
|
import typer
|
|
|
|
from cmapi_server.exceptions import CMAPIBasicError
|
|
from mcs_cluster_tool.decorators import handle_output
|
|
|
|
|
|
logger = logging.getLogger('mcs_cli')
|
|
app = typer.Typer(
|
|
help='CMAPI itself related commands.'
|
|
)
|
|
|
|
|
|
@app.command()
|
|
@handle_output
|
|
def is_ready(
|
|
node: Annotated[
|
|
str,
|
|
typer.Option(
|
|
'--node',
|
|
help=('Which node to check the CMAPI is ready to handle requests.')
|
|
)
|
|
] = '127.0.0.1'
|
|
):
|
|
"""Check CMAPI is ready to handle requests."""
|
|
url = f'https://{node}:8640/cmapi/ready'
|
|
try:
|
|
resp = requests.get(url, verify=False, timeout=15)
|
|
resp.raise_for_status()
|
|
r_json = resp.json()
|
|
except requests.exceptions.HTTPError as err:
|
|
if err.response.status_code == 503:
|
|
raise CMAPIBasicError('CMAPI is not ready.') from err
|
|
else:
|
|
raise CMAPIBasicError(
|
|
'Got unexpected HTTP return code '
|
|
f'"{err.response.status_code}" while getting CMAPI ready '
|
|
'state.'
|
|
) from err
|
|
except Exception as err:
|
|
raise CMAPIBasicError('Got an error getting CMAPI ready state.') from err
|
|
logger.debug('Successfully get CMAPI ready state.')
|
|
return r_json
|