Skip to content

Commit f62f986

Browse files
authored
Merge pull request #322 from SomberNight/202510_fix_replydc
session: fix ReplyAndDisconnect. was broken for PaddedRSTransport
2 parents 36af30a + 6bbd311 commit f62f986

File tree

1 file changed

+7
-0
lines changed

1 file changed

+7
-0
lines changed

src/electrumx/server/transport.py

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -109,3 +109,10 @@ def connection_made(self, transport: asyncio.BaseTransport):
109109
super().connection_made(transport)
110110
coro = self.session.taskgroup.spawn(self._poll_sbuffer())
111111
self._sbuffer_task = self.loop.create_task(coro)
112+
113+
async def close(self, *args, **kwargs):
114+
'''Close the connection and return when closed.'''
115+
# Flush buffer before disconnecting. This makes ReplyAndDisconnect work:
116+
self._force_send = True
117+
self._maybe_consume_sbuffer()
118+
await super().close(*args, **kwargs)

0 commit comments

Comments
 (0)