d6140cbe8d
# What this PR does Related to https://github.com/grafana/oncall/issues/2392 - Re-enable the following `mypy` rules + fix their pre-existing errors: - `no-redef` - `valid-type` - `var-annotated` - Add stronger return typing to the `GrafanaAPIClient` by use of generics + add some links to documentation in the method docstrings ## Checklist - [x] Unit, integration, and e2e (if applicable) tests updated - [x] Documentation added (or `pr:no public docs` PR label added if not required) - [x] `CHANGELOG.md` updated (or `pr:no changelog` PR label added if not required)
95 lines
3.9 KiB
Python
95 lines
3.9 KiB
Python
import typing
|
|
|
|
from apps.alerts.constants import AlertGroupState
|
|
from apps.metrics_exporter.helpers import (
|
|
get_response_time_period,
|
|
metrics_update_alert_groups_response_time_cache,
|
|
metrics_update_alert_groups_state_cache,
|
|
)
|
|
|
|
|
|
class MetricsCacheManager:
|
|
class _TeamsDiff(typing.TypedDict):
|
|
team_name: str | None
|
|
deleted: bool
|
|
|
|
TeamsDiffMap = typing.Dict[int, _TeamsDiff]
|
|
|
|
@staticmethod
|
|
def get_default_teams_diff_dict() -> _TeamsDiff:
|
|
return {
|
|
"team_name": None,
|
|
"deleted": False,
|
|
}
|
|
|
|
@staticmethod
|
|
def update_team_diff(
|
|
teams_diff: TeamsDiffMap, team_id: int, new_name: str | None = None, deleted: bool = False
|
|
) -> TeamsDiffMap:
|
|
teams_diff.setdefault(team_id, MetricsCacheManager.get_default_teams_diff_dict())
|
|
teams_diff[team_id]["team_name"] = new_name
|
|
teams_diff[team_id]["deleted"] = deleted
|
|
return teams_diff
|
|
|
|
@staticmethod
|
|
def get_default_states_diff_dict():
|
|
default_dict = {
|
|
"previous_states": {state.value: 0 for state in AlertGroupState},
|
|
"new_states": {state.value: 0 for state in AlertGroupState},
|
|
}
|
|
return default_dict
|
|
|
|
@staticmethod
|
|
def update_integration_states_diff(metrics_dict, integration_id, previous_state=None, new_state=None):
|
|
metrics_dict.setdefault(integration_id, MetricsCacheManager.get_default_states_diff_dict())
|
|
if previous_state:
|
|
state_value = previous_state.value
|
|
metrics_dict[integration_id]["previous_states"][state_value] += 1
|
|
if new_state:
|
|
state_value = new_state.value
|
|
metrics_dict[integration_id]["new_states"][state_value] += 1
|
|
return metrics_dict
|
|
|
|
@staticmethod
|
|
def update_integration_response_time_diff(metrics_dict, integration_id, response_time_seconds):
|
|
metrics_dict.setdefault(integration_id, [])
|
|
metrics_dict[integration_id].append(response_time_seconds)
|
|
return metrics_dict
|
|
|
|
@staticmethod
|
|
def metrics_update_state_cache_for_alert_group(integration_id, organization_id, old_state=None, new_state=None):
|
|
"""
|
|
Update state metric cache for one alert group.
|
|
Run the task to update async if organization_id is None due to an additional request to db
|
|
"""
|
|
metrics_state_diff = MetricsCacheManager.update_integration_states_diff(
|
|
{}, integration_id, previous_state=old_state, new_state=new_state
|
|
)
|
|
metrics_update_alert_groups_state_cache(metrics_state_diff, organization_id)
|
|
|
|
@staticmethod
|
|
def metrics_update_response_time_cache_for_alert_group(integration_id, organization_id, response_time_seconds):
|
|
"""
|
|
Update response time metric cache for one alert group.
|
|
Run the task to update async if organization_id is None due to an additional request to db
|
|
"""
|
|
metrics_response_time = MetricsCacheManager.update_integration_response_time_diff(
|
|
{}, integration_id, response_time_seconds
|
|
)
|
|
metrics_update_alert_groups_response_time_cache(metrics_response_time, organization_id)
|
|
|
|
@staticmethod
|
|
def metrics_update_cache_for_alert_group(
|
|
integration_id, organization_id, old_state=None, new_state=None, response_time=None, started_at=None
|
|
):
|
|
"""Call methods to update state and response time metrics cache for one alert group."""
|
|
|
|
if response_time and old_state == AlertGroupState.FIRING and started_at > get_response_time_period():
|
|
response_time_seconds = int(response_time.total_seconds())
|
|
MetricsCacheManager.metrics_update_response_time_cache_for_alert_group(
|
|
integration_id, organization_id, response_time_seconds
|
|
)
|
|
if old_state or new_state:
|
|
MetricsCacheManager.metrics_update_state_cache_for_alert_group(
|
|
integration_id, organization_id, old_state, new_state
|
|
)
|