IQ.Lvbs/pycapnp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

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
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():