130 lines
6.5 KiB
Python
130 lines
6.5 KiB
Python
import asyncio
|
|
from sqlalchemy import event
|
|
from orm.author import Author, AuthorFollower
|
|
from orm.reaction import Reaction
|
|
from orm.shout import Shout, ShoutAuthor
|
|
from orm.topic import Topic, TopicFollower
|
|
from services.cache import cache_author, get_cached_author, cache_topic, get_cached_topic
|
|
from services.logger import root_logger as logger
|
|
|
|
|
|
class CacheRevalidationManager:
|
|
"""Управление периодической ревалидацией кэша."""
|
|
|
|
def __init__(self):
|
|
self.items_to_revalidate = {"authors": set(), "topics": set()}
|
|
self.revalidation_interval = 60 # Интервал ревалидации в секундах
|
|
|
|
def start(self):
|
|
loop = asyncio.get_event_loop()
|
|
loop.run_until_complete(self.revalidate_cache())
|
|
loop.run_forever()
|
|
|
|
async def revalidate_cache(self):
|
|
"""Периодическая ревалидация кэша."""
|
|
while True:
|
|
await asyncio.sleep(self.revalidation_interval)
|
|
await self.process_revalidation()
|
|
|
|
async def process_revalidation(self):
|
|
"""Ревалидация кэша для отмеченных сущностей."""
|
|
for entity_type, ids in self.items_to_revalidate.items():
|
|
for entity_id in ids:
|
|
if entity_type == "authors":
|
|
# Ревалидация кэша автора
|
|
author = await get_cached_author(entity_id)
|
|
if author:
|
|
await cache_author(author)
|
|
elif entity_type == "topics":
|
|
# Ревалидация кэша темы
|
|
topic = await get_cached_topic(entity_id)
|
|
if topic:
|
|
await cache_topic(topic)
|
|
ids.clear()
|
|
|
|
def mark_for_revalidation(self, entity_id, entity_type):
|
|
"""Отметить сущность для ревалидации."""
|
|
self.items_to_revalidate[entity_type].add(entity_id)
|
|
|
|
# Инициализация и запуск менеджера ревалидации
|
|
revalidation_manager = CacheRevalidationManager()
|
|
revalidation_manager.start()
|
|
|
|
|
|
def after_update_handler(mapper, connection, target):
|
|
"""Обработчик обновления сущности."""
|
|
entity_type = "authors" if isinstance(target, Author) else "topics" if isinstance(target, Topic) else "shouts"
|
|
revalidation_manager.mark_for_revalidation(target.id, entity_type)
|
|
|
|
|
|
def after_follower_insert_update_handler(mapper, connection, target):
|
|
"""Обработчик добавления или обновления подписки."""
|
|
if isinstance(target, AuthorFollower):
|
|
# Пометить автора и подписчика для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.author_id, "authors")
|
|
revalidation_manager.mark_for_revalidation(target.follower_id, "authors")
|
|
elif isinstance(target, TopicFollower):
|
|
# Пометить тему и подписчика для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.topic_id, "topics")
|
|
revalidation_manager.mark_for_revalidation(target.follower_id, "authors")
|
|
|
|
|
|
def after_follower_delete_handler(mapper, connection, target):
|
|
"""Обработчик удаления подписки."""
|
|
if isinstance(target, AuthorFollower):
|
|
# Пометить автора и подписчика для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.author_id, "authors")
|
|
revalidation_manager.mark_for_revalidation(target.follower_id, "authors")
|
|
elif isinstance(target, TopicFollower):
|
|
# Пометить тему и подписчика для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.topic_id, "topics")
|
|
revalidation_manager.mark_for_revalidation(target.follower_id, "authors")
|
|
|
|
|
|
def after_reaction_update_handler(mapper, connection, reaction):
|
|
"""Обработчик изменений реакций."""
|
|
# Пометить shout для ревалидации
|
|
revalidation_manager.mark_for_revalidation(reaction.shout_id, "shouts")
|
|
# Пометить автора реакции для ревалидации
|
|
revalidation_manager.mark_for_revalidation(reaction.created_by, "authors")
|
|
|
|
|
|
def after_shout_author_insert_update_handler(mapper, connection, target):
|
|
"""Обработчик добавления или обновления авторства публикации."""
|
|
# Пометить shout и автора для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.shout_id, "shouts")
|
|
revalidation_manager.mark_for_revalidation(target.author_id, "authors")
|
|
|
|
|
|
def after_shout_author_delete_handler(mapper, connection, target):
|
|
"""Обработчик удаления авторства публикации."""
|
|
# Пометить shout и автора для ревалидации
|
|
revalidation_manager.mark_for_revalidation(target.shout_id, "shouts")
|
|
revalidation_manager.mark_for_revalidation(target.author_id, "authors")
|
|
|
|
|
|
def events_register():
|
|
"""Регистрация обработчиков событий для всех сущностей."""
|
|
event.listen(ShoutAuthor, "after_insert", after_shout_author_insert_update_handler)
|
|
event.listen(ShoutAuthor, "after_update", after_shout_author_insert_update_handler)
|
|
event.listen(ShoutAuthor, "after_delete", after_shout_author_delete_handler)
|
|
|
|
event.listen(AuthorFollower, "after_insert", after_follower_insert_update_handler)
|
|
event.listen(AuthorFollower, "after_update", after_follower_insert_update_handler)
|
|
event.listen(AuthorFollower, "after_delete", after_follower_delete_handler)
|
|
event.listen(TopicFollower, "after_insert", after_follower_insert_update_handler)
|
|
event.listen(TopicFollower, "after_update", after_follower_insert_update_handler)
|
|
event.listen(TopicFollower, "after_delete", after_follower_delete_handler)
|
|
event.listen(Reaction, "after_update", after_reaction_update_handler)
|
|
|
|
event.listen(Author, "after_update", after_update_handler)
|
|
event.listen(Topic, "after_update", after_update_handler)
|
|
event.listen(Shout, "after_update", after_update_handler)
|
|
event.listen(
|
|
Reaction,
|
|
"after_update",
|
|
lambda mapper, connection, target: revalidation_manager.mark_for_revalidation(target.shout, "shouts"),
|
|
)
|
|
|
|
logger.info("Event handlers registered successfully.")
|