Always set result in PyFutureAwaitable

emmett-framework · Jan 30, 2025 · 338f813 · 338f813
1 parent 73b5771
commit 338f813
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/callbacks.rs b/src/callbacks.rs
@@ -571,6 +571,8 @@ impl PyFutureAwaitable {
 
     pub(crate) fn set_result(pyself: Py<Self>, py: Python, result: FutureResultToPy) {
         let rself = pyself.get();
+
+        _ = rself.result.set(result.into_pyobject(py).map(Bound::unbind));
         if rself
             .state
             .compare_exchange(
@@ -584,7 +586,6 @@ impl PyFutureAwaitable {
             return;
         }
 
-        let _ = rself.result.set(result.into_pyobject(py).map(Bound::unbind));
         let ack = rself.ack.read().unwrap();
         if let Some((cb, ctx)) = &*ack {
             let _ = rself.event_loop.clone_ref(py).call_method(