Transport: Add termination protocol signature

This commit is contained in:
Josh Wu
2023-09-10 02:38:33 +08:00
parent 1e00c8f456
commit 548d5597c0

View File

@@ -63,6 +63,8 @@ class TransportSink(Protocol):
class TransportSource(Protocol): class TransportSource(Protocol):
terminated: asyncio.Future[None]
def set_packet_sink(self, sink: TransportSink) -> None: def set_packet_sink(self, sink: TransportSink) -> None:
... ...
@@ -430,6 +432,7 @@ class SnoopingTransport(Transport):
def __init__(self, source: TransportSource, snooper: Snooper): def __init__(self, source: TransportSource, snooper: Snooper):
self.source = source self.source = source
self.snooper = snooper self.snooper = snooper
self.terminated = source.terminated
def set_packet_sink(self, sink: TransportSink) -> None: def set_packet_sink(self, sink: TransportSink) -> None:
self.sink = sink self.sink = sink