mirrors/pycapnp
1
0
Fork 0
mirror of https://github.com/capnproto/pycapnp.git synced 2025-03-04 08:24:43 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

Bugfix: Attach server to on_disconnect to prevent early closing

Browse source
This commit is contained in:
Lasse Blaauwbroek 2023-06-08 08:11:23 +02:00
parent 7a8175ed84
commit 770be41b6d
5 changed files with 6 additions and 9 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
capnp/lib/capnp.pyx
View file

@ -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:

3
examples/async_calculator_server.py
View file

@ -110,8 +110,7 @@ class CalculatorImpl(calculator_capnp.Calculator.Server):
async def new_connection(stream):
server = capnp.TwoPartyServer(stream, bootstrap=CalculatorImpl())
await server.on_disconnect()
await capnp.TwoPartyServer(stream, bootstrap=CalculatorImpl()).on_disconnect()
def parse_args():

3
examples/async_server.py
View file

@ -25,8 +25,7 @@ class ExampleImpl(thread_capnp.Example.Server):
async def new_connection(stream):
server = capnp.TwoPartyServer(stream, bootstrap=ExampleImpl())
await server.on_disconnect()
await capnp.TwoPartyServer(stream, bootstrap=ExampleImpl()).on_disconnect()
def parse_args():

3
examples/async_ssl_calculator_server.py
View file

@ -125,8 +125,7 @@ def parse_args():
async def new_connection(stream):
server = capnp.TwoPartyServer(stream, bootstrap=CalculatorImpl())
await server.on_disconnect()
await capnp.TwoPartyServer(stream, bootstrap=CalculatorImpl()).on_disconnect()
async def main():

3
examples/async_ssl_server.py
View file

@ -33,8 +33,7 @@ class ExampleImpl(thread_capnp.Example.Server):
async def new_connection(stream):
server = capnp.TwoPartyServer(stream, bootstrap=ExampleImpl())
await server.on_disconnect()
await capnp.TwoPartyServer(stream, bootstrap=ExampleImpl()).on_disconnect()
def parse_args():