Upstream/aioesphomeapi
1
0
Fork 0
mirror of https://github.com/esphome/aioesphomeapi.git synced 2024-11-29 13:15:10 +01:00
Code Issues Projects Releases Wiki Activity

Allow any message from the remote to cancel the pong timer (#401)

Browse Source
This commit is contained in:
J. Nick Koston 2023-03-13 09:56:36 -10:00 committed by GitHub
parent 471abb5fea
commit eacf758900
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
aioesphomeapi/connection.py
View File

@ -51,7 +51,7 @@ _LOGGER = logging.getLogger(__name__)
BUFFER_SIZE = 1024 * 1024 # Set buffer limit to 1MB BUFFER_SIZE = 1024 * 1024 # Set buffer limit to 1MB
INTERNAL_MESSAGE_TYPES = {GetTimeRequest, PingRequest, PingResponse, DisconnectRequest} INTERNAL_MESSAGE_TYPES = {GetTimeRequest, PingRequest, DisconnectRequest}
PING_REQUEST_MESSAGE = PingRequest() PING_REQUEST_MESSAGE = PingRequest()
PING_RESPONSE_MESSAGE = PingResponse() PING_RESPONSE_MESSAGE = PingResponse()
@ -679,6 +679,11 @@ class APIConnection:
_LOGGER.debug("%s: Got message of type %s: %s", self.log_name, msg_type, msg) _LOGGER.debug("%s: Got message of type %s: %s", self.log_name, msg_type, msg)
if self._pong_timer:
# Any valid message from the remote cancels the pong timer
# as we know the connection is still alive
self._async_cancel_pong_timer()
for handler in self._message_handlers.get(msg_type, [])[:]: for handler in self._message_handlers.get(msg_type, [])[:]:
handler(msg) handler(msg)
@ -687,11 +692,7 @@ class APIConnection:
if msg_type not in INTERNAL_MESSAGE_TYPES: if msg_type not in INTERNAL_MESSAGE_TYPES:
return return
if msg_type is PingResponse: if msg_type is DisconnectRequest:
# We got a pong so we know the ESP is alive, cancel the timer
# that will disconnect us
self._async_cancel_pong_timer()
elif msg_type is DisconnectRequest:
self.send_message(DisconnectResponse()) self.send_message(DisconnectResponse())
self._connection_state = ConnectionState.CLOSED self._connection_state = ConnectionState.CLOSED
self._expected_disconnect = True self._expected_disconnect = True