fabaff/platypush
1
0
Fork 0
forked from platypush/platypush
Code Issues Pull requests Projects Releases Packages Wiki Activity

Handle MQTT listeners by host with lists of topics, so we only need one listener thread per host

Browse source
This commit is contained in:
Fabio Manganiello 2019-03-07 23:03:12 +01:00
parent 84099b2ab7
commit 022262eb78
1 changed files with 8 additions and 7 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
platypush/backend/mqtt.py
View file

@ -116,8 +116,9 @@ class MqttBackend(Backend):
client.connect(host, port, 60)
client.loop_forever()
def on_connect(topic):
def on_connect(topics):
def handler(client, userdata, flags, rc):
for topic in topics:
client.subscribe(topic)
return handler
@ -135,18 +136,18 @@ class MqttBackend(Backend):
for i, listener in enumerate(listeners_conf):
host = listener.get('host')
port = listener.get('port', self._default_mqtt_port)
topic = listener.get('topic')
topics = listener.get('topics')
username = listener.get('username')
password = listener.get('password')
tls_cafile = listener.get('tls_cafile')
if not host or not topic:
self.logger.warning('No host nor topic specified for listener ' +
'n.{}'.format(i+1))
if not host or not topics:
self.logger.warning('No host nor list of topics specified for ' +
'listener n.{}'.format(i+1))
continue
client = mqtt.Client()
client.on_connect = on_connect(topic)
client.on_connect = on_connect(topics)
client.on_message = on_message
if username and password: