saltstack/salt
1
0
Fork 0
mirror of https://github.com/saltstack/salt.git synced 2025-04-17 10:10:20 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 4

Use async with for lock

Browse source
This commit is contained in:
Daniel A. Wozniak 2023-07-31 14:39:59 -07:00 • committed by Gareth J. Greenaway
parent 3a84dadfd2
commit 43a3863532
1 changed files with 9 additions and 11 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
salt/transport/zeromq.py
View file

@ -1119,17 +1119,15 @@ class RequestClient(salt.transport.base.RequestClient):
async def _send_recv(self, message):
message = salt.payload.dumps(message)
await self.sending.acquire()
try:
await self.socket.send(message)
ret = await self.socket.recv()
except zmq.error.ZMQError:
self.close()
await self.connect()
await self.socket.send(message)
ret = await self.socket.recv()
finally:
self.sending.release()
async with self.sending:
try:
await self.socket.send(message)
ret = await self.socket.recv()
except zmq.error.ZMQError:
self.close()
await self.connect()
await self.socket.send(message)
ret = await self.socket.recv()
return salt.payload.loads(ret)
async def send(self, load, timeout=60):