Skip to content

Commit

Permalink
FLEX: Reorganize the Coordinator and gsctl to adapt to the unified sc…
Browse files Browse the repository at this point in the history
…hema
  • Loading branch information
lidongze0629 committed May 16, 2024
1 parent f15605b commit 36f4417
Show file tree
Hide file tree
Showing 257 changed files with 16,700 additions and 27,483 deletions.
83 changes: 44 additions & 39 deletions flex/coordinator/.openapi-generator/FILES
Original file line number Diff line number Diff line change
Expand Up @@ -8,55 +8,60 @@ gs_flex_coordinator/__init__.py
gs_flex_coordinator/__main__.py
gs_flex_coordinator/encoder.py
gs_flex_coordinator/models/__init__.py
gs_flex_coordinator/models/alert_message.py
gs_flex_coordinator/models/alert_receiver.py
gs_flex_coordinator/models/alert_rule.py
gs_flex_coordinator/models/base_edge_type.py
gs_flex_coordinator/models/base_edge_type_vertex_type_pair_relations_inner.py
gs_flex_coordinator/models/base_edge_type_vertex_type_pair_relations_inner_x_csr_params.py
gs_flex_coordinator/models/base_model.py
gs_flex_coordinator/models/base_property_meta.py
gs_flex_coordinator/models/base_vertex_type.py
gs_flex_coordinator/models/base_vertex_type_x_csr_params.py
gs_flex_coordinator/models/column_mapping.py
gs_flex_coordinator/models/connection.py
gs_flex_coordinator/models/connection_status.py
gs_flex_coordinator/models/data_source.py
gs_flex_coordinator/models/deployment_info.py
gs_flex_coordinator/models/deployment_info_graphs_info_value.py
gs_flex_coordinator/models/deployment_status.py
gs_flex_coordinator/models/edge_data_source.py
gs_flex_coordinator/models/column_mapping_column.py
gs_flex_coordinator/models/create_alert_receiver_request.py
gs_flex_coordinator/models/create_alert_rule_request.py
gs_flex_coordinator/models/create_dataloading_job_response.py
gs_flex_coordinator/models/create_edge_type.py
gs_flex_coordinator/models/create_graph_request.py
gs_flex_coordinator/models/create_graph_response.py
gs_flex_coordinator/models/create_graph_schema_request.py
gs_flex_coordinator/models/create_procedure_request.py
gs_flex_coordinator/models/create_procedure_response.py
gs_flex_coordinator/models/create_property_meta.py
gs_flex_coordinator/models/create_vertex_type.py
gs_flex_coordinator/models/dataloading_job_config.py
gs_flex_coordinator/models/dataloading_job_config_edges_inner.py
gs_flex_coordinator/models/dataloading_job_config_loading_config.py
gs_flex_coordinator/models/dataloading_job_config_loading_config_format.py
gs_flex_coordinator/models/dataloading_job_config_vertices_inner.py
gs_flex_coordinator/models/edge_mapping.py
gs_flex_coordinator/models/edge_mapping_destination_vertex_mappings_inner.py
gs_flex_coordinator/models/edge_mapping_source_vertex_mappings_inner.py
gs_flex_coordinator/models/edge_mapping_source_vertex_mappings_inner_column.py
gs_flex_coordinator/models/edge_mapping_type_triplet.py
gs_flex_coordinator/models/edge_type.py
gs_flex_coordinator/models/edge_type_vertex_type_pair_relations_inner.py
gs_flex_coordinator/models/edge_type_vertex_type_pair_relations_inner_x_csr_params.py
gs_flex_coordinator/models/graph.py
gs_flex_coordinator/models/graph_stored_procedures.py
gs_flex_coordinator/models/groot_dataloading_job_config.py
gs_flex_coordinator/models/groot_dataloading_job_config_edges_inner.py
gs_flex_coordinator/models/groot_edge_type.py
gs_flex_coordinator/models/groot_edge_type_relations_inner.py
gs_flex_coordinator/models/groot_graph.py
gs_flex_coordinator/models/groot_graph_gremlin_interface.py
gs_flex_coordinator/models/groot_property.py
gs_flex_coordinator/models/groot_schema.py
gs_flex_coordinator/models/groot_vertex_type.py
gs_flex_coordinator/models/error.py
gs_flex_coordinator/models/get_alert_message_response.py
gs_flex_coordinator/models/get_alert_receiver_response.py
gs_flex_coordinator/models/get_alert_rule_response.py
gs_flex_coordinator/models/get_edge_type.py
gs_flex_coordinator/models/get_graph_response.py
gs_flex_coordinator/models/get_graph_schema_response.py
gs_flex_coordinator/models/get_procedure_response.py
gs_flex_coordinator/models/get_property_meta.py
gs_flex_coordinator/models/get_vertex_type.py
gs_flex_coordinator/models/gs_data_type.py
gs_flex_coordinator/models/job_status.py
gs_flex_coordinator/models/model_property.py
gs_flex_coordinator/models/model_schema.py
gs_flex_coordinator/models/node_status.py
gs_flex_coordinator/models/procedure.py
gs_flex_coordinator/models/procedure_params_inner.py
gs_flex_coordinator/models/property_property_type.py
gs_flex_coordinator/models/long_text.py
gs_flex_coordinator/models/parameter.py
gs_flex_coordinator/models/primitive_type.py
gs_flex_coordinator/models/running_deployment_info.py
gs_flex_coordinator/models/schema_mapping.py
gs_flex_coordinator/models/schema_mapping_loading_config.py
gs_flex_coordinator/models/schema_mapping_loading_config_data_source.py
gs_flex_coordinator/models/schema_mapping_loading_config_format.py
gs_flex_coordinator/models/service_status.py
gs_flex_coordinator/models/service_status_sdk_endpoints.py
gs_flex_coordinator/models/start_service_request.py
gs_flex_coordinator/models/update_alert_messages_request.py
gs_flex_coordinator/models/vertex_data_source.py
gs_flex_coordinator/models/stored_procedure_meta.py
gs_flex_coordinator/models/string_type.py
gs_flex_coordinator/models/string_type_string.py
gs_flex_coordinator/models/update_alert_message_status_request.py
gs_flex_coordinator/models/update_procedure_request.py
gs_flex_coordinator/models/upload_file_response.py
gs_flex_coordinator/models/vertex_mapping.py
gs_flex_coordinator/models/vertex_type.py
gs_flex_coordinator/openapi/openapi.yaml
gs_flex_coordinator/test/__init__.py
gs_flex_coordinator/typing_utils.py
Expand Down
172 changes: 87 additions & 85 deletions flex/coordinator/gs_flex_coordinator/controllers/alert_controller.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,157 +3,159 @@
from typing import Tuple
from typing import Union

