mirror of
https://github.com/kaspanet/kaspad.git
synced 2025-10-14 00:59:33 +00:00
1ce7f21026
* [NOD-380] Add MQTT to the project. * [NOD-380] Add MQTT params to config. * [NOD-380] Implement connecting to an mqtt broker. * [NOD-380] Fix a comment. * [NOD-380] Removed unnecessary option. * [NOD-380] Added comments to MQTT functions. * [NOD-380] Fix copy+paste error. * [NOD-380] Make it so that all the mqtt flags must be passed together. * [NOD-380] Use activeConfig instead of passing it everywhere.
51 lines
1.2 KiB
Go
51 lines
1.2 KiB
Go
package mqtt
|
|
|
|
import (
|
|
"errors"
|
|
"github.com/daglabs/btcd/apiserver/config"
|
|
mqtt "github.com/eclipse/paho.mqtt.golang"
|
|
)
|
|
|
|
// client is an instance of the MQTT client, in case we have an active connection
|
|
var client mqtt.Client
|
|
|
|
// GetClient returns an instance of the MQTT client, in case we have an active connection
|
|
func GetClient() (mqtt.Client, error) {
|
|
if client == nil {
|
|
return nil, errors.New("MQTT is not connected")
|
|
}
|
|
return client, nil
|
|
}
|
|
|
|
// Connect initiates a connection to the MQTT server, if defined
|
|
func Connect() error {
|
|
cfg := config.ActiveConfig()
|
|
if cfg.MQTTBrokerAddress == "" {
|
|
// MQTT broker not defined -- nothing to do
|
|
return nil
|
|
}
|
|
|
|
options := mqtt.NewClientOptions()
|
|
options.AddBroker(cfg.MQTTBrokerAddress)
|
|
options.SetUsername(cfg.MQTTUser)
|
|
options.SetPassword(cfg.MQTTPassword)
|
|
options.SetAutoReconnect(true)
|
|
|
|
newClient := mqtt.NewClient(options)
|
|
if token := newClient.Connect(); token.Wait() && token.Error() != nil {
|
|
return token.Error()
|
|
}
|
|
client = newClient
|
|
|
|
return nil
|
|
}
|
|
|
|
// Close closes the connection to the MQTT server, if previously connected
|
|
func Close() {
|
|
if client == nil {
|
|
return
|
|
}
|
|
client.Disconnect(250)
|
|
client = nil
|
|
}
|