Honor timeouts when possible

This commit is contained in:
Daniel A. Wozniak 2023-07-25 12:25:33 -07:00 committed by Gareth J. Greenaway
parent a04fe85ffa
commit dadacf2685
3 changed files with 4 additions and 5 deletions

View file

@ -1475,7 +1475,7 @@ class TCPPublishServer(salt.transport.base.DaemonizedPublishServer):
async def publish_payload(self, payload, *args):
return await self.pub_server.publish_payload(payload)
def connect(self):
def connect(self, timeout=None):
self.pub_sock = salt.utils.asynchronous.SyncWrapper(
_TCPPubServerPublisher,
(
@ -1485,7 +1485,7 @@ class TCPPublishServer(salt.transport.base.DaemonizedPublishServer):
),
loop_kwarg="io_loop",
)
self.pub_sock.connect()
self.pub_sock.connect(timeout=timeout)
async def publish(self, payload, **kwargs):
"""

View file

@ -976,7 +976,7 @@ class PublishServer(salt.transport.base.DaemonizedPublishServer):
args=(self.publish_payload,),
)
def connect(self):
def connect(self, timeout=None):
"""
Create and connect this thread's zmq socket. If a publisher socket
already exists "pub_close" is called before creating and connecting a

View file

@ -384,8 +384,7 @@ class SaltEvent:
),
)
try:
# self.pusher.connect(timeout=timeout)
self.pusher.connect()
self.pusher.connect(timeout=timeout)
self.cpush = True
except tornado.iostream.StreamClosedError as exc:
log.debug("Unable to connect pusher: %s", exc)