asyncio의 queue를 써보자!

asyncio는 비동기 프로그래밍을 위한 라이브러리. 모든 것을 정리하자니 귀찮아서 샘플 코드 하나로 끝낸다.

• asyncio.ensure_future
  • loop에 schedule한다.
  • coroutine을 future로 감싸서 return한다.
• queue.put & queue.get
  • 둘 다 coroutine
• asyncio.get_event_loop
  • event loop을 가져온다.
  • 최신 버젼 만료
• loop.add_reader(fd, callback)
  • fd의 변화를 감지해서 callback을 호출한다.
• loop.run_until_complete(task or future)
  • 인자로 들어온 task나 future가 끝나기 전까지 돈다.
  • 실제로는 future.add_done_callback(_run_until_complete_cb)를 호출하며,
  • _run_until_complete_cb는 future._loop.stop()을 호출한다.
  • 즉 future가 끝나면 future가 실행되고있는 loop을 stop한다.
  • source code
• loop.remove_reader(fd)
  • fd 감시를 그만둔다.

import asyncio
import sys
from functools import partial

def handle_stdin(queue):
    data = sys.stdin.readline().strip()
    if data == 'q':
        loop = asyncio.get_event_loop()
        loop.remove_reader(sys.stdin)
    asyncio.ensure_future(queue.put(data)) 

async def tick(queue):
    stop = False
    while not stop:
        
        data = await queue.get()
        print('Data received: {}'.format(data))
        if data == 'q':
            stop = True
    print("tick finished.")

def main(): 
    queue = asyncio.Queue()
    loop = asyncio.get_event_loop()
    loop.add_reader(sys.stdin, partial(handle_stdin, queue))
    loop.run_until_complete(tick(queue))
    loop.close()

if __name__ == '__main__':
    main()`

asyncio: 비동기 프로그래밍과 관련된 함수와 클래스를 포함합니다.

sys: 시스템 특정 파라미터와 함수에 접근할 수 있도록 합니다.

partial: 함수를 부분적으로 호출할 수 있게 해줍니다.

1. handle_stdin 함수:
   • 표준 입력에서 데이터를 읽어옵니다.
   • 'q'를 입력받으면 이벤트 루프에서 표준 입력 리더를 제거합니다.
   • 읽은 데이터는 asyncio 큐에 추가됩니다.
2. tick 함수:
   • 비동기 함수로서, 큐에 데이터가 들어올 때까지 기다렸다가 데이터를 가져와 출력합니다.
   • 'q' 문자열을 받으면 이 함수의 반복 실행을 중단합니다.
3. main 함수:
   • asyncio의 이벤트 루프를 설정하고 실행합니다.
   • loop.add_reader: 표준 입력에 데이터가 들어올 때마다 handle_stdin 함수를 호출합니다.
   • loop.run_until_complete(tick(queue)): tick 함수가 완료될 때까지 이벤트 루프를 실행합니다.
   • 이후 이벤트 루프를 닫습니다.
4. if __name__ == '__main__'::
   • 이 코드가 스크립트로 직접 실행될 때 main 함수를 호출합니다.

**asyncio.Queue**는 async/await 문법에 적합하게 설계되어 있어서 비동기 작업에서의 데이터 교환에 최적화되어 있습니다.

주요 특징: