Skip to content

Commit 2c9b22c

Browse files
ArnoutDbouwew
authored andcommitted
Only up the NodeResponse expected counter when the Stick sends an Accept
1 parent d53077d commit 2c9b22c

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

plugwise_usb/connection/manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -38,7 +38,7 @@ def __init__(self) -> None:
3838

3939
@property
4040
def queue_depth(self) -> int:
41-
return self._sender.processed_messages - self._receiver.processed_messages
41+
return self._sender.expected_responses - self._receiver.processed_messages
4242

4343
def correct_received_messages(self, correction: int) -> None:
4444
self._receiver.correct_processed_messages(correction)

plugwise_usb/connection/sender.py

Lines changed: 5 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -38,16 +38,16 @@ def __init__(self, stick_receiver: StickReceiver, transport: Transport) -> None:
3838
self._loop = get_running_loop()
3939
self._receiver = stick_receiver
4040
self._transport = transport
41-
self._processed_msgs = 0
41+
self._expected_responses = 0
4242
self._stick_response: Future[StickResponse] | None = None
4343
self._stick_lock = Lock()
4444
self._current_request: None | PlugwiseRequest = None
4545
self._unsubscribe_stick_response: Callable[[], None] | None = None
4646

4747
@property
48-
def processed_messages(self) -> int:
48+
def expected_responses(self) -> int:
4949
"""Return the number of processed messages."""
50-
return self._processed_msgs
50+
return self._expected_responses
5151

5252
async def start(self) -> None:
5353
"""Start the sender."""
@@ -115,6 +115,7 @@ async def write_request_to_port(self, request: PlugwiseRequest) -> None:
115115
)
116116
)
117117
else:
118+
self._expected_responses += 1
118119
request.seq_id = response.seq_id
119120
await request.subscribe_to_response(
120121
self._receiver.subscribe_to_stick_responses,
@@ -141,7 +142,7 @@ async def write_request_to_port(self, request: PlugwiseRequest) -> None:
141142
finally:
142143
self._stick_response.cancel()
143144
self._stick_lock.release()
144-
self._processed_msgs += 1
145+
145146

146147
async def _process_stick_response(self, response: StickResponse) -> None:
147148
"""Process stick response."""

0 commit comments

Comments
 (0)