Optimize run_in_context calls

MagicStack · Sep 8, 2024 · 50497f9 · 50497f9
1 parent 79d5dcd
commit 50497f9
Showing 1 changed file with 15 additions and 3 deletions.
diff --git a/uvloop/loop.pyx b/uvloop/loop.pyx
@@ -98,23 +98,35 @@ cdef inline run_in_context(context, method):
     # See also: edgedb/edgedb#2222
     Py_INCREF(method)
     try:
-        return context.run(method)
+        Context_Enter(context)
+        try:
+            return method()
+        finally:
+            Context_Exit(context)
     finally:
         Py_DECREF(method)
 
 
 cdef inline run_in_context1(context, method, arg):
     Py_INCREF(method)
     try:
-        return context.run(method, arg)
+        Context_Enter(context)
+        try:
+            return method(arg)
+        finally:
+            Context_Exit(context)
     finally:
         Py_DECREF(method)
 
 
 cdef inline run_in_context2(context, method, arg1, arg2):
     Py_INCREF(method)
     try:
-        return context.run(method, arg1, arg2)
+        Context_Enter(context)
+        try:
+            return method(arg1, arg2)
+        finally:
+            Context_Exit(context)
     finally:
         Py_DECREF(method)