Skip to content

Commit 5615061

Browse files
committed
Add private method to allow closing of channels
Signed-off-by: Mathias L. Baumann <[email protected]>
1 parent 4be4a3b commit 5615061

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

src/frequenz/sdk/actor/_channel_registry.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -49,3 +49,15 @@ def new_receiver(self, key: str) -> Receiver[Any]:
4949
if key not in self._channels:
5050
self._channels[key] = Broadcast(f"{self._name}-{key}")
5151
return self._channels[key].new_receiver()
52+
53+
async def _close_channel(self, key: str) -> None:
54+
"""Close a channel with the given key.
55+
56+
This method is private and should only be used in special cases.
57+
58+
Args:
59+
key: A key to identify the channel.
60+
"""
61+
if key in self._channels:
62+
if channel := self._channels.pop(key, None):
63+
await channel.close()

0 commit comments

Comments
 (0)