Adding upstream version 1.34.4.

Signed-off-by: Daniel Baumann <daniel@debian.org>

plugins/common/mqtt/mqtt_v5.go

@@ -0,0 +1,165 @@
+package mqtt
+
+import (
+	"context"
+	"fmt"
+	"net/url"
+	"time"
+
+	mqttv5auto "github.com/eclipse/paho.golang/autopaho"
+	mqttv5 "github.com/eclipse/paho.golang/paho"
+	paho "github.com/eclipse/paho.mqtt.golang"
+
+	"github.com/influxdata/telegraf/config"
+	"github.com/influxdata/telegraf/internal"
+	"github.com/influxdata/telegraf/logger"
+)
+
+type mqttv5Client struct {
+	client      *mqttv5auto.ConnectionManager
+	options     mqttv5auto.ClientConfig
+	username    config.Secret
+	password    config.Secret
+	timeout     time.Duration
+	qos         int
+	retain      bool
+	clientTrace bool
+	properties  *mqttv5.PublishProperties
+}
+
+func NewMQTTv5Client(cfg *MqttConfig) (*mqttv5Client, error) {
+	opts := mqttv5auto.ClientConfig{
+		KeepAlive:      uint16(cfg.KeepAlive),
+		OnConnectError: cfg.OnConnectionLost,
+	}
+	opts.ConnectPacketBuilder = func(c *mqttv5.Connect, _ *url.URL) (*mqttv5.Connect, error) {
+		c.CleanStart = cfg.PersistentSession
+		return c, nil
+	}
+
+	if time.Duration(cfg.ConnectionTimeout) >= 1*time.Second {
+		opts.ConnectTimeout = time.Duration(cfg.ConnectionTimeout)
+	}
+
+	if cfg.ClientID != "" {
+		opts.ClientID = cfg.ClientID
+	} else {
+		id, err := internal.RandomString(5)
+		if err != nil {
+			return nil, fmt.Errorf("generating random client ID failed: %w", err)
+		}
+		opts.ClientID = "Telegraf-Output-" + id
+	}
+
+	tlsCfg, err := cfg.ClientConfig.TLSConfig()
+	if err != nil {
+		return nil, err
+	}
+
+	if tlsCfg != nil {
+		opts.TlsCfg = tlsCfg
+	}
+
+	brokers := make([]*url.URL, 0)
+	servers, err := parseServers(cfg.Servers)
+	if err != nil {
+		return nil, err
+	}
+
+	for _, server := range servers {
+		if tlsCfg != nil {
+			server.Scheme = "tls"
+		}
+		brokers = append(brokers, server)
+	}
+	opts.BrokerUrls = brokers
+
+	// Build the v5 specific publish properties if they are present in the config.
+	// These should not change during the lifecycle of the client.
+	var properties *mqttv5.PublishProperties
+	if cfg.PublishPropertiesV5 != nil {
+		properties = &mqttv5.PublishProperties{
+			ContentType:   cfg.PublishPropertiesV5.ContentType,
+			ResponseTopic: cfg.PublishPropertiesV5.ResponseTopic,
+			TopicAlias:    cfg.PublishPropertiesV5.TopicAlias,
+		}
+
+		messageExpiry := time.Duration(cfg.PublishPropertiesV5.MessageExpiry)
+		if expirySeconds := uint32(messageExpiry.Seconds()); expirySeconds > 0 {
+			properties.MessageExpiry = &expirySeconds
+		}
+
+		properties.User = make([]mqttv5.UserProperty, 0, len(cfg.PublishPropertiesV5.UserProperties))
+		for k, v := range cfg.PublishPropertiesV5.UserProperties {
+			properties.User.Add(k, v)
+		}
+	}
+
+	return &mqttv5Client{
+		options:     opts,
+		timeout:     time.Duration(cfg.Timeout),
+		username:    cfg.Username,
+		password:    cfg.Password,
+		qos:         cfg.QoS,
+		retain:      cfg.Retain,
+		properties:  properties,
+		clientTrace: cfg.ClientTrace,
+	}, nil
+}
+
+func (m *mqttv5Client) Connect() (bool, error) {
+	user, err := m.username.Get()
+	if err != nil {
+		return false, fmt.Errorf("getting username failed: %w", err)
+	}
+	defer user.Destroy()
+	pass, err := m.password.Get()
+	if err != nil {
+		return false, fmt.Errorf("getting password failed: %w", err)
+	}
+	defer pass.Destroy()
+	m.options.ConnectUsername = user.String()
+	m.options.ConnectPassword = []byte(pass.String())
+
+	if m.clientTrace {
+		log := mqttLogger{logger.New("paho", "", "")}
+		m.options.Debug = log
+		m.options.Errors = log
+	}
+
+	client, err := mqttv5auto.NewConnection(context.Background(), m.options)
+	if err != nil {
+		return false, err
+	}
+	m.client = client
+	return false, client.AwaitConnection(context.Background())
+}
+
+func (m *mqttv5Client) Publish(topic string, body []byte) error {
+	ctx, cancel := context.WithTimeout(context.Background(), m.timeout)
+	defer cancel()
+
+	_, err := m.client.Publish(ctx, &mqttv5.Publish{
+		Topic:      topic,
+		QoS:        byte(m.qos),
+		Retain:     m.retain,
+		Payload:    body,
+		Properties: m.properties,
+	})
+
+	return err
+}
+
+func (*mqttv5Client) SubscribeMultiple(filters map[string]byte, callback paho.MessageHandler) error {
+	_, _ = filters, callback
+	panic("not implemented")
+}
+
+func (*mqttv5Client) AddRoute(topic string, callback paho.MessageHandler) {
+	_, _ = topic, callback
+	panic("not implemented")
+}
+
+func (m *mqttv5Client) Close() error {
+	return m.client.Disconnect(context.Background())
+}