From 41413242bdb2b87615b28554d42f1bd6f91778bd Mon Sep 17 00:00:00 2001
From: knst-kotov <knst.kotov@gmail.com>
Date: Wed, 1 Sep 2021 13:28:42 +0300
Subject: [PATCH] cache top shouts

---
 main.py           |  4 +--
 resolvers/zine.py | 88 ++++++++++++++++++++++++++++++++---------------
 2 files changed, 62 insertions(+), 30 deletions(-)

diff --git a/main.py b/main.py
index 3ef0979b..a473aaca 100644
--- a/main.py
+++ b/main.py
@@ -13,7 +13,7 @@ from auth.oauth import oauth_login, oauth_authorize
 from auth.email import email_authorize
 from redis import redis
 from resolvers.base import resolvers
-from resolvers.zine import GitTask
+from resolvers.zine import GitTask, TopShouts
 
 import asyncio
 
@@ -28,7 +28,7 @@ middleware = [
 async def start_up():
 	await redis.connect()
 	git_task = asyncio.create_task(GitTask.git_task_worker())
-
+	top_shouts_task = asyncio.create_task(TopShouts.worker())
 
 async def shutdown():
 	await redis.disconnect()
diff --git a/resolvers/zine.py b/resolvers/zine.py
index e8bd40a4..9f43f481 100644
--- a/resolvers/zine.py
+++ b/resolvers/zine.py
@@ -67,40 +67,72 @@ class GitTask:
 				print("git task worker error = %s" % (err))
 
 
+class TopShouts:
+	limit = 50
+	period = 60*60 #1 hour
+
+	lock = asyncio.Lock()
+
+	@staticmethod
+	async def prepare_shouts_by_rating():
+		month_ago = datetime.now() - timedelta(days = 30)
+		with local_session() as session:
+			stmt = select(Shout, func.sum(ShoutRating.value).label("rating")).\
+				join(ShoutRating).\
+				where(ShoutRating.ts > month_ago).\
+				group_by(Shout.id).\
+				order_by(desc("rating")).\
+				limit(TopShouts.limit)
+			shouts = []
+			for row in session.execute(stmt):
+				shout = row.Shout
+				shout.rating = row.rating
+				shouts.append(shout)
+		async with TopShouts.lock:
+			TopShouts.shouts_by_rating = shouts
+
+	@staticmethod
+	async def prepare_shouts_by_view():
+		month_ago = datetime.now() - timedelta(days = 30)
+		with local_session() as session:
+			stmt = select(Shout, func.sum(ShoutViewByDay.value).label("view")).\
+				join(ShoutViewByDay).\
+				where(ShoutViewByDay.day > month_ago).\
+				group_by(Shout.id).\
+				order_by(desc("view")).\
+				limit(TopShouts.limit)
+			shouts = []
+			for row in session.execute(stmt):
+				shout = row.Shout
+				shout.view = row.view
+				shouts.append(shout)
+		async with TopShouts.lock:
+			TopShouts.shouts_by_view = shouts
+
+	@staticmethod
+	async def worker():
+		print("top shouts worker start")
+		while True:
+			try:
+				print("top shouts: update cache")
+				await TopShouts.prepare_shouts_by_rating()
+				await TopShouts.prepare_shouts_by_view()
+				print("top shouts: update finished")
+			except Exception as err:
+				print("top shouts worker error = %s" % (err))
+			await asyncio.sleep(TopShouts.period)
+
+
 @query.field("topShoutsByView")
 async def top_shouts_by_view(_, info, limit):
-	month_ago = datetime.now() - timedelta(days = 30)
-	with local_session() as session:
-		stmt = select(Shout, func.sum(ShoutViewByDay.value).label("view")).\
-			join(ShoutViewByDay).\
-			where(ShoutViewByDay.day > month_ago).\
-			group_by(Shout.id).\
-			order_by(desc("view")).\
-			limit(limit)
-		shouts = []
-		for row in session.execute(stmt):
-			shout = row.Shout
-			shout.view = row.view
-			shouts.append(shout)
-	return shouts
+	async with TopShouts.lock:
+		return TopShouts.shouts_by_view[:limit]
 
 
 @query.field("topShoutsByRating")
 async def top_shouts(_, info, limit):
-	month_ago = datetime.now() - timedelta(days = 30)
-	with local_session() as session:
-		stmt = select(Shout, func.sum(ShoutRating.value).label("rating")).\
-			join(ShoutRating).\
-			where(ShoutRating.ts > month_ago).\
-			group_by(Shout.id).\
-			order_by(desc("rating")).\
-			limit(limit)
-		shouts = []
-		for row in session.execute(stmt):
-			shout = row.Shout
-			shout.rating = row.rating
-			shouts.append(shout)
-	return shouts
+	async with TopShouts.lock:
+		return TopShouts.shouts_by_rating[:limit]
 
 
 @mutation.field("createShout")