Skip to content

Commit

Permalink
add celery task for rb_notes_summary refresh
Browse files Browse the repository at this point in the history
  • Loading branch information
Julien Corny committed Jan 6, 2025
1 parent baca9d2 commit 6b37d13
Showing 1 changed file with 25 additions and 1 deletion.
26 changes: 25 additions & 1 deletion backend/gn_module_zh/tasks.py
Original file line number Diff line number Diff line change
Expand Up @@ -3,7 +3,7 @@
from datetime import datetime, timedelta
from pathlib import Path

from sqlalchemy import func
from sqlalchemy import func, text
from celery.utils.log import get_task_logger
from celery.schedules import crontab

Expand Down Expand Up @@ -33,9 +33,33 @@ def setup_periodic_tasks(sender, **kwargs):
)


@celery_app.on_after_finalize.connect
def setup_periodic_tasks_for_vm_rb_rules(sender, **kwargs):
minute, hour, day_of_month, month_of_year, day_of_week = "* * * * *".split(" ")
sender.add_periodic_task(
crontab(
minute=minute,
hour=hour,
day_of_week=day_of_week,
day_of_month=day_of_month,
month_of_year=month_of_year,
),
refresh_vm_rb_rules.s(),
name="Refresh vm rb rules",
)


@celery_app.task(bind=True)
def refresh_taxon_vm(self):
logger.info("Refresh taxon vms...")
db.session.execute(func.pr_zh.refresh_taxon_materialized_views())
db.session.commit()
logger.info("Taxon vms refreshed.")


@celery_app.task(bind=True)
def refresh_vm_rb_rules(self):
logger.info("Refresh vm rb rules...")
db.session.execute(text("REFRESH MATERIALIZED VIEW pr_zh.rb_notes_summary"))
db.session.commit()
logger.info("vm rb rules refreshed.")

0 comments on commit 6b37d13

Please sign in to comment.