diff --git a/janus/__init__.py b/janus/__init__.py index ac3bebe..479abc0 100644 --- a/janus/__init__.py +++ b/janus/__init__.py @@ -8,11 +8,17 @@ from queue import Empty as SyncQueueEmpty from queue import Full as SyncQueueFull -__version__ = '0.2.1' +__version__ = '0.2.2' log = logging.getLogger(__package__) +try: + ensure_future = asyncio.ensure_future +except NameError: + ensure_future = asyncio.async + + class Queue: def __init__(self, maxsize=0, *, loop=None): if loop is None: @@ -128,7 +134,7 @@ def f(): self._async_not_empty.notify() def task_maker(): - task = asyncio.async(f(), loop=self._loop) + task = ensure_future(f(), loop=self._loop) task.add_done_callback(self._pending.discard) self._pending.add(task) @@ -144,7 +150,7 @@ def f(): self._async_not_full.notify() def task_maker(): - task = asyncio.async(f(), loop=self._loop) + task = ensure_future(f(), loop=self._loop) task.add_done_callback(self._pending.discard) self._pending.add(task)