renaming plugins -> inputs
This commit is contained in:
Cameron Sparr
184
plugins/inputs/memcached/memcached.go
Normal file
184
plugins/inputs/memcached/memcached.go
Normal file
@@ -0,0 +1,184 @@
|
||||
package memcached
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"bytes"
|
||||
"fmt"
|
||||
"net"
|
||||
"strconv"
|
||||
"time"
|
||||
|
||||
"github.com/influxdb/telegraf/plugins/inputs"
|
||||
)
|
||||
|
||||
// Memcached is a memcached plugin
|
||||
type Memcached struct {
|
||||
Servers []string
|
||||
UnixSockets []string
|
||||
}
|
||||
|
||||
var sampleConfig = `
|
||||
# An array of address to gather stats about. Specify an ip on hostname
|
||||
# with optional port. ie localhost, 10.0.0.1:11211, etc.
|
||||
#
|
||||
# If no servers are specified, then localhost is used as the host.
|
||||
servers = ["localhost:11211"]
|
||||
# unix_sockets = ["/var/run/memcached.sock"]
|
||||
`
|
||||
|
||||
var defaultTimeout = 5 * time.Second
|
||||
|
||||
// The list of metrics that should be sent
|
||||
var sendMetrics = []string{
|
||||
"get_hits",
|
||||
"get_misses",
|
||||
"evictions",
|
||||
"limit_maxbytes",
|
||||
"bytes",
|
||||
"uptime",
|
||||
"curr_items",
|
||||
"total_items",
|
||||
"curr_connections",
|
||||
"total_connections",
|
||||
"connection_structures",
|
||||
"cmd_get",
|
||||
"cmd_set",
|
||||
"delete_hits",
|
||||
"delete_misses",
|
||||
"incr_hits",
|
||||
"incr_misses",
|
||||
"decr_hits",
|
||||
"decr_misses",
|
||||
"cas_hits",
|
||||
"cas_misses",
|
||||
"evictions",
|
||||
"bytes_read",
|
||||
"bytes_written",
|
||||
"threads",
|
||||
"conn_yields",
|
||||
}
|
||||
|
||||
// SampleConfig returns sample configuration message
|
||||
func (m *Memcached) SampleConfig() string {
|
||||
return sampleConfig
|
||||
}
|
||||
|
||||
// Description returns description of Memcached plugin
|
||||
func (m *Memcached) Description() string {
|
||||
return "Read metrics from one or many memcached servers"
|
||||
}
|
||||
|
||||
// Gather reads stats from all configured servers accumulates stats
|
||||
func (m *Memcached) Gather(acc inputs.Accumulator) error {
|
||||
if len(m.Servers) == 0 && len(m.UnixSockets) == 0 {
|
||||
return m.gatherServer(":11211", false, acc)
|
||||
}
|
||||
|
||||
for _, serverAddress := range m.Servers {
|
||||
if err := m.gatherServer(serverAddress, false, acc); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
for _, unixAddress := range m.UnixSockets {
|
||||
if err := m.gatherServer(unixAddress, true, acc); err != nil {
|
||||
return err
|
||||
}
|
||||
}
|
||||
|
||||
return nil
|
||||
}
|
||||
|
||||
func (m *Memcached) gatherServer(
|
||||
address string,
|
||||
unix bool,
|
||||
acc inputs.Accumulator,
|
||||
) error {
|
||||
var conn net.Conn
|
||||
if unix {
|
||||
conn, err := net.DialTimeout("unix", address, defaultTimeout)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer conn.Close()
|
||||
} else {
|
||||
_, _, err := net.SplitHostPort(address)
|
||||
if err != nil {
|
||||
address = address + ":11211"
|
||||
}
|
||||
|
||||
conn, err = net.DialTimeout("tcp", address, defaultTimeout)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
defer conn.Close()
|
||||
}
|
||||
|
||||
// Extend connection
|
||||
conn.SetDeadline(time.Now().Add(defaultTimeout))
|
||||
|
||||
// Read and write buffer
|
||||
rw := bufio.NewReadWriter(bufio.NewReader(conn), bufio.NewWriter(conn))
|
||||
|
||||
// Send command
|
||||
if _, err := fmt.Fprint(rw, "stats\r\n"); err != nil {
|
||||
return err
|
||||
}
|
||||
if err := rw.Flush(); err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
values, err := parseResponse(rw.Reader)
|
||||
if err != nil {
|
||||
return err
|
||||
}
|
||||
|
||||
// Add server address as a tag
|
||||
tags := map[string]string{"server": address}
|
||||
|
||||
// Process values
|
||||
fields := make(map[string]interface{})
|
||||
for _, key := range sendMetrics {
|
||||
if value, ok := values[key]; ok {
|
||||
// Mostly it is the number
|
||||
if iValue, errParse := strconv.ParseInt(value, 10, 64); errParse == nil {
|
||||
fields[key] = iValue
|
||||
} else {
|
||||
fields[key] = value
|
||||
}
|
||||
}
|
||||
}
|
||||
acc.AddFields("memcached", fields, tags)
|
||||
return nil
|
||||
}
|
||||
|
||||
func parseResponse(r *bufio.Reader) (map[string]string, error) {
|
||||
values := make(map[string]string)
|
||||
|
||||
for {
|
||||
// Read line
|
||||
line, _, errRead := r.ReadLine()
|
||||
if errRead != nil {
|
||||
return values, errRead
|
||||
}
|
||||
// Done
|
||||
if bytes.Equal(line, []byte("END")) {
|
||||
break
|
||||
}
|
||||
// Read values
|
||||
s := bytes.SplitN(line, []byte(" "), 3)
|
||||
if len(s) != 3 || !bytes.Equal(s[0], []byte("STAT")) {
|
||||
return values, fmt.Errorf("unexpected line in stats response: %q", line)
|
||||
}
|
||||
|
||||
// Save values
|
||||
values[string(s[1])] = string(s[2])
|
||||
}
|
||||
return values, nil
|
||||
}
|
||||
|
||||
func init() {
|
||||
inputs.Add("memcached", func() inputs.Input {
|
||||
return &Memcached{}
|
||||
})
|
||||
}
|
||||
160
plugins/inputs/memcached/memcached_test.go
Normal file
160
plugins/inputs/memcached/memcached_test.go
Normal file
@@ -0,0 +1,160 @@
|
||||
package memcached
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/influxdb/telegraf/testutil"
|
||||
"github.com/stretchr/testify/assert"
|
||||
"github.com/stretchr/testify/require"
|
||||
)
|
||||
|
||||
func TestMemcachedGeneratesMetrics(t *testing.T) {
|
||||
if testing.Short() {
|
||||
t.Skip("Skipping integration test in short mode")
|
||||
}
|
||||
|
||||
m := &Memcached{
|
||||
Servers: []string{testutil.GetLocalHost()},
|
||||
}
|
||||
|
||||
var acc testutil.Accumulator
|
||||
|
||||
err := m.Gather(&acc)
|
||||
require.NoError(t, err)
|
||||
|
||||
intMetrics := []string{"get_hits", "get_misses", "evictions",
|
||||
"limit_maxbytes", "bytes", "uptime", "curr_items", "total_items",
|
||||
"curr_connections", "total_connections", "connection_structures", "cmd_get",
|
||||
"cmd_set", "delete_hits", "delete_misses", "incr_hits", "incr_misses",
|
||||
"decr_hits", "decr_misses", "cas_hits", "cas_misses", "evictions",
|
||||
"bytes_read", "bytes_written", "threads", "conn_yields"}
|
||||
|
||||
for _, metric := range intMetrics {
|
||||
assert.True(t, acc.HasIntField("memcached", metric), metric)
|
||||
}
|
||||
}
|
||||
|
||||
func TestMemcachedParseMetrics(t *testing.T) {
|
||||
r := bufio.NewReader(strings.NewReader(memcachedStats))
|
||||
values, err := parseResponse(r)
|
||||
require.NoError(t, err, "Error parsing memcached response")
|
||||
|
||||
tests := []struct {
|
||||
key string
|
||||
value string
|
||||
}{
|
||||
{"pid", "23235"},
|
||||
{"uptime", "194"},
|
||||
{"time", "1449174679"},
|
||||
{"version", "1.4.14 (Ubuntu)"},
|
||||
{"libevent", "2.0.21-stable"},
|
||||
{"pointer_size", "64"},
|
||||
{"rusage_user", "0.000000"},
|
||||
{"rusage_system", "0.007566"},
|
||||
{"curr_connections", "5"},
|
||||
{"total_connections", "6"},
|
||||
{"connection_structures", "6"},
|
||||
{"reserved_fds", "20"},
|
||||
{"cmd_get", "0"},
|
||||
{"cmd_set", "0"},
|
||||
{"cmd_flush", "0"},
|
||||
{"cmd_touch", "0"},
|
||||
{"get_hits", "0"},
|
||||
{"get_misses", "0"},
|
||||
{"delete_misses", "0"},
|
||||
{"delete_hits", "0"},
|
||||
{"incr_misses", "0"},
|
||||
{"incr_hits", "0"},
|
||||
{"decr_misses", "0"},
|
||||
{"decr_hits", "0"},
|
||||
{"cas_misses", "0"},
|
||||
{"cas_hits", "0"},
|
||||
{"cas_badval", "0"},
|
||||
{"touch_hits", "0"},
|
||||
{"touch_misses", "0"},
|
||||
{"auth_cmds", "0"},
|
||||
{"auth_errors", "0"},
|
||||
{"bytes_read", "7"},
|
||||
{"bytes_written", "0"},
|
||||
{"limit_maxbytes", "67108864"},
|
||||
{"accepting_conns", "1"},
|
||||
{"listen_disabled_num", "0"},
|
||||
{"threads", "4"},
|
||||
{"conn_yields", "0"},
|
||||
{"hash_power_level", "16"},
|
||||
{"hash_bytes", "524288"},
|
||||
{"hash_is_expanding", "0"},
|
||||
{"expired_unfetched", "0"},
|
||||
{"evicted_unfetched", "0"},
|
||||
{"bytes", "0"},
|
||||
{"curr_items", "0"},
|
||||
{"total_items", "0"},
|
||||
{"evictions", "0"},
|
||||
{"reclaimed", "0"},
|
||||
}
|
||||
|
||||
for _, test := range tests {
|
||||
value, ok := values[test.key]
|
||||
if !ok {
|
||||
t.Errorf("Did not find key for metric %s in values", test.key)
|
||||
continue
|
||||
}
|
||||
if value != test.value {
|
||||
t.Errorf("Metric: %s, Expected: %s, actual: %s",
|
||||
test.key, test.value, value)
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
var memcachedStats = `STAT pid 23235
|
||||
STAT uptime 194
|
||||
STAT time 1449174679
|
||||
STAT version 1.4.14 (Ubuntu)
|
||||
STAT libevent 2.0.21-stable
|
||||
STAT pointer_size 64
|
||||
STAT rusage_user 0.000000
|
||||
STAT rusage_system 0.007566
|
||||
STAT curr_connections 5
|
||||
STAT total_connections 6
|
||||
STAT connection_structures 6
|
||||
STAT reserved_fds 20
|
||||
STAT cmd_get 0
|
||||
STAT cmd_set 0
|
||||
STAT cmd_flush 0
|
||||
STAT cmd_touch 0
|
||||
STAT get_hits 0
|
||||
STAT get_misses 0
|
||||
STAT delete_misses 0
|
||||
STAT delete_hits 0
|
||||
STAT incr_misses 0
|
||||
STAT incr_hits 0
|
||||
STAT decr_misses 0
|
||||
STAT decr_hits 0
|
||||
STAT cas_misses 0
|
||||
STAT cas_hits 0
|
||||
STAT cas_badval 0
|
||||
STAT touch_hits 0
|
||||
STAT touch_misses 0
|
||||
STAT auth_cmds 0
|
||||
STAT auth_errors 0
|
||||
STAT bytes_read 7
|
||||
STAT bytes_written 0
|
||||
STAT limit_maxbytes 67108864
|
||||
STAT accepting_conns 1
|
||||
STAT listen_disabled_num 0
|
||||
STAT threads 4
|
||||
STAT conn_yields 0
|
||||
STAT hash_power_level 16
|
||||
STAT hash_bytes 524288
|
||||
STAT hash_is_expanding 0
|
||||
STAT expired_unfetched 0
|
||||
STAT evicted_unfetched 0
|
||||
STAT bytes 0
|
||||
STAT curr_items 0
|
||||
STAT total_items 0
|
||||
STAT evictions 0
|
||||
STAT reclaimed 0
|
||||
END
|
||||
`
|
||||
Reference in New Issue
Block a user