from gs_flex_coordinator.core.alert import alert_manager
from gs_flex_coordinator.core import handle_api_exception
from gs_flex_coordinator.models.alert_message import AlertMessage # noqa: E501
from gs_flex_coordinator.models.alert_receiver import AlertReceiver # noqa: E501
from gs_flex_coordinator.models.alert_rule import AlertRule # noqa: E501
from gs_flex_coordinator.models.update_alert_messages_request import UpdateAlertMessagesRequest # noqa: E501
from gs_flex_coordinator.models.create_alert_receiver_request import CreateAlertReceiverRequest # noqa: E501
from gs_flex_coordinator.models.create_alert_rule_request import CreateAlertRuleRequest # noqa: E501
from gs_flex_coordinator.models.error import Error # noqa: E501
from gs_flex_coordinator.models.get_alert_message_response import GetAlertMessageResponse # noqa: E501
from gs_flex_coordinator.models.get_alert_receiver_response import GetAlertReceiverResponse # noqa: E501
from gs_flex_coordinator.models.get_alert_rule_response import GetAlertRuleResponse # noqa: E501
from gs_flex_coordinator.models.update_alert_message_status_request import UpdateAlertMessageStatusRequest # noqa: E501
from gs_flex_coordinator import util


@handle_api_exception()
def delete_alert_rule_by_name(rule_name): # noqa: E501
"""delete_alert_rule
def create_alert_receiver(create_alert_receiver_request): # noqa: E501
"""create_alert_receiver
# noqa: E501
Create a new alert receiver # noqa: E501
:param rule_name:
:type rule_name: str
:param create_alert_receiver_request:
:type create_alert_receiver_request: dict | bytes
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
return alert_manager.delete_alert_rule_by_name(rule_name)
if connexion.request.is_json:
create_alert_receiver_request = CreateAlertReceiverRequest.from_dict(connexion.request.get_json()) # noqa: E501
return 'do some magic!'


@handle_api_exception()
def delete_receiver_by_id(receiver_id): # noqa: E501
"""delete_receiver_by_id
def delete_alert_message_in_batch(message_ids): # noqa: E501
"""delete_alert_message_in_batch
# noqa: E501
Delete alert message in batch # noqa: E501
:param message_ids: A list of message id separated by comma, e.g. id1,id2,id3
:type message_ids: str
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
return 'do some magic!'


