修正限流机制,将信号量从 50 更改为 5,并优化任务传递逻辑以确保正确的参数传递
All checks were successful
Sync Container Images / sync-images (push) Successful in 2m47s

This commit is contained in:
远野千束 2025-04-17 22:16:51 +08:00
parent 82517a10a7
commit 104a6822bc

View File

@ -52,12 +52,13 @@ async def docker_task(source: str, target: str) -> int | None:
return r
return 0
semaphore = asyncio.Semaphore(50)
semaphore = asyncio.Semaphore(5)
async def limited_task[T: Any](semaphore: asyncio.Semaphore, task: Callable[[], Coroutine[None, None, T]]) -> T:
async with semaphore:
return await task()
async def main():
async with aiofiles.open('images.yaml', 'r') as file:
config = await file.read()
@ -72,9 +73,9 @@ async def main():
for image in config.images:
if len(image.tags) > 0:
for tag in image.tags:
tasks.append(limited_task(semaphore, lambda: docker_task(f"{image.source}:{tag}", f"{image.target}:{tag}")))
tasks.append(limited_task(semaphore, lambda s=f"{image.source}:{tag}", t=f"{image.target}:{tag}": docker_task(s, t)))
else:
tasks.append(limited_task(semaphore, lambda: docker_task(image.source, image.target)))
tasks.append(limited_task(semaphore, lambda s=image.source, t=image.target: docker_task(s, t)))
results = await asyncio.gather(*tasks)
failed_tasks = sum(1 for result in results if result is not None and result != 0)