telegraf/plugins/inputs/zookeeper/zookeeper.go

110 lines
2.4 KiB
Go
Raw Normal View History

package zookeeper
import (
"bufio"
"fmt"
"net"
"os"
"regexp"
"strconv"
"strings"
"time"
2016-01-20 20:19:03 +00:00
"github.com/influxdata/telegraf/plugins/inputs"
)
// Zookeeper is a zookeeper plugin
type Zookeeper struct {
Servers []string
}
var sampleConfig = `
# An array of address to gather stats about. Specify an ip or hostname
# with port. ie localhost:2181, 10.0.0.1:2181, etc.
# If no servers are specified, then localhost is used as the host.
# If no port is specified, 2181 is used
servers = [":2181"]
`
var defaultTimeout = time.Second * time.Duration(5)
// SampleConfig returns sample configuration message
func (z *Zookeeper) SampleConfig() string {
return sampleConfig
}
// Description returns description of Zookeeper plugin
func (z *Zookeeper) Description() string {
return `Reads 'mntr' stats from one or many zookeeper servers`
}
// Gather reads stats from all configured servers accumulates stats
2016-01-07 20:39:43 +00:00
func (z *Zookeeper) Gather(acc inputs.Accumulator) error {
if len(z.Servers) == 0 {
return nil
}
for _, serverAddress := range z.Servers {
if err := z.gatherServer(serverAddress, acc); err != nil {
return err
}
}
return nil
}
2016-01-07 20:39:43 +00:00
func (z *Zookeeper) gatherServer(address string, acc inputs.Accumulator) error {
_, _, err := net.SplitHostPort(address)
if err != nil {
address = address + ":2181"
}
c, err := net.DialTimeout("tcp", address, defaultTimeout)
if err != nil {
fmt.Fprintln(os.Stderr, err)
return err
}
defer c.Close()
fmt.Fprintf(c, "%s\n", "mntr")
rdr := bufio.NewReader(c)
scanner := bufio.NewScanner(rdr)
2015-12-15 21:11:23 +00:00
service := strings.Split(address, ":")
if len(service) != 2 {
return fmt.Errorf("Invalid service address: %s", address)
}
tags := map[string]string{"server": service[0], "port": service[1]}
fields := make(map[string]interface{})
for scanner.Scan() {
line := scanner.Text()
re := regexp.MustCompile(`^zk_(\w+)\s+([\w\.\-]+)`)
parts := re.FindStringSubmatch(string(line))
2015-12-15 21:11:23 +00:00
if len(parts) != 3 {
return fmt.Errorf("unexpected line in mntr response: %q", line)
}
measurement := strings.TrimPrefix(parts[1], "zk_")
sValue := string(parts[2])
iVal, err := strconv.ParseInt(sValue, 10, 64)
if err == nil {
2015-12-15 21:11:23 +00:00
fields[measurement] = iVal
} else {
2015-12-15 21:11:23 +00:00
fields[measurement] = sValue
}
}
2015-12-15 21:11:23 +00:00
acc.AddFields("zookeeper", fields, tags)
return nil
}
func init() {
2016-01-07 20:39:43 +00:00
inputs.Add("zookeeper", func() inputs.Input {
return &Zookeeper{}
})
}