postfixing-reimplemented-cache

resolvers/author.py

@@ -73,7 +73,7 @@ async def get_author(_, _info, slug="", author_id=0):
             if found_author:
                 logger.debug(f"found author id: {found_author.id}")
                 author_id = found_author.id if found_author.id else author_id
-                author_dict = await get_cached_author(author_id)
+                author_dict = await get_cached_author(int(author_id), get_with_stat)
 
         # update stat from db
         if not author_dict or not author_dict.get("stat"):
@@ -99,7 +99,7 @@ async def get_author_by_user_id(user_id: str):
     logger.info(f"getting author id for {user_id}")
     author = None
     try:
-        author = await get_cached_author_by_user_id(user_id)
+        author = await get_cached_author_by_user_id(user_id, get_with_stat)
         if author:
             return author
 
@@ -150,7 +150,7 @@ async def load_authors_by(_, _info, by, limit, offset):
         for [a] in authors_nostat:
             author_dict = None
             if isinstance(a, Author):
-                author_dict = await get_cached_author(a.id)
+                author_dict = await get_cached_author(a.id, get_with_stat)
                 if not author_dict or not isinstance(author_dict.get("shouts"), int):
                     break
 

resolvers/follower.py

@@ -135,7 +135,7 @@ async def unfollow(_, info, what, slug):
 async def get_follows_by_user_id(user_id: str):
     if not user_id:
         return {"error": "unauthorized"}
-    author = await get_cached_author_by_user_id(user_id)
+    author = await get_cached_author_by_user_id(user_id, get_with_stat)
     if not author:
         with local_session() as session:
             author = session.query(Author).filter(Author.user == user_id).first()

resolvers/stat.py

@@ -56,9 +56,9 @@ def get_topic_shouts_stat(topic_id: int):
     return result[0] if result else 0
 
 
-def get_topic_authors_query(topic_id):
-    return (
-        select(ShoutAuthor.author)
+def get_topic_authors_stat(topic_id: int):
+    count_query = (
+        select(func.count(distinct(ShoutAuthor.author)))
         .select_from(join(ShoutTopic, Shout, ShoutTopic.shout == Shout.id))
         .join(ShoutAuthor, ShoutAuthor.shout == Shout.id)
         .filter(
@@ -70,14 +70,6 @@ def get_topic_authors_query(topic_id):
         )
     )
 
-
-def get_topic_authors_stat(topic_id: int):
-    # authors query
-    topic_authors_query = get_topic_authors_query(topic_id)
-
-    # Оборачиваем запрос в другой запрос, чтобы посчитать уникальных авторов
-    count_query = select(func.count(distinct(topic_authors_query.subquery().c.author)))
-
     # Выполняем запрос и получаем результат
     result = local_session().execute(count_query).scalar()
     return result if result is not None else 0

resolvers/topic.py

@@ -1,7 +1,7 @@
-from sqlalchemy import distinct, func, select
+from sqlalchemy import and_, distinct, func, join, select
 
 from orm.author import Author
-from orm.shout import ShoutTopic
+from orm.shout import Shout, ShoutAuthor, ShoutTopic
 from orm.topic import Topic
 from resolvers.stat import get_with_stat
 from services.auth import login_required
@@ -144,5 +144,17 @@ async def get_topic_authors(_, _info, slug: str):
     topic_query = select(Topic.id).filter(Topic.slug == slug).first()
     topic_id_result = local_session().execute(topic_query)
     topic_id = topic_id_result[0] if topic_id_result else None
-    authors = await get_cached_topic_authors(topic_id)
+    topic_authors_query = (
+        select(ShoutAuthor.author)
+        .select_from(join(ShoutTopic, Shout, ShoutTopic.shout == Shout.id))
+        .join(ShoutAuthor, ShoutAuthor.shout == Shout.id)
+        .filter(
+            and_(
+                ShoutTopic.topic == topic_id,
+                Shout.published_at.is_not(None),
+                Shout.deleted_at.is_(None),
+            )
+        )
+    )
+    authors = await get_cached_topic_authors(topic_id, topic_authors_query)
     return authors