WIP: leaderboard state by action_id

This commit is contained in:
norohind committed 2021-12-15 12:10:21 +03:00
1 parent 3a568a9c89
commit 3f1efe0ac7
4 files changed
+47 -4

No files matched your search

+4
View File
@@ -27,3 +27,7 @@ class AbstractModel(abc.ABC):
@abc.abstractmethod
def get_leaderboard_sum_history(self, platform: str, leaderboard_type: str) -> list[dict]:
raise NotImplemented
@abc.abstractmethod
def get_leaderboard_by_action_id(self, action_id: int) -> list[dict]:
raise NotImplemented
+24
View File
@@ -166,3 +166,27 @@ class PostgresModel(AbstractModel):
cache.set(cache_key, json.dumps(result))
return result
def get_leaderboard_by_action_id(self, action_id: int) -> list[dict]:
cache_key = f'leaderboard_by_action_id_{action_id}'
cached_result: typing.Union[str, None] = cache.get(cache_key)
if cached_result is not None:
logger.debug(f'Cached result for {cache_key}')
return json.loads(cached_result)
with self.db.cursor(cursor_factory=psycopg2.extras.RealDictCursor) as cursor:
cursor.execute(
postgres_sql_requests.select_last_action_id,
{
'action_id': action_id
}
)
result: list[dict] = cursor.fetchall()
if not cache.disabled:
cache.set(cache_key, json.dumps(result))
return result
+8 -1
View File
@@ -94,4 +94,11 @@ where leaderboard_type = %(LB_type)s and platform = %(platform)s
group by action_id
order by "Timestamp UTC" desc
limit 1000;
"""
"""
select_leaderboard_by_action_id = """select
*
from squads_stats_states
where action_id = %(action_id)s
order by score desc;
"""