@@ -45,7 +45,7 @@ def __init__( # pylint: disable=too-many-arguments
4545 self ._resampler_request_channel = Broadcast [ComponentMetricRequest ](
4646 "resampler-request"
4747 )
48- self ._basic_receivers : dict [str , list [Receiver [Sample [Quantity ]]]] = {}
48+ self ._input_channels_receivers : dict [str , list [Receiver [Sample [Quantity ]]]] = {}
4949
5050 def power_senders (
5151 comp_ids : list [int ],
@@ -58,7 +58,7 @@ def power_senders(
5858 Sample [Quantity ], name
5959 ).new_sender ()
6060 )
61- self ._basic_receivers [name ] = [
61+ self ._input_channels_receivers [name ] = [
6262 self ._channel_registry .get_or_create (
6363 Sample [Quantity ], name
6464 ).new_receiver ()
@@ -77,7 +77,7 @@ def frequency_senders(
7777 Sample [Quantity ], name
7878 ).new_sender ()
7979 )
80- self ._basic_receivers [name ] = [
80+ self ._input_channels_receivers [name ] = [
8181 self ._channel_registry .get_or_create (
8282 Sample [Quantity ], name
8383 ).new_receiver (name )
@@ -119,19 +119,19 @@ def multi_phase_senders(
119119 ).new_sender (),
120120 ]
121121 )
122- self ._basic_receivers [p1_name ] = [
122+ self ._input_channels_receivers [p1_name ] = [
123123 self ._channel_registry .get_or_create (
124124 Sample [Quantity ], p1_name
125125 ).new_receiver ()
126126 for _ in range (namespaces )
127127 ]
128- self ._basic_receivers [p2_name ] = [
128+ self ._input_channels_receivers [p2_name ] = [
129129 self ._channel_registry .get_or_create (
130130 Sample [Quantity ], p2_name
131131 ).new_receiver ()
132132 for _ in range (namespaces )
133133 ]
134- self ._basic_receivers [p3_name ] = [
134+ self ._input_channels_receivers [p3_name ] = [
135135 self ._channel_registry .get_or_create (
136136 Sample [Quantity ], p3_name
137137 ).new_receiver ()
@@ -222,15 +222,18 @@ async def _handle_resampling_requests(self) -> None:
222222 name = request .get_channel_name ()
223223 if name in self ._forward_tasks :
224224 continue
225- basic_recv_name = f"{ request .component_id } :{ request .metric_id } "
226- recv = self ._basic_receivers [basic_recv_name ].pop ()
227- assert recv is not None
225+ input_chan_recv_name = f"{ request .component_id } :{ request .metric_id } "
226+ input_chan_recv = self ._input_channels_receivers [input_chan_recv_name ].pop ()
227+ assert input_chan_recv is not None
228+ output_chan_sender : Sender [Sample [Quantity ]] = (
229+ self ._channel_registry .get_or_create (
230+ Sample [Quantity ], name
231+ ).new_sender ()
232+ )
228233 task = asyncio .create_task (
229234 self ._channel_forward_messages (
230- recv ,
231- self ._channel_registry .get_or_create (
232- Sample [Quantity ], name
233- ).new_sender (),
235+ input_chan_recv ,
236+ output_chan_sender ,
234237 ),
235238 name = name ,
236239 )
0 commit comments