mirror of
https://github.com/PiBrewing/craftbeerpi4.git
synced 2024-12-22 13:34:55 +01:00
just add topic filters for mqttsensor instead of crating tasks to accommodate new aiomqtt single message queue
This commit is contained in:
parent
f659eea65b
commit
1d1a7e8cfb
3 changed files with 12 additions and 40 deletions
|
@ -1,3 +1,3 @@
|
|||
__version__ = "4.3.2.a10"
|
||||
__version__ = "4.4.0.a1"
|
||||
__codename__ = "Winter Storm"
|
||||
|
||||
|
|
|
@ -157,29 +157,10 @@ class SatelliteController:
|
|||
except Exception as e:
|
||||
self.logger.warning("Failed to send sensorupdate via mqtt: {}".format(e))
|
||||
|
||||
def subcribe(self, topic, method):
|
||||
task = asyncio.create_task(self._subcribe(topic, method))
|
||||
return task
|
||||
|
||||
async def _subcribe(self, topic, method):
|
||||
self.error=False
|
||||
while True:
|
||||
try:
|
||||
if self.client._connected.done():
|
||||
await self.client.subscribe(topic)
|
||||
async for message in self.client.messages:
|
||||
if message.topic.matches(topic):
|
||||
await method(message.payload.decode())
|
||||
except asyncio.CancelledError:
|
||||
# Cancel
|
||||
self.logger.warning("Subscription {} Cancelled".format(topic))
|
||||
self.error=True
|
||||
except MqttError as e:
|
||||
self.logger.error("Sub MQTT Exception: {}".format(e))
|
||||
except Exception as e:
|
||||
self.logger.error("Sub Exception: {}".format(e))
|
||||
# wait before try to resubscribe
|
||||
if self.error == True:
|
||||
break
|
||||
else:
|
||||
await asyncio.sleep(5)
|
||||
def subscribe(self, topic, method):
|
||||
self.topic_filters.append((topic,method))
|
||||
return True
|
||||
|
||||
def unsubscribe(self, topic, method):
|
||||
self.topic_filters.remove((topic,method))
|
||||
return True
|
||||
|
|
|
@ -24,7 +24,7 @@ class MQTTSensor(CBPiSensor):
|
|||
self.payload_text = self.props.get("PayloadDictionary", None)
|
||||
if self.payload_text != None:
|
||||
self.payload_text = self.payload_text.split('.')
|
||||
self.mqtt_task = self.cbpi.satellite.subcribe(self.Topic, self.on_message)
|
||||
self.subscribed = self.cbpi.satellite.subscribe(self.Topic, self.on_message)
|
||||
self.value: float = 999
|
||||
self.timeout=int(self.props.get("Timeout", 60))
|
||||
self.starttime = time.time()
|
||||
|
@ -55,7 +55,7 @@ class MQTTSensor(CBPiSensor):
|
|||
pass
|
||||
|
||||
async def on_message(self, message):
|
||||
val = json.loads(message)
|
||||
val = json.loads(message.payload.decode())
|
||||
try:
|
||||
if self.payload_text is not None:
|
||||
for key in self.payload_text:
|
||||
|
@ -130,17 +130,8 @@ class MQTTSensor(CBPiSensor):
|
|||
return dict(value=self.value)
|
||||
|
||||
async def on_stop(self):
|
||||
was_cancelled=False
|
||||
if not self.mqtt_task.done():
|
||||
logging.info("Task not done -> cancelling")
|
||||
was_cancelled = self.mqtt_task.cancel()
|
||||
try:
|
||||
logging.info("Trying to call cancelled task")
|
||||
await self.mqtt_task
|
||||
except asyncio.CancelledError:
|
||||
logging.info("Task has been Cancelled")
|
||||
pass
|
||||
logging.info("Task cancelled: {}".format(was_cancelled))
|
||||
self.subscribed = self.cbpi.satellite.unsubscribe(self.Topic, self.on_message)
|
||||
|
||||
|
||||
def setup(cbpi):
|
||||
'''
|
||||
|
|
Loading…
Reference in a new issue