mirror of
https://github.com/gravitl/netmaker.git
synced 2025-10-05 16:57:51 +08:00
add ids to mqtt
This commit is contained in:
3
mq/mq.go
3
mq/mq.go
@@ -14,6 +14,7 @@ import (
|
|||||||
"github.com/gravitl/netmaker/logger"
|
"github.com/gravitl/netmaker/logger"
|
||||||
"github.com/gravitl/netmaker/logic"
|
"github.com/gravitl/netmaker/logic"
|
||||||
"github.com/gravitl/netmaker/models"
|
"github.com/gravitl/netmaker/models"
|
||||||
|
"github.com/gravitl/netmaker/netclient/ncutils"
|
||||||
"github.com/gravitl/netmaker/servercfg"
|
"github.com/gravitl/netmaker/servercfg"
|
||||||
"github.com/gravitl/netmaker/serverctl"
|
"github.com/gravitl/netmaker/serverctl"
|
||||||
)
|
)
|
||||||
@@ -192,6 +193,8 @@ func SetupMQTT() mqtt.Client {
|
|||||||
opts := mqtt.NewClientOptions()
|
opts := mqtt.NewClientOptions()
|
||||||
broker := servercfg.GetMessageQueueEndpoint()
|
broker := servercfg.GetMessageQueueEndpoint()
|
||||||
opts.AddBroker(broker)
|
opts.AddBroker(broker)
|
||||||
|
id := ncutils.MakeRandomString(23)
|
||||||
|
opts.ClientID = id
|
||||||
client := mqtt.NewClient(opts)
|
client := mqtt.NewClient(opts)
|
||||||
tperiod := time.Now().Add(10 * time.Second)
|
tperiod := time.Now().Add(10 * time.Second)
|
||||||
for {
|
for {
|
||||||
|
@@ -93,6 +93,8 @@ func SetupMQTT(cfg *config.ClientConfig) mqtt.Client {
|
|||||||
opts := mqtt.NewClientOptions()
|
opts := mqtt.NewClientOptions()
|
||||||
server := getServerAddress(cfg)
|
server := getServerAddress(cfg)
|
||||||
opts.AddBroker(server + ":1883")
|
opts.AddBroker(server + ":1883")
|
||||||
|
id := ncutils.MakeRandomString(23)
|
||||||
|
opts.ClientID = id
|
||||||
opts.SetDefaultPublishHandler(All)
|
opts.SetDefaultPublishHandler(All)
|
||||||
client := mqtt.NewClient(opts)
|
client := mqtt.NewClient(opts)
|
||||||
tperiod := time.Now().Add(12 * time.Second)
|
tperiod := time.Now().Add(12 * time.Second)
|
||||||
|
@@ -32,6 +32,9 @@ import (
|
|||||||
// Version - version of the netclient
|
// Version - version of the netclient
|
||||||
var Version = "dev"
|
var Version = "dev"
|
||||||
|
|
||||||
|
// src - for random strings
|
||||||
|
var src = rand.NewSource(time.Now().UnixNano())
|
||||||
|
|
||||||
// MAX_NAME_LENGTH - maximum node name length
|
// MAX_NAME_LENGTH - maximum node name length
|
||||||
const MAX_NAME_LENGTH = 62
|
const MAX_NAME_LENGTH = 62
|
||||||
|
|
||||||
@@ -62,6 +65,14 @@ const DEFAULT_GC_PERCENT = 10
|
|||||||
// KEY_SIZE = ideal length for keys
|
// KEY_SIZE = ideal length for keys
|
||||||
const KEY_SIZE = 2048
|
const KEY_SIZE = 2048
|
||||||
|
|
||||||
|
// constants for random strings
|
||||||
|
const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
|
||||||
|
const (
|
||||||
|
letterIdxBits = 6 // 6 bits to represent a letter index
|
||||||
|
letterIdxMask = 1<<letterIdxBits - 1 // All 1-bits, as many as letterIdxBits
|
||||||
|
letterIdxMax = 63 / letterIdxBits // # of letter indices fitting in 63 bits
|
||||||
|
)
|
||||||
|
|
||||||
// Log - logs a message
|
// Log - logs a message
|
||||||
func Log(message string) {
|
func Log(message string) {
|
||||||
log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
|
log.SetFlags(log.Flags() &^ (log.Llongfile | log.Lshortfile))
|
||||||
@@ -618,3 +629,23 @@ func BoxDecrypt(encrypted []byte, senderPublicKey *[32]byte, recipientPrivateKey
|
|||||||
}
|
}
|
||||||
return decrypted, nil
|
return decrypted, nil
|
||||||
}
|
}
|
||||||
|
|
||||||
|
// MakeRandomString - generates a random string of len n
|
||||||
|
func MakeRandomString(n int) string {
|
||||||
|
sb := strings.Builder{}
|
||||||
|
sb.Grow(n)
|
||||||
|
// A src.Int63() generates 63 random bits, enough for letterIdxMax characters!
|
||||||
|
for i, cache, remain := n-1, src.Int63(), letterIdxMax; i >= 0; {
|
||||||
|
if remain == 0 {
|
||||||
|
cache, remain = src.Int63(), letterIdxMax
|
||||||
|
}
|
||||||
|
if idx := int(cache & letterIdxMask); idx < len(letterBytes) {
|
||||||
|
sb.WriteByte(letterBytes[idx])
|
||||||
|
i--
|
||||||
|
}
|
||||||
|
cache >>= letterIdxBits
|
||||||
|
remain--
|
||||||
|
}
|
||||||
|
|
||||||
|
return sb.String()
|
||||||
|
}
|
||||||
|
Reference in New Issue
Block a user