Fix presence events on TCP transport

This commit is contained in:
Jakub Recman 2022-10-06 13:57:09 +02:00 committed by Megan Wilhite
parent 2bb2604f74
commit 49a9532475
2 changed files with 4 additions and 2 deletions

View file

@ -774,7 +774,9 @@ class PubServerChannel:
secrets = kwargs.get("secrets", None)
if secrets is not None:
salt.master.SMaster.secrets = secrets
self.transport.publish_daemon(self.publish_payload, self.presence_callback)
self.transport.publish_daemon(
self.publish_payload, self.presence_callback, self.remove_presence_callback
)
def presence_callback(self, subscriber, msg):
if msg["enc"] != "aes":

View file

@ -938,7 +938,7 @@ class PubServer(salt.ext.tornado.tcpserver.TCPServer):
"Subscriber at %s has disconnected from publisher", client.address
)
client.close()
self._remove_client_present(client)
self.remove_presence_callback(client)
self.clients.discard(client)
log.trace("TCP PubServer finished publishing payload")