mirror
/
tile38
mirror of https://github.com/tidwall/tile38.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use WaitTimeout for MQTT

This commit is contained in:
tidwall 2019-10-08 11:13:18 -07:00
parent 6b2025ab0c
commit b092cea0d2
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
internal/endpoint/mqtt.go
View File

@ -3,9 +3,9 @@ package endpoint
import ( import (
"crypto/tls" "crypto/tls"
"crypto/x509" "crypto/x509"
"math/rand"
"fmt" "fmt"
"io/ioutil" "io/ioutil"
"math/rand"
"sync" "sync"
"time" "time"
@ -14,7 +14,8 @@ import (
) )
const ( const (
mqttExpiresAfter = time.Second * 30 mqttExpiresAfter = time.Second * 30
mqttPublishTimeout = time.Second * 5
) )
// MQTTConn is an endpoint connection // MQTTConn is an endpoint connection
@ -85,12 +86,12 @@ func (conn *MQTTConn) Send(msg string) error {
} }
ops = ops.SetTLSConfig(&config) ops = ops.SetTLSConfig(&config)
} }
//generate UUID for the client-id. //generate UUID for the client-id.
b := make([]byte, 16) b := make([]byte, 16)
_, err := rand.Read(b) _, err := rand.Read(b)
if err != nil { if err != nil {
log.Debugf("Failed to generate guid for the mqtt client. The endpoint will not work") log.Debugf("Failed to generate guid for the mqtt client. The endpoint will not work")
return err; 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:]) uuid := fmt.Sprintf("tile38-%x-%x-%x-%x-%x", b[0:4], b[4:6], b[6:8], b[8:10], b[10:])
@ -106,9 +107,8 @@ func (conn *MQTTConn) Send(msg string) error {
t := conn.conn.Publish(conn.ep.MQTT.QueueName, conn.ep.MQTT.Qos, t := conn.conn.Publish(conn.ep.MQTT.QueueName, conn.ep.MQTT.Qos,
conn.ep.MQTT.Retained, msg) conn.ep.MQTT.Retained, msg)
t.Wait()
if t.Error() != nil { if !t.WaitTimeout(mqttPublishTimeout) || t.Error() != nil {
conn.close() conn.close()
return t.Error() return t.Error()
} }