Bugfix: Attach server to on_disconnect to prevent early closing

2023-06-08 08:11:23 +02:00
parent 7a8175ed84
commit 770be41b6d
5 changed files with 6 additions and 9 deletions
--- a/capnp/lib/capnp.pyx
+++ b/capnp/lib/capnp.pyx
@@ -2241,7 +2241,8 @@ cdef class TwoPartyServer:
        return _CapabilityClient()._init(helpers.bootstrapHelperServer(deref(self.thisptr)), self)

    cpdef on_disconnect(self) except +reraise_kj_exception:
-        return self._network.on_disconnect()
+        return _voidpromise_to_asyncio(deref(self._network.thisptr).onDisconnect()
+                                       .attach(capnp.heap[PyRefCounter](<PyObject*>self)))


 cdef class _AsyncIoStream: