Bennett/g0v0-server
1
0
Fork 1
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat(beatmapset): add mechanism to prevent duplicate missing beatmapset additions

Browse Source
This commit is contained in:
MingxuanGame
2025-10-02 07:37:33 +00:00
parent 3f6776847e
commit 6252e04253
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
app/service/beatmapset_update_service.py
View File

@@ -123,6 +123,7 @@ class ProcessingBeatmapset:
class BeatmapsetUpdateService: class BeatmapsetUpdateService:
def __init__(self, fetcher: "Fetcher"): def __init__(self, fetcher: "Fetcher"):
self.fetcher = fetcher self.fetcher = fetcher
self._adding_missing = False
async def add_missing_beatmapset(self, beatmapset_id: int) -> bool: async def add_missing_beatmapset(self, beatmapset_id: int) -> bool:
beatmapset = await self.fetcher.get_beatmapset(beatmapset_id) beatmapset = await self.fetcher.get_beatmapset(beatmapset_id)
@@ -136,6 +137,9 @@ class BeatmapsetUpdateService:
return True return True
async def add_missing_beatmapsets(self): async def add_missing_beatmapsets(self):
if self._adding_missing:
return
self._adding_missing = True
async with with_db() as session: async with with_db() as session:
missings = await session.exec( missings = await session.exec(
select(Beatmapset.id).where( select(Beatmapset.id).where(
@@ -156,6 +160,7 @@ class BeatmapsetUpdateService:
total += 1 total += 1
if total > 0: if total > 0:
logger.opt(colors=True).info(f"<cyan>[BeatmapsetUpdateService]</cyan> added {total} missing beatmapset") logger.opt(colors=True).info(f"<cyan>[BeatmapsetUpdateService]</cyan> added {total} missing beatmapset")
self._adding_missing = False
async def add(self, beatmapset: BeatmapsetResp): async def add(self, beatmapset: BeatmapsetResp):
if ( if (