discours.io/core
5
0
Fork 0
Code Pull Requests Actions Projects Activity

custom-encoder-fix-3

Browse Source
This commit is contained in:
Untone 2024-03-06 22:05:17 +03:00
parent 7a5cbf7438
commit 5c7b28de90
2 changed files with 15 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
resolvers/author.py
View File

@ -162,14 +162,14 @@ async def get_author_follows(_, _info, slug='', user=None, author_id=None):
) )
if not cached: if not cached:
prepared = [author.dict() for author in authors] prepared = [author.dict() for author in authors]
await redis.execute('SET', rkey, 24 * 60 * 60, json.dumps(prepared, cls=CustomJSONEncoder)) await redis.execute('SET', rkey, json.dumps(prepared, cls=CustomJSONEncoder))
rkey = f'author:{author_id}:follows-topics' rkey = f'author:{author_id}:follows-topics'
cached = await redis.execute('GET', rkey) cached = await redis.execute('GET', rkey)
topics = json.loads(cached) if cached else author_follows_topics(author_id) topics = json.loads(cached) if cached else author_follows_topics(author_id)
if not cached: if not cached:
prepared = [topic.dict() for topic in topics] prepared = [topic.dict() for topic in topics]
await redis.execute('SET', rkey, 24 * 60 * 60, json.dumps(prepared, cls=CustomJSONEncoder)) await redis.execute('SET', rkey, json.dumps(prepared, cls=CustomJSONEncoder))
return { return {
'topics': topics, 'topics': topics,
'authors': authors, 'authors': authors,
@ -198,7 +198,7 @@ async def get_author_follows_topics(_, _info, slug='', user=None, author_id=None
topics = json.loads(cached) if cached else author_follows_topics(author_id) topics = json.loads(cached) if cached else author_follows_topics(author_id)
if not cached: if not cached:
prepared = [topic.dict() for topic in topics] prepared = [topic.dict() for topic in topics]
await redis.execute('SET', rkey, 24 * 60 * 60, json.dumps(prepared, cls=CustomJSONEncoder)) await redis.execute('SET', rkey, json.dumps(prepared, cls=CustomJSONEncoder))
return topics return topics
else: else:
raise ValueError('Author not found') raise ValueError('Author not found')
@ -223,7 +223,7 @@ async def get_author_follows_authors(_, _info, slug='', user=None, author_id=Non
) )
if not cached: if not cached:
prepared = [author.dict() for author in authors] prepared = [author.dict() for author in authors]
await redis.execute('SET', rkey, 24 * 60 * 60, json.dumps(prepared, cls=CustomJSONEncoder)) await redis.execute('SET', rkey, json.dumps(prepared, cls=CustomJSONEncoder))
return authors return authors
else: else:
raise ValueError('Author not found') raise ValueError('Author not found')

23
services/cache.py
View File

@ -20,21 +20,21 @@ DEFAULT_FOLLOWS = {
} }
async def set_author_cache(author: dict, ttl=25 * 60 * 60): async def set_author_cache(author: dict):
payload = json.dumps(author, cls=CustomJSONEncoder) payload = json.dumps(author, cls=CustomJSONEncoder)
await redis.execute('SET', f'user:{author.get("user")}:author', ttl, payload) await redis.execute('SET', f'user:{author.get("user")}:author', payload)
await redis.execute('SET', f'id:{author.get("id")}:author', ttl, payload) await redis.execute('SET', f'id:{author.get("id")}:author', payload)
async def update_author_followers_cache(author_id: int, followers, ttl=25 * 60 * 60): async def update_author_followers_cache(author_id: int, followers):
payload = json.dumps(followers, cls=CustomJSONEncoder) payload = json.dumps(followers, cls=CustomJSONEncoder)
await redis.execute('SET', f'author:{author_id}:followers', ttl, payload) await redis.execute('SET', f'author:{author_id}:followers', payload)
async def set_follows_topics_cache(follows, author_id: int, ttl=25 * 60 * 60): async def set_follows_topics_cache(follows, author_id: int):
try: try:
payload = json.dumps(follows, cls=CustomJSONEncoder) payload = json.dumps(follows, cls=CustomJSONEncoder)
await redis.execute('SET', f'author:{author_id}:follows-topics', ttl, payload) await redis.execute('SET', f'author:{author_id}:follows-topics', payload)
except Exception as exc: except Exception as exc:
logger.error(exc) logger.error(exc)
import traceback import traceback
@ -43,15 +43,14 @@ async def set_follows_topics_cache(follows, author_id: int, ttl=25 * 60 * 60):
logger.error(exc) logger.error(exc)
async def set_follows_authors_cache(follows, author_id: int, ttl=25 * 60 * 60): async def set_follows_authors_cache(follows, author_id: int):
try: try:
payload = json.dumps(follows, cls=CustomJSONEncoder) payload = json.dumps(follows, cls=CustomJSONEncoder)
await redis.execute( await redis.execute('SET', f'author:{author_id}:follows-authors', payload)
'SET', f'author:{author_id}:follows-authors', ttl, payload except Exception as exc:
)
except Exception:
import traceback import traceback
logger.error(exc)
exc = traceback.format_exc() exc = traceback.format_exc()
logger.error(exc) logger.error(exc)