[superset] branch master updated: chore(reports): add metrics to report schedule and log prune (#25080)

[vi...@apache.org]

This is an automated email from the ASF dual-hosted git repository.

villebro pushed a commit to branch master
in repository https://gitbox.apache.org/repos/asf/superset.git


The following commit(s) were added to refs/heads/master by this push:
     new 04bb73cf27 chore(reports): add metrics to report schedule and log prune (#25080)
04bb73cf27 is described below

commit 04bb73cf279727f9d5852cfbd0917f08eea452b9
Author: Ville Brofeldt <33...@users.noreply.github.com>
AuthorDate: Fri Aug 25 09:39:26 2023 -0700

    chore(reports): add metrics to report schedule and log prune (#25080)
---
 superset/tasks/scheduler.py | 10 ++++++++++
 1 file changed, 10 insertions(+)

diff --git a/superset/tasks/scheduler.py b/superset/tasks/scheduler.py
index 0133ee34ca..90df90ff15 100644
--- a/superset/tasks/scheduler.py
+++ b/superset/tasks/scheduler.py
@@ -26,6 +26,7 @@ from superset.extensions import celery_app
 from superset.reports.commands.exceptions import ReportScheduleUnexpectedError
 from superset.reports.commands.execute import AsyncExecuteReportScheduleCommand
 from superset.reports.commands.log_prune import AsyncPruneReportScheduleLogCommand
+from superset.stats_logger import BaseStatsLogger
 from superset.tasks.cron_util import cron_schedule_window
 from superset.utils.celery import session_scope
 from superset.utils.core import LoggerLevel
@@ -39,6 +40,9 @@ def scheduler() -> None:
     """
     Celery beat main scheduler for reports
     """
+    stats_logger: BaseStatsLogger = app.config["STATS_LOGGER"]
+    stats_logger.incr("reports.scheduler")
+
     if not is_feature_enabled("ALERT_REPORTS"):
         return
     with session_scope(nullpool=True) as session:
@@ -68,6 +72,9 @@ def scheduler() -> None:
 
 @celery_app.task(name="reports.execute", bind=True)
 def execute(self: Celery.task, report_schedule_id: int) -> None:
+    stats_logger: BaseStatsLogger = app.config["STATS_LOGGER"]
+    stats_logger.incr("reports.execute")
+
     task_id = None
     try:
         task_id = execute.request.id
@@ -100,6 +107,9 @@ def execute(self: Celery.task, report_schedule_id: int) -> None:
 
 @celery_app.task(name="reports.prune_log")
 def prune_log() -> None:
+    stats_logger: BaseStatsLogger = app.config["STATS_LOGGER"]
+    stats_logger.incr("reports.prune_log")
+
     try:
         AsyncPruneReportScheduleLogCommand().run()
     except SoftTimeLimitExceeded as ex: