From 77f3c03f2881f7bbf403025d6b69ab4bb55e47b6 Mon Sep 17 00:00:00 2001 From: "Daniel A. Wozniak" Date: Wed, 10 Apr 2024 14:42:37 -0700 Subject: [PATCH] Make sure on_recv callbacks are coroutines --- salt/minion.py | 17 ++++------------- 1 file changed, 4 insertions(+), 13 deletions(-) diff --git a/salt/minion.py b/salt/minion.py index 67886c30a56..3d48378a1ac 100644 --- a/salt/minion.py +++ b/salt/minion.py @@ -3366,7 +3366,6 @@ class Syndic(Minion): data["jid"], data["to"], io_loop=self.io_loop, - callback=lambda _: None, **kwargs, ) @@ -3693,11 +3692,7 @@ class SyndicManager(MinionBase): self.raw_events = [] def reconnect_event_bus(self, something): - # XXX: set_event_handler does not return anything! - # pylint: disable=assignment-from-no-return - future = self.local.event.set_event_handler(self._process_event) - self.io_loop.add_future(future, self.reconnect_event_bus) - # pylint: enable=assignment-from-no-return + self.local.event.set_event_handler(self._process_event) # Syndic Tune In def tune_in(self): @@ -3718,11 +3713,7 @@ class SyndicManager(MinionBase): self.raw_events = [] self._reset_event_aggregation() - # XXX: set_event_handler does not return anything! - # pylint: disable=assignment-from-no-return - future = self.local.event.set_event_handler(self._process_event) - self.io_loop.add_future(future, self.reconnect_event_bus) - # pylint: enable=assignment-from-no-return + self.local.event.set_event_handler(self._process_event) # forward events every syndic_event_forward_timeout self.forward_events = tornado.ioloop.PeriodicCallback( @@ -3736,7 +3727,7 @@ class SyndicManager(MinionBase): self.io_loop.start() - def _process_event(self, raw): + async def _process_event(self, raw): # TODO: cleanup: Move down into event class mtag, data = self.local.event.unpack(raw) log.trace("Got event %s", mtag) # pylint: disable=no-member @@ -3923,7 +3914,7 @@ class ProxyMinion(Minion): mp_call = _metaproxy_call(self.opts, "target_load") return mp_call(self, load) - def _handle_payload(self, payload): + async def _handle_payload(self, payload): mp_call = _metaproxy_call(self.opts, "handle_payload") return mp_call(self, payload)