Skip to content

Commit

Permalink
ref(escalating-issues): Add more logs to forecasting cron job (#47716)
Browse files Browse the repository at this point in the history
Add extra logs to the escalating issues cron job to ensure forecasts are
being saved in nodestore
  • Loading branch information
jangjodi committed Apr 20, 2023
1 parent 1ba7d31 commit a669f80
Showing 1 changed file with 11 additions and 0 deletions.
11 changes: 11 additions & 0 deletions src/sentry/issues/forecasts.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
This module is for helper functions for escalating issues forecasts.
"""

import logging
from datetime import datetime
from typing import List

Expand All @@ -14,6 +15,8 @@
from sentry.issues.escalating_issues_alg import generate_issue_forecast
from sentry.models import Group

logger = logging.getLogger(__name__)


def save_forecast_per_group(
until_escalating_groups: List[Group], group_counts: ParsedGroupsCount
Expand All @@ -33,6 +36,10 @@ def save_forecast_per_group(
group_dict[group_id].project.id, group_id, forecasts_list, datetime.now()
)
escalating_group_forecast.save()
logger.info(
"Saved forecasts in nodestore",
extra={"num_groups": len(group_counts.keys())},
)


def get_forecasts(groups: List[Group]) -> None:
Expand All @@ -41,5 +48,9 @@ def get_forecasts(groups: List[Group]) -> None:
`groups`: List of groups to be forecasted
"""
past_counts = query_groups_past_counts(groups)
logger.info(
"Queried groups from snuba",
extra={"num_groups": len(past_counts)},
)
group_counts = parse_groups_past_counts(past_counts)
save_forecast_per_group(groups, group_counts)

0 comments on commit a669f80

Please sign in to comment.