Experimental Future.eager_set_result

Author: taras

Lib/asyncio/futures.py

@@ -72,6 +72,8 @@ class Future:
 
     __log_traceback = False
 
+    _eager_result = False
+
     def __init__(self, *, loop=None):
         """Initialize the future.
 
@@ -168,6 +170,30 @@ def cancel(self, msg=None):
         self.__schedule_callbacks()
         return True
 
+    def __execute_callbacks(self):
+        callbacks = self._callbacks[:]
+        if not callbacks:
+            return
+
+        self._callbacks[:] = []
+        for callback, ctx in callbacks:
+            # TODO: run_in_context
+            try:
+                callback(self)
+            except BaseException as exc:
+                cb = format_helpers._format_callback_source(
+                    callback, (self,),
+                    debug=self._loop.get_debug())
+                msg = f'Exception in callback {cb}'
+                context = {
+                    'message': msg,
+                    'exception': exc,
+                    'handle': self,
+                }
+                if self._source_traceback:
+                    context['source_traceback'] = self._source_traceback
+                self._loop.call_exception_handler(context)
+
     def __schedule_callbacks(self):
         """Internal: Ask the event loop to call all callbacks.
 
@@ -270,6 +296,19 @@ def set_result(self, result):
         self._state = _FINISHED
         self.__schedule_callbacks()
 
+    def eager_set_result(self, result):
+        """Mark the future done and set its result.
+
+        If the future is already done when this method is called, raises
+        InvalidStateError.
+        """
+        if self._state != _PENDING:
+            raise exceptions.InvalidStateError(f'{self._state}: {self!r}')
+        self._result = result
+        self._state = _FINISHED
+        self._eager_result = True
+        self.__execute_callbacks()
+
     def set_exception(self, exception):
         """Mark the future done and set an exception.
 

Lib/asyncio/tasks.py

@@ -370,7 +370,10 @@ def __wakeup(self, future):
             # Python eval loop would use `.send(value)` method call,
             # instead of `__next__()`, which is slower for futures
             # that return non-generator iterators from their `__iter__`.
-            self.__step()
+            if future._eager_result:
+                self.__eager_start()
+            else:
+                self.__step()
         self = None  # Needed to break cycles when an exception occurs.