This commit is contained in:
tonyrewin 2022-09-07 19:56:26 +03:00
parent bae0be1c7d
commit 79e112c36f
3 changed files with 19 additions and 20 deletions

View File

@ -16,39 +16,39 @@ from sqlalchemy.orm import selectinload
@query.field("topViewed") @query.field("topViewed")
async def top_viewed(_, info, page, size): async def top_viewed(_, _info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_top_viewed[((page - 1) * size) : (page * size)] return ShoutsCache.get_top_viewed()[((page - 1) * size) : (page * size)]
@query.field("topMonth") @query.field("topMonth")
async def top_month(_, info, page, size): async def top_month(_, _info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_top_month[((page - 1) * size) : (page * size)] return ShoutsCache.get_top_month()[((page - 1) * size) : (page * size)]
@query.field("topOverall") @query.field("topOverall")
async def top_overall(_, info, page, size): async def top_overall(_, _info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_top_overall[((page - 1) * size) : (page * size)] return ShoutsCache.get_top_overall()[((page - 1) * size) : (page * size)]
@query.field("recentPublished") @query.field("recentPublished")
async def recent_published(_, info, page, size): async def recent_published(_, _info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_recent_published[((page - 1) * size) : (page * size)] return ShoutsCache.get_recent_published()[((page - 1) * size) : (page * size)]
@query.field("recentAll") @query.field("recentAll")
async def recent_all(_, info, page, size): async def recent_all(_, _info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_recent_all[((page - 1) * size) : (page * size)] return ShoutsCache.get_recent_all()[((page - 1) * size) : (page * size)]
@query.field("recentReacted") @query.field("recentReacted")
async def recent_reacted(_, info, page, size): async def recent_reacted(_, info, page, size):
async with ShoutsCache.lock: async with ShoutsCache.lock:
return ShoutsCache.get_recent_reacted[((page - 1) * size) : (page * size)] return ShoutsCache.get_recent_reacted()[((page - 1) * size) : (page * size)]
@mutation.field("viewShout") @mutation.field("viewShout")

View File

@ -216,11 +216,11 @@ class ReactedStorage:
@staticmethod @staticmethod
async def flush_changes(session): async def flush_changes(session):
self = ReactedStorage() self = ReactedStorage
async with self.lock: async with self.lock:
for slug in dict(self.reacted['shouts']).keys(): for slug in dict(self.reacted['shouts']).keys():
topics = session.query(ShoutTopic.topic).where(ShoutTopic.shout == slug).all() topics = session.query(ShoutTopic.topic).where(ShoutTopic.shout == slug).all()
reactions = self.reacted['shouts'][slug] reactions = self.reacted['shouts'].get(slug, [])
for ts in list(topics): for ts in list(topics):
try: try:
tslug = ts.pop() tslug = ts.pop()
@ -238,6 +238,7 @@ class ReactedStorage:
session.add(reaction) session.add(reaction)
flag_modified(reaction, "value") flag_modified(reaction, "value")
reaction.modified = False reaction.modified = False
print('flushing')
for reaction in self.to_flush: for reaction in self.to_flush:
session.add(reaction) session.add(reaction)
self.to_flush.clear() self.to_flush.clear()

View File

@ -22,18 +22,16 @@ class TopicStat:
for shout_topic in shout_topics: for shout_topic in shout_topics:
topic = shout_topic.topic topic = shout_topic.topic
shout = shout_topic.shout shout = shout_topic.shout
if topic in self.shouts_by_topic: if not self.shouts_by_topic.get(topic):
self.shouts_by_topic[topic] = []
if shout not in self.shouts_by_topic[topic]:
self.shouts_by_topic[topic].append(shout) self.shouts_by_topic[topic].append(shout)
else:
self.shouts_by_topic[topic] = [
shout,
]
authors = await ShoutAuthorStorage.get_authors(shout) authors = await ShoutAuthorStorage.get_authors(shout)
if topic in self.authors_by_topic: if topic in self.authors_by_topic:
self.authors_by_topic[topic].update(authors) self.authors_by_topic[topic].update(authors)
else: else:
self.authors_by_topic[topic] = set(authors) self.authors_by_topic[topic] = list(set(authors))
print("[stat.topics] authors sorted") print("[stat.topics] authors sorted")
print("[stat.topics] shouts sorted") print("[stat.topics] shouts sorted")
@ -46,7 +44,7 @@ class TopicStat:
if topic in self.followers_by_topic: if topic in self.followers_by_topic:
self.followers_by_topic[topic].append(user) self.followers_by_topic[topic].append(user)
else: else:
self.followers_by_topic[topic] = [user] self.followers_by_topic[topic] = [user, ]
print("[stat.topics] followers sorted") print("[stat.topics] followers sorted")
@staticmethod @staticmethod