新增rabbitmq、haproxy组件采集 (#575)

* add

* add prober plugin for rabbitmq

* add prober plugin for haproxy

Co-authored-by: root <root@localhost.localdomain>
Co-authored-by: UlricQin <ulric.qin@gmail.com>

src/modules/monapi/plugins/all/all.go

@@ -11,6 +11,8 @@ import (
 	_ "github.com/didi/nightingale/src/modules/monapi/plugins/redis"
 	_ "github.com/didi/nightingale/src/modules/monapi/plugins/nginx"
 	_ "github.com/didi/nightingale/src/modules/monapi/plugins/elasticsearch"
+	_ "github.com/didi/nightingale/src/modules/monapi/plugins/rabbitmq"
+	_ "github.com/didi/nightingale/src/modules/monapi/plugins/haproxy"
 	_ "github.com/didi/nightingale/src/modules/monapi/plugins/tengine"
 	_ "github.com/didi/nightingale/src/modules/monapi/plugins/zookeeper"
 

src/modules/monapi/plugins/haproxy/haproxy.go

@@ -0,0 +1,71 @@
+package haproxy
+
+import (
+	"fmt"
+	"github.com/didi/nightingale/src/modules/monapi/collector"
+	"github.com/didi/nightingale/src/modules/monapi/plugins"
+	"github.com/didi/nightingale/src/toolkits/i18n"
+	"github.com/influxdata/telegraf"
+	"github.com/didi/nightingale/src/modules/monapi/plugins/haproxy/haproxy"
+)
+
+func init() {
+	collector.CollectorRegister(NewHaproxyCollector()) // for monapi
+	i18n.DictRegister(langDict)
+}
+
+type HaproxyCollector struct {
+	*collector.BaseCollector
+}
+
+func NewHaproxyCollector() *HaproxyCollector {
+	return &HaproxyCollector{BaseCollector: collector.NewBaseCollector(
+		"haproxy",
+		collector.RemoteCategory,
+		func() collector.TelegrafPlugin { return &HaproxyRule{} },
+	)}
+}
+
+var (
+	langDict = map[string]map[string]string{
+		"zh": map[string]string{
+			"Servers":          "Servers",
+			"Username":         "用户名",
+			"Password":         "密码",
+		},
+	}
+)
+
+type HaproxyRule struct {
+	Servers               []string `label:"Servers" json:"servers,required" example:"http://myhaproxy.com:1936/haproxy?stats"`
+	KeepFieldNames        bool     `label:"KeepFieldNames" json:"keepFieldNames" default:"false" description:"Setting this option to true results in the plugin keeping the original"`
+	Username              string   `label:"Username" json:"username" description:"specify username"`
+	Password              string   `label:"Password" json:"password" format:"password" description:"specify server password"`
+
+	plugins.ClientConfig
+}
+
+func (p *HaproxyRule) Validate() error {
+	if len(p.Servers) == 0 || p.Servers[0] == "" {
+		return fmt.Errorf("haproxy.rule.servers must be set")
+	}
+	return nil
+}
+
+func (p *HaproxyRule) TelegrafInput() (telegraf.Input, error) {
+	if err := p.Validate(); err != nil {
+		return nil, err
+	}
+
+	ha := &haproxy.Haproxy{
+
+		Servers:               p.Servers,
+        KeepFieldNames:		   p.KeepFieldNames,
+		Username:              p.Username,
+		Password:              p.Password,
+		ClientConfig:          p.ClientConfig.TlsClientConfig(),
+	}
+
+	return ha, nil
+}
+

src/modules/monapi/plugins/rabbitmq/rabbitmq.go

@@ -0,0 +1,98 @@
+package rabbitmq
+
+import (
+	"fmt"
+	"time"
+	"reflect"
+	"github.com/didi/nightingale/src/modules/monapi/collector"
+	"github.com/didi/nightingale/src/modules/monapi/plugins"
+	"github.com/didi/nightingale/src/toolkits/i18n"
+	"github.com/influxdata/telegraf"
+	"github.com/influxdata/telegraf/plugins/inputs/rabbitmq"
+)
+
+func init() {
+	collector.CollectorRegister(NewRabbitMQCollector()) // for monapi
+	i18n.DictRegister(langDict)
+}
+
+type RabbitMQCollector struct {
+	*collector.BaseCollector
+}
+
+func NewRabbitMQCollector() *RabbitMQCollector {
+	return &RabbitMQCollector{BaseCollector: collector.NewBaseCollector(
+		"rabbitMQ",
+		collector.RemoteCategory,
+		func() collector.TelegrafPlugin { return &RabbitMQRule{} },
+	)}
+}
+
+var (
+	langDict = map[string]map[string]string{
+		"zh": map[string]string{
+			"URL":              "URL",
+			"Name":             "节点名称",
+			"Username":         "用户名",
+			"Password":         "密码",
+			"header time out":  "请求超时时间",
+			"client time out":  "连接超时时间",
+			"nodes":            "MQ节点",
+			"queues":           "队列",
+			"exchanges":        "Exchange交换机",
+			"QueueNameInclude": "包含队列",
+			"QueueNameExclude": "排除队列",
+
+		},
+	}
+)
+
+type RabbitMQRule struct {
+	URL                   string   `label:"URL" json:"url,required" example:"http://localhost:15672"`
+	Name                  string   `label:"Name" json:"Name" description:"Tag added to rabbitmq_overview series"`
+	Username              string   `label:"Username" json:"username,required" description:"specify username"`
+	Password              string   `label:"Password" json:"password,required" format:"password" description:"specify server password"`
+	ResponseHeaderTimeout int      `label:"header time out" json:"header_timeout" default:"3" description:"for a server's response headers after fully writing the request"`
+	ClientTimeout         int      `label:"client time out" json:"client_timeout" default:"4" description:"for a server's response headers after fully writing the request"`
+	Nodes                 []string `label:"nodes" json:"nodes" description:"A list of nodes to gather as the rabbitmq_node measurement"`
+	Queues                []string `label:"queues" json:"queues" description:"A list of queues to gather as the rabbitmq_queue measurement"`
+	Exchanges             []string `label:"exchanges" json:"exchanges" description:"A list of exchanges to gather as the rabbitmq_exchange measurement"`
+	QueueNameInclude      []string `label:"queue name include" json:"queue_name_include" description:"Queues to include."`
+	QueueNameExclude      []string `label:"queue name exclude" json:"queue_name_exclude" description:"Queues to exclude."`
+	FederationUpstreamInclude []string `label:"FederationUpstreamInclude" json:"federation_upstream_include" description:"exchange filters include"`
+	FederationUpstreamExclude []string `label:"FederationUpstreamExclude" json:"federation_upstream_exclude" description:"exchange filters exclude"`
+	plugins.ClientConfig
+}
+
+func (p *RabbitMQRule) Validate() error {
+	if len(p.URL) == 0 || p.URL == "" {
+		return fmt.Errorf("rabbitmq.rule.servers must be set")
+	}
+	return nil
+}
+
+func (p *RabbitMQRule) TelegrafInput() (telegraf.Input, error) {
+	if err := p.Validate(); err != nil {
+		return nil, err
+	}
+
+	mq := &rabbitmq.RabbitMQ{
+
+		URL:                   p.URL,
+		Name:                  p.Name,
+		Username:              p.Username,
+		Password:              p.Password,
+		Nodes:                 p.Nodes,
+		Queues:                p.Queues,
+		Exchanges:             p.Exchanges,
+		QueueInclude:          p.QueueNameInclude,
+		QueueExclude:          p.QueueNameExclude,
+		ClientConfig:          p.ClientConfig.TlsClientConfig(),
+	}
+	v := reflect.ValueOf(&(mq.ResponseHeaderTimeout.Duration)).Elem()
+	v.Set(reflect.ValueOf(time.Second * time.Duration(p.ResponseHeaderTimeout)))
+	v1 := reflect.ValueOf(&(mq.ClientTimeout.Duration)).Elem()
+	v1.Set(reflect.ValueOf(time.Second * time.Duration(p.ClientTimeout)))
+	return mq, nil
+}
+