1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
|
package main
import (
"fmt"
"log"
"time"
mqtt "github.com/eclipse/paho.mqtt.golang"
)
const (
BROKER = `tcp://127.0.0.1:1883`
CLIENT_ID = `siaserver`
TOPIC_PREFIX = `sia`
QOS = byte(1)
RETAINED = true
MQTT_CONNECT_TIMEOUT = 1 * time.Second
MQTT_DISCONNECT_TIMEOUT_US = 500
MQTT_KEEPALIVE_PERIOD = 2 * time.Second
)
var (
mqttServerHealthTopic string
)
type MQTTMessage struct {
Topic string
Payload []byte
}
func init() {
mqttServerHealthTopic = fmt.Sprintf("%s/server/health", TOPIC_PREFIX)
}
func MQTTRun(tx chan MQTTMessage) {
opts := mqtt.NewClientOptions()
opts.AddBroker(BROKER)
opts.SetClientID(CLIENT_ID)
opts.SetCleanSession(true)
opts.SetOnConnectHandler(MQTTOnConnectHandler)
opts.SetConnectionLostHandler(MQTTConnectionLostHandler)
opts.SetAutoReconnect(true)
opts.SetConnectRetry(true)
opts.SetConnectTimeout(MQTT_CONNECT_TIMEOUT)
opts.SetKeepAlive(MQTT_KEEPALIVE_PERIOD)
opts.SetWill(mqttServerHealthTopic, `bad`, QOS, true)
client := mqtt.NewClient(opts)
token := client.Connect()
success := token.WaitTimeout(MQTT_CONNECT_TIMEOUT)
if !success {
log.Fatal("Initial connection to MQTT broker failed.")
}
defer client.Disconnect(MQTT_DISCONNECT_TIMEOUT_US)
for message := range tx {
client.Publish(message.Topic, QOS, RETAINED, message.Payload)
}
}
func MQTTOnConnectHandler(c mqtt.Client) {
log.Printf("Connected to MQTT broker (%s)", BROKER)
c.Publish(mqttServerHealthTopic, QOS, true, []byte(`good`))
}
func MQTTConnectionLostHandler(c mqtt.Client, err error) {
log.Printf("Connection to MQTT broker lost: %v", err)
}
|
