8.4. Gather

이전 장에서는 asyncio.create_task() 가 코루틴을 스케줄링하면서도 그것을 기다리지 않는다는 것을 보여주었습니다. 짝이 되는 연산 – 여러 awaitable을 동시에 실행하고 그것들 모두를 기다리는 것 – 은 asyncio.gather() 입니다.

8.4.1. 기본 형태

import asyncio

async def fetch(name, delay_ms):
    await asyncio.sleep_ms(delay_ms)
    return name

async def main():
    results = await asyncio.gather(
        fetch("a", 100),
        fetch("b", 200),
        fetch("c", 50),
    )
    print(results)

asyncio.run(main())

출력:

['a', 'b', 'c']

두 가지를 주목하세요. 첫째, 결과 리스트는 코루틴이 완료된 순서가 아니라 gather에 전달된 인자의 순서입니다 – "c" 가 먼저 반환되었지만 여전히 세 번째 항목입니다. 둘째, 호출은 350 ms 가 아니라 총 200 ms 가 걸렸습니다: 세 개의 sleep이 동시에 실행되었고, gather() 는 가장 느린 것이 완료되는 즉시 반환합니다.

두 사실 모두 같은 원천에서 나옵니다. gather() 는 아직 작업이 아닌 각 인자(예제의 코루틴들)를 감싸서 루프에 스케줄링하고, 그것들이 모두 완료될 때까지 호출하는 코루틴을 일시 중단한 다음, 원래 순서대로 결과를 반환합니다.

8.4.2. 언제 사용해야 하는가

애플리케이션이 N개의 awaitable 연산을 가지고 있고 계속하기 전에 그것들 모두의 결과를 원하는 모든 경우입니다. 전형적인 예:

  • 여러 네트워크 요청을 병렬로 발행하고 모든 응답을 기다리기.

  • 결합된 결과를 처리하기 전에 여러 센서로부터 동시에 읽기.

  • 애플리케이션 단계의 끝에서 수명이 짧은 여러 헬퍼 작업을 합치기.

애플리케이션이 시작하여 프로그램 수명 동안 계속 실행되도록 놔두는 장기 실행 백그라운드 작업에는 적합한 도구가 아닙니다 – 그것들은 여전히 create_task() 의 일입니다. gather() 는 팬아웃 / 팬인 패턴을 위한 것입니다: 작업을 나누고, 동시에 수행하고, 다시 합칩니다.

8.4.3. 그룹 내의 예외

수집된 awaitable 중 하나라도 예외를 발생시키면, 기본 동작은 gather 호출 밖으로 그 예외를 다시 발생시키는 것입니다. 아직 완료되지 않은 형제들은 백그라운드에서 취소됩니다.

그것은 보통 애플리케이션이 원하는 바입니다 – N개의 병렬 작업 중 하나가 실패했으니 결합된 연산이 실패한 것이고, 따라서 나머지에 시간을 쓰는 것을 멈춥니다. 때로는 애플리케이션이 반대를 원합니다: 각 awaitable이 독립적으로 완료(또는 실패)하도록 두고, 이후에 결과를 검사하는 것입니다. 그것을 위해서는 return_exceptions=True 를 전달하세요:

results = await asyncio.gather(
    fetch_or_fail("ok"),
    fetch_or_fail("bad"),
    return_exceptions=True,
)
# results == ["ok-value", OSError(...)]

이제 반환된 리스트의 각 항목은 정상적인 반환값 또는 해당 awaitable이 발생시킨 예외 중 하나입니다. 호출자는 isinstance(r, Exception) 을 확인하여 둘을 구별합니다.

8.4.4. 취소

gather 자체를 취소하는 것 – 그것을 대기 중이던 작업을 취소함으로써 – 은 그 안에서 아직 실행 중인 모든 awaitable을 취소합니다. 타임아웃과 취소 페이지는 취소가 호출 체인을 통해 어떻게 전파되는지 자세히 다룹니다.