blob: 1f935cf97eefd42fb7b4ab92c7ada6c62420cab7 (
plain)
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
|
package main
import (
"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_RECONNECT_PERIOD = 2 * time.Second
)
type MQTTMessage struct {
Topic string
Payload []byte
}
func MQTTRun(tx chan MQTTMessage) {
opts := mqtt.NewClientOptions()
opts.AddBroker(BROKER)
opts.SetClientID(CLIENT_ID)
opts.SetCleanSession(true)
opts.SetOnConnectHandler(MQTTOnConnectHandler)
opts.SetConnectionLostHandler(MQTTConnectionLostHandler)
client := mqtt.NewClient(opts)
token := client.Connect()
success := token.WaitTimeout(MQTT_CONNECT_TIMEOUT)
if !success {
log.Printf("Timed out after %v.", MQTT_CONNECT_TIMEOUT)
return
}
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)
}
func MQTTConnectionLostHandler(c mqtt.Client, err error) {
log.Printf("Connection to MQTT broker lost: %v", err)
}
|
