Zabbix

/*

** Zabbix

** Copyright (C) 2001-2023 Zabbix SIA

**

** This program is free software; you can redistribute it and/or modify

** it under the terms of the GNU General Public License as published by

** the Free Software Foundation; either version 2 of the License, or

** (at your option) any later version.

**

** This program is distributed in the hope that it will be useful,

** but WITHOUT ANY WARRANTY; without even the implied warranty of

** MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the

** GNU General Public License for more details.

**

** You should have received a copy of the GNU General Public License

** along with this program; if not, write to the Free Software

** Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.

**

**/

​

/*

** We use the library Eclipse Paho (eclipse/paho.mqtt.golang), which is

** distributed under the terms of the Eclipse Distribution License 1.0 (The 3-Clause BSD License)

** available at https://www.eclipse.org/org/documents/edl-v10.php

**/

​

package mqtt

​

import (

    "crypto/rand"

    "crypto/tls"

    "encoding/json"

    "errors"

    "fmt"

    "net/url"

    "strings"

    "time"

​

    "git.zabbix.com/ap/plugin-support/metric"

    "git.zabbix.com/ap/plugin-support/plugin"

    "git.zabbix.com/ap/plugin-support/tlsconfig"

    "git.zabbix.com/ap/plugin-support/zbxerr"

    mqtt "github.com/eclipse/paho.mqtt.golang"

    "zabbix.com/pkg/itemutil"

    "zabbix.com/pkg/version"

    "zabbix.com/pkg/watch"

)

​

const (

    pluginName = "MQTT"

)

​

type mqttClient struct {

    client    mqtt.Client

    broker    broker

    subs      map[string]*mqttSub

    opts      *mqtt.ClientOptions

    connected bool

}

​

type mqttSub struct {

    broker   broker

    topic    string

    wildCard bool

}

​

type broker struct {

    url      string

    username string

    password string

}

type Plugin struct {

    plugin.Base

    options     Options

    manager     *watch.Manager

    mqttClients map[broker]*mqttClient

}

​

var impl Plugin

​

func (p *Plugin) createOptions(

    clientid, username, password string, b broker, details tlsconfig.Details) (*mqtt.ClientOptions, error) {

    opts := mqtt.NewClientOptions().AddBroker(b.url).SetClientID(clientid).SetCleanSession(true).SetConnectTimeout(

        time.Duration(impl.options.Timeout) * time.Second)

    if username != "" {

        opts.SetUsername(username)

        if password != "" {

            opts.SetPassword(password)

        }

    }

​