No more need for promise joining

capnp/helpers/capabilityHelper.cpp

@@ -58,32 +58,25 @@ void check_py_error() {
     }
 }
 
-inline kj::Promise<kj::Own<PyRefCounter>> maybeUnwrapPromise(PyObject * result) {
-  check_py_error();
-  auto promise = extract_promise(result);
-  Py_DECREF(result);
-  return kj::mv(*promise);
-}
-
 kj::Promise<kj::Own<PyRefCounter>> wrapPyFunc(kj::Own<PyRefCounter> func, kj::Own<PyRefCounter> arg) {
     GILAcquire gil;
-    // Creates an owned reference, which will be destroyed in maybeUnwrapPromise
     PyObject * result = PyObject_CallFunctionObjArgs(func->obj, arg->obj, NULL);
-    return maybeUnwrapPromise(result);
+    check_py_error();
+    return stealPyRef(result);
 }
 
 kj::Promise<kj::Own<PyRefCounter>> wrapPyFuncNoArg(kj::Own<PyRefCounter> func) {
     GILAcquire gil;
-    // Creates an owned reference, which will be destroyed in maybeUnwrapPromise
     PyObject * result = PyObject_CallFunctionObjArgs(func->obj, NULL);
-    return maybeUnwrapPromise(result);
+    check_py_error();
+    return stealPyRef(result);
 }
 
 kj::Promise<kj::Own<PyRefCounter>> wrapRemoteCall(kj::Own<PyRefCounter> func, capnp::Response<capnp::DynamicStruct> & arg) {
     GILAcquire gil;
-    // Creates an owned reference, which will be destroyed in maybeUnwrapPromise
+    PyObject * result = wrap_remote_call(func->obj, arg);
-    PyObject * ret = wrap_remote_call(func->obj, arg);
+    check_py_error();
-    return maybeUnwrapPromise(ret);
+    return stealPyRef(result);
 }
 
 ::kj::Promise<kj::Own<PyRefCounter>> then(kj::Own<kj::Promise<kj::Own<PyRefCounter>>> promise,

capnp/lib/capnp.pyx

@@ -155,19 +155,6 @@ cdef api VoidPromise * call_server_method(object server,
                 .format(method_name, str(ret)))
 
 
-cdef api Own[Promise[Own[PyRefCounter]]] extract_promise(object obj):
-    if type(obj) is _Promise:
-        return move((<_Promise>obj).thisptr)
-    elif type(obj) is _RemotePromise:
-        parent = (<_RemotePromise>obj)._parent
-        # We don't need parent anymore. Setting to none allows quicker garbage collection
-        (<_RemotePromise>obj)._parent = None
-        return capnp.heap[PyPromise](helpers.convert_to_pypromise(move((<_RemotePromise>obj).thisptr))
-                                     .attach(capnp.heap[PyRefCounter](<PyObject *>parent)))
-    else:
-        return capnp.heap[PyPromise](capnp.heap[PyRefCounter](<PyObject *>obj))
-
-
 cdef extern from "<kj/string.h>" namespace " ::kj":
     String strStructReader" ::kj::str"(C_DynamicStruct.Reader)
     String strStructBuilder" ::kj::str"(DynamicStruct_Builder)