5227ee3798
# What this PR does Related to https://github.com/grafana/oncall/pull/5287 Few random "clean-ups", type improvements, etc. Additionally, fixes a change made in #5292; we should wait to read from `slack_message.channel.slack_id`, until we've performed the data-migration mentioned in that PR (in the mean-time we should continue to use `slack_message._channel_id`). ## 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] Added the relevant release notes label (see labels prefixed w/ `release:`). These labels dictate how your PR will show up in the autogenerated release notes.
57 lines
2.3 KiB
Python
57 lines
2.3 KiB
Python
from functools import partial
|
|
|
|
from django.conf import settings
|
|
from django.db import transaction
|
|
|
|
from common.custom_celery_tasks import shared_dedicated_queue_retry_task
|
|
|
|
from .send_alert_group_signal import send_alert_group_signal
|
|
from .task_logger import task_logger
|
|
|
|
|
|
@shared_dedicated_queue_retry_task(
|
|
autoretry_for=(Exception,), retry_backoff=True, max_retries=1 if settings.DEBUG else None
|
|
)
|
|
def unsilence_task(alert_group_pk):
|
|
from apps.alerts.models import AlertGroup, AlertGroupLogRecord
|
|
|
|
task_logger.info(f"Start unsilence_task for alert_group {alert_group_pk}")
|
|
|
|
with transaction.atomic():
|
|
try:
|
|
alert_group = AlertGroup.objects.filter(pk=alert_group_pk).select_for_update()[0] # Lock alert_group:
|
|
except IndexError:
|
|
task_logger.info(f"unsilence_task. alert_group {alert_group_pk} doesn't exist")
|
|
return
|
|
|
|
if unsilence_task.request.id != alert_group.unsilence_task_uuid:
|
|
task_logger.info(
|
|
f"unsilence_task. alert_group {alert_group.pk}.ID mismatch.Active: {alert_group.unsilence_task_uuid}"
|
|
)
|
|
return
|
|
|
|
if alert_group.status == AlertGroup.SILENCED and alert_group.is_root_alert_group:
|
|
initial_state = alert_group.state
|
|
task_logger.info(f"unsilence alert_group {alert_group_pk} and start escalation if needed")
|
|
|
|
alert_group.un_silence()
|
|
# update metrics
|
|
alert_group._update_metrics(
|
|
organization_id=alert_group.channel.organization_id,
|
|
previous_state=initial_state,
|
|
state=alert_group.state,
|
|
)
|
|
alert_group.start_escalation_if_needed()
|
|
un_silence_log_record = AlertGroupLogRecord(
|
|
type=AlertGroupLogRecord.TYPE_UN_SILENCE,
|
|
alert_group=alert_group,
|
|
reason="auto unsilence",
|
|
)
|
|
un_silence_log_record.save()
|
|
transaction.on_commit(partial(send_alert_group_signal.apply_async, (un_silence_log_record.pk,)))
|
|
else:
|
|
task_logger.info(
|
|
f"Failed to unsilence alert_group {alert_group_pk}: alert_group status: {alert_group.status}, "
|
|
f"is root: {alert_group.is_root_alert_group}"
|
|
)
|
|
task_logger.info(f"Finish unsilence_task for alert_group {alert_group_pk}")
|