Resubscribe if not using persistent sessions

This commit is contained in:
Pascal Larin 2016-04-12 21:43:25 -04:00
parent 3648c85e16
commit b151a79fa5
1 changed files with 16 additions and 10 deletions

View File

@ -46,6 +46,8 @@ type MQTTConsumer struct {
// keep the accumulator internally:
acc telegraf.Accumulator
started bool false
}
var sampleConfig = `
@ -134,6 +136,7 @@ func (m *MQTTConsumer) Start(acc telegraf.Accumulator) error {
return nil
}
func onConnect(c *MQTT.Client) {
if (!m.PersistentSession || !m.started) {
topics := make(map[string]byte)
for _, topic := range m.Topics {
topics[topic] = byte(m.QoS)
@ -144,6 +147,8 @@ func (m *MQTTConsumer) Start(acc telegraf.Accumulator) error {
log.Printf("MQTT SUBSCRIBE ERROR\ntopics: %s\nerror: %s",
string(m.Topics), err.Error())
}
m.started = true;
}
}
// receiver() reads all incoming messages from the consumer, and parses them into
// influxdb metric points.
@ -178,6 +183,7 @@ func (m *MQTTConsumer) Stop() {
defer m.Unlock()
close(m.done)
m.client.Disconnect(200)
m.started = false
}
func (m *MQTTConsumer) Gather(acc telegraf.Accumulator) error {