Skip to content

Commit

Permalink
Removes compat module, uses asyncio.ensure_future
Browse files Browse the repository at this point in the history
  • Loading branch information
dzen committed Jan 4, 2019
1 parent 88f71bc commit 8607c6a
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 15 deletions.
11 changes: 0 additions & 11 deletions aioamqp/compat.py

This file was deleted.

7 changes: 3 additions & 4 deletions aioamqp/protocol.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,6 @@
from . import frame as amqp_frame
from . import exceptions
from . import version
from .compat import ensure_future


logger = logging.getLogger(__name__)
Expand Down Expand Up @@ -244,7 +243,7 @@ async def start_connection(self, host, port, login, password, virtualhost, ssl=F

await self.ensure_open()
# for now, we read server's responses asynchronously
self.worker = ensure_future(self.run(), loop=self._loop)
self.worker = asyncio.ensure_future(self.run(), loop=self._loop)

async def get_frame(self):
"""Read the frame, and only decode its header
Expand Down Expand Up @@ -305,7 +304,7 @@ def _close_channels(self, reply_code=None, reply_text=None, exception=None):

if self._on_error_callback:
if asyncio.iscoroutinefunction(self._on_error_callback):
ensure_future(self._on_error_callback(exception), loop=self._loop)
asyncio.ensure_future(self._on_error_callback(exception), loop=self._loop)
else:
self._on_error_callback(exceptions.ChannelClosed(exception))

Expand Down Expand Up @@ -367,7 +366,7 @@ def _heartbeat_timer_send_reset(self):
self.server_heartbeat,
self._heartbeat_trigger_send.set)
if self._heartbeat_worker is None:
self._heartbeat_worker = ensure_future(self._heartbeat(), loop=self._loop)
self._heartbeat_worker = asyncio.ensure_future(self._heartbeat(), loop=self._loop)

def _heartbeat_stop(self):
self.server_heartbeat = None
Expand Down

0 comments on commit 8607c6a

Please sign in to comment.