lkva/NDOJ
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Add cache for related problems

Browse source
This commit is contained in:
cuom1999 2023-01-28 10:46:17 -06:00
parent c3f2930d4a
commit e46775301c
1 changed files with 7 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
judge/utils/problems.py
View file

@ -238,6 +238,10 @@ def hot_problems(duration, limit):
def get_related_problems(profile, problem, limit=8): def get_related_problems(profile, problem, limit=8):
if not profile or not settings.ML_OUTPUT_PATH: if not profile or not settings.ML_OUTPUT_PATH:
return None return None
cache_key = "related_problems:%d:%d" % (profile.id, problem.id)
qs = cache.get(cache_key)
if qs is not None:
return qs
problemset = Problem.get_visible_problems(profile.user).values_list("id", flat=True) problemset = Problem.get_visible_problems(profile.user).values_list("id", flat=True)
problemset = problemset.exclude(id__in=user_completed_ids(profile)) problemset = problemset.exclude(id__in=user_completed_ids(profile))
problemset = problemset.exclude(id=problem.id) problemset = problemset.exclude(id=problem.id)
@ -249,4 +253,6 @@ def get_related_problems(profile, problem, limit=8):
seed = datetime.now().strftime("%d%m%Y") seed = datetime.now().strftime("%d%m%Y")
random.Random(seed).shuffle(results) random.Random(seed).shuffle(results)
results = results[:limit] results = results[:limit]
return [Problem.objects.get(id=i) for i in results] results = [Problem.objects.get(id=i) for i in results]
cache.set(cache_key, results, 21600)
return results