aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--lib/_emerge/BinpkgFetcher.py32
1 files changed, 17 insertions, 15 deletions
diff --git a/lib/_emerge/BinpkgFetcher.py b/lib/_emerge/BinpkgFetcher.py
index 640eead91..e788cb05d 100644
--- a/lib/_emerge/BinpkgFetcher.py
+++ b/lib/_emerge/BinpkgFetcher.py
@@ -16,6 +16,7 @@ import portage
from portage import os
from portage.util._async.AsyncTaskFuture import AsyncTaskFuture
from portage.util._pty import _create_pty_or_pipe
+from portage.util.futures import asyncio
from portage.util.futures.compat_coroutine import coroutine
if sys.hexversion >= 0x3000000:
@@ -205,6 +206,7 @@ class _BinpkgFetcherProcess(SpawnProcess):
except OSError:
pass
+ @coroutine
def async_lock(self):
"""
This raises an AlreadyLocked exception if lock() is called
@@ -215,22 +217,22 @@ class _BinpkgFetcherProcess(SpawnProcess):
if self._lock_obj is not None:
raise self.AlreadyLocked((self._lock_obj,))
- result = self.scheduler.create_future()
-
- def acquired_lock(async_lock):
- if async_lock.wait() == os.EX_OK:
- self.locked = True
- result.set_result(None)
- else:
- result.set_exception(AssertionError(
- "AsynchronousLock failed with returncode %s"
- % (async_lock.returncode,)))
-
- self._lock_obj = AsynchronousLock(path=self.pkg_path,
+ async_lock = self._lock_obj = AsynchronousLock(path=self.pkg_path,
scheduler=self.scheduler)
- self._lock_obj.addExitListener(acquired_lock)
- self._lock_obj.start()
- return result
+ try:
+ yield async_lock.async_start()
+ yield async_lock.async_wait()
+ except asyncio.CancelledError:
+ if async_lock.poll() is None:
+ async_lock.cancel()
+ raise
+
+ if async_lock.returncode != os.EX_OK:
+ raise AssertionError(
+ "AsynchronousLock failed with returncode %s"
+ % (async_lock.returncode,))
+
+ self.locked = True
class AlreadyLocked(portage.exception.PortageException):
pass