From 548d5597c06c3a01f8eabb9f703c25c9e82d6024 Mon Sep 17 00:00:00 2001 From: Josh Wu Date: Sun, 10 Sep 2023 02:38:33 +0800 Subject: [PATCH] Transport: Add termination protocol signature --- bumble/transport/common.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/bumble/transport/common.py b/bumble/transport/common.py index 5d5bdf1d..c0303086 100644 --- a/bumble/transport/common.py +++ b/bumble/transport/common.py @@ -63,6 +63,8 @@ class TransportSink(Protocol): class TransportSource(Protocol): + terminated: asyncio.Future[None] + def set_packet_sink(self, sink: TransportSink) -> None: ... @@ -430,6 +432,7 @@ class SnoopingTransport(Transport): def __init__(self, source: TransportSource, snooper: Snooper): self.source = source self.snooper = snooper + self.terminated = source.terminated def set_packet_sink(self, sink: TransportSink) -> None: self.sink = sink