tile38/internal/endpoint/mqtt.go

125 lines
2.5 KiB
Go
Raw Normal View History

2017-03-08 00:15:18 +03:00
package endpoint
import (
"crypto/tls"
"crypto/x509"
"math/rand"
2017-03-08 00:15:18 +03:00
"fmt"
"io/ioutil"
2017-03-08 00:15:18 +03:00
"sync"
"time"
paho "github.com/eclipse/paho.mqtt.golang"
"github.com/tidwall/tile38/internal/log"
2017-03-08 00:15:18 +03:00
)
const (
mqttExpiresAfter = time.Second * 30
)
2018-04-19 19:25:39 +03:00
// MQTTConn is an endpoint connection
type MQTTConn struct {
2017-03-08 00:15:18 +03:00
mu sync.Mutex
ep Endpoint
conn paho.Client
ex bool
t time.Time
}
2018-04-19 19:25:39 +03:00
// Expired returns true if the connection has expired
func (conn *MQTTConn) Expired() bool {
2017-03-08 00:15:18 +03:00
conn.mu.Lock()
defer conn.mu.Unlock()
if !conn.ex {
if time.Now().Sub(conn.t) > mqttExpiresAfter {
conn.close()
conn.ex = true
}
}
return conn.ex
}
2018-04-19 19:25:39 +03:00
func (conn *MQTTConn) close() {
2017-03-08 00:15:18 +03:00
if conn.conn != nil {
if conn.conn.IsConnected() {
conn.conn.Disconnect(250)
}
conn.conn = nil
}
}
2018-04-19 19:25:39 +03:00
// Send sends a message
func (conn *MQTTConn) Send(msg string) error {
2017-03-08 00:15:18 +03:00
conn.mu.Lock()
defer conn.mu.Unlock()
if conn.ex {
return errExpired
}
conn.t = time.Now()
if conn.conn == nil {
uri := fmt.Sprintf("tcp://%s:%d", conn.ep.MQTT.Host, conn.ep.MQTT.Port)
ops := paho.NewClientOptions()
if conn.ep.MQTT.CertFile != "" || conn.ep.MQTT.KeyFile != "" ||
conn.ep.MQTT.CACertFile != "" {
var config tls.Config
if conn.ep.MQTT.CertFile != "" || conn.ep.MQTT.KeyFile != "" {
cert, err := tls.LoadX509KeyPair(conn.ep.MQTT.CertFile,
conn.ep.MQTT.KeyFile)
if err != nil {
return err
}
config.Certificates = append(config.Certificates, cert)
}
if conn.ep.MQTT.CACertFile != "" {
// Load CA cert
caCert, err := ioutil.ReadFile(conn.ep.MQTT.CACertFile)
if err != nil {
return err
}
caCertPool := x509.NewCertPool()
caCertPool.AppendCertsFromPEM(caCert)
config.RootCAs = caCertPool
}
ops = ops.SetTLSConfig(&config)
}
//generate UUID for the client-id.
b := make([]byte, 16)
_, err := rand.Read(b)
if err != nil {
log.Debugf("Failed to generate guid for the mqtt client. The endpoint will not work")
return err;
}
uuid := fmt.Sprintf("tile38-%x-%x-%x-%x-%x", b[0:4], b[4:6], b[6:8], b[8:10], b[10:])
ops = ops.SetClientID(uuid).AddBroker(uri)
2017-03-08 00:15:18 +03:00
c := paho.NewClient(ops)
if token := c.Connect(); token.Wait() && token.Error() != nil {
return token.Error()
}
conn.conn = c
}
t := conn.conn.Publish(conn.ep.MQTT.QueueName, conn.ep.MQTT.Qos,
conn.ep.MQTT.Retained, msg)
2017-03-08 00:15:18 +03:00
t.Wait()
if t.Error() != nil {
conn.close()
return t.Error()
}
return nil
}
2018-04-19 19:25:39 +03:00
func newMQTTConn(ep Endpoint) *MQTTConn {
return &MQTTConn{
2017-03-08 00:15:18 +03:00
ep: ep,
t: time.Now(),
}
}