def delete_alert_receiver_by_id(receiver_id): # noqa: E501
"""delete_alert_receiver_by_id
Delete the alert receiver by ID # noqa: E501
:param receiver_id:
:type receiver_id: str
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
return alert_manager.delete_receiver_by_id(receiver_id)
return 'do some magic!'


@handle_api_exception()
def list_alert_messages(alert_type=None, status=None, severity=None, start_time=None, end_time=None): # noqa: E501
"""list_alert_messages
def delete_alert_rule_by_id(rule_id): # noqa: E501
"""delete_alert_rule_by_id
# noqa: E501
:param alert_type:
:param rule_id:
:type rule_id: str
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
return 'do some magic!'


def list_alert_messages(alert_type=None, status=None, severity=None, start_time=None, end_time=None, limit=None): # noqa: E501
"""list_alert_messages
List all alert messages # noqa: E501
:param alert_type:
:type alert_type: str
:param status:
:param status:
:type status: str
:param severity:
:param severity:
:type severity: str
:param start_time: format with \"2023-02-21-11-56-30\"
:type start_time: str
:param end_time: format with \"2023-02-21-11-56-30\"
:type end_time: str
:param limit:
:type limit: int
:rtype: Union[List[AlertMessage], Tuple[List[AlertMessage], int], Tuple[List[AlertMessage], int, Dict[str, str]]
"""
return alert_manager.list_alert_messages(alert_type, status, severity, start_time, end_time)


@handle_api_exception()
def list_alert_rules(): # noqa: E501
"""list_alert_rules
# noqa: E501
:rtype: Union[List[AlertRule], Tuple[List[AlertRule], int], Tuple[List[AlertRule], int, Dict[str, str]]
:rtype: Union[List[GetAlertMessageResponse], Tuple[List[GetAlertMessageResponse], int], Tuple[List[GetAlertMessageResponse], int, Dict[str, str]]
"""
return alert_manager.list_alert_rules()
return 'do some magic!'


@handle_api_exception()
def list_receivers(): # noqa: E501
"""list_receivers
def list_alert_receivers(): # noqa: E501
"""list_alert_receivers
# noqa: E501
List all alert receivers # noqa: E501
:rtype: Union[List[AlertReceiver], Tuple[List[AlertReceiver], int], Tuple[List[AlertReceiver], int, Dict[str, str]]
:rtype: Union[List[GetAlertReceiverResponse], Tuple[List[GetAlertReceiverResponse], int], Tuple[List[GetAlertReceiverResponse], int, Dict[str, str]]
"""
return alert_manager.list_receivers()
return 'do some magic!'


@handle_api_exception()
def register_receiver(alert_receiver): # noqa: E501
"""register_receiver
def list_alert_rules(): # noqa: E501
"""list_alert_rules
# noqa: E501
List all alert rules # noqa: E501
:param alert_receiver:
:type alert_receiver: dict | bytes
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
:rtype: Union[List[GetAlertRuleResponse], Tuple[List[GetAlertRuleResponse], int], Tuple[List[GetAlertRuleResponse], int, Dict[str, str]]
"""
if connexion.request.is_json:
alert_receiver = AlertReceiver.from_dict(connexion.request.get_json()) # noqa: E501
return alert_manager.register_receiver(alert_receiver)
return 'do some magic!'


@handle_api_exception()
def update_alert_messages(update_alert_messages_request=None): # noqa: E501
"""update_alert_messages
def update_alert_message_in_batch(update_alert_message_status_request=None): # noqa: E501
"""update_alert_message_in_batch
Update alert messages in batch # noqa: E501
Update the message status in batch # noqa: E501
:param update_alert_messages_request:
:type update_alert_messages_request: dict | bytes
:param update_alert_message_status_request:
:type update_alert_message_status_request: dict | bytes
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
if connexion.request.is_json:
update_alert_messages_request = UpdateAlertMessagesRequest.from_dict(connexion.request.get_json()) # noqa: E501
return alert_manager.update_alert_messages(
update_alert_messages_request.messages,
update_alert_messages_request.batch_status,
update_alert_messages_request.batch_delete
)
update_alert_message_status_request = UpdateAlertMessageStatusRequest.from_dict(connexion.request.get_json()) # noqa: E501
return 'do some magic!'


@handle_api_exception()
def update_alert_rule_by_name(rule_name, alert_rule=None): # noqa: E501
"""update_alert_rule_by_name
def update_alert_receiver_by_id(receiver_id, create_alert_receiver_request=None): # noqa: E501
"""update_alert_receiver_by_id
# noqa: E501
Update alert receiver by ID # noqa: E501
:param rule_name:
:type rule_name: str
:param alert_rule:
:type alert_rule: dict | bytes
:param receiver_id:
:type receiver_id: str
:param create_alert_receiver_request:
:type create_alert_receiver_request: dict | bytes
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
if connexion.request.is_json:
alert_rule = AlertRule.from_dict(connexion.request.get_json()) # noqa: E501
return alert_manager.update_alert_rule_by_name(rule_name, alert_rule)
create_alert_receiver_request = CreateAlertReceiverRequest.from_dict(connexion.request.get_json()) # noqa: E501
return 'do some magic!'


@handle_api_exception()
def update_receiver_by_id(receiver_id, alert_receiver=None): # noqa: E501
"""update_receiver_by_id
def update_alert_rule_by_id(rule_id, create_alert_rule_request=None): # noqa: E501
"""update_alert_rule_by_id
# noqa: E501
:param receiver_id:
:type receiver_id: str
:param alert_receiver:
:type alert_receiver: dict | bytes
:param rule_id:
:type rule_id: str
:param create_alert_rule_request:
:type create_alert_rule_request: dict | bytes
:rtype: Union[str, Tuple[str, int], Tuple[str, int, Dict[str, str]]
"""
if connexion.request.is_json:
alert_receiver = AlertReceiver.from_dict(connexion.request.get_json()) # noqa: E501
return alert_manager.update_receiver_by_id(receiver_id, alert_receiver)
create_alert_rule_request = CreateAlertRuleRequest.from_dict(connexion.request.get_json()) # noqa: E501
return 'do some magic!'

This file was deleted.

Loading

0 comments on commit 36f4417

Please sign in to comment.