140 lines
3.3 KiB
Go
140 lines
3.3 KiB
Go
package redis
|
|
|
|
import (
|
|
"bufio"
|
|
"errors"
|
|
"fmt"
|
|
"net"
|
|
"strconv"
|
|
"strings"
|
|
|
|
"github.com/influxdb/tivan/plugins"
|
|
)
|
|
|
|
type Gatherer struct {
|
|
Disabled bool
|
|
Address string
|
|
|
|
c net.Conn
|
|
buf []byte
|
|
}
|
|
|
|
var Tracking = map[string]string{
|
|
"uptime_in_seconds": "redis_uptime",
|
|
"connected_clients": "redis_clients",
|
|
"used_memory": "redis_used_memory",
|
|
"used_memory_rss": "redis_used_memory_rss",
|
|
"used_memory_peak": "redis_used_memory_peak",
|
|
"used_memory_lua": "redis_used_memory_lua",
|
|
"rdb_changes_since_last_save": "redis_rdb_changes_since_last_save",
|
|
"total_connections_received": "redis_total_connections_received",
|
|
"total_commands_processed": "redis_total_commands_processed",
|
|
"instantaneous_ops_per_sec": "redis_instantaneous_ops_per_sec",
|
|
"sync_full": "redis_sync_full",
|
|
"sync_partial_ok": "redis_sync_partial_ok",
|
|
"sync_partial_err": "redis_sync_partial_err",
|
|
"expired_keys": "redis_expired_keys",
|
|
"evicted_keys": "redis_evicted_keys",
|
|
"keyspace_hits": "redis_keyspace_hits",
|
|
"keyspace_misses": "redis_keyspace_misses",
|
|
"pubsub_channels": "redis_pubsub_channels",
|
|
"pubsub_patterns": "redis_pubsub_patterns",
|
|
"latest_fork_usec": "redis_latest_fork_usec",
|
|
"connected_slaves": "redis_connected_slaves",
|
|
"master_repl_offset": "redis_master_repl_offset",
|
|
"repl_backlog_active": "redis_repl_backlog_active",
|
|
"repl_backlog_size": "redis_repl_backlog_size",
|
|
"repl_backlog_histlen": "redis_repl_backlog_histlen",
|
|
"mem_fragmentation_ratio": "redis_mem_fragmentation_ratio",
|
|
"used_cpu_sys": "redis_used_cpu_sys",
|
|
"used_cpu_user": "redis_used_cpu_user",
|
|
"used_cpu_sys_children": "redis_used_cpu_sys_children",
|
|
"used_cpu_user_children": "redis_used_cpu_user_children",
|
|
}
|
|
|
|
var ErrProtocolError = errors.New("redis protocol error")
|
|
|
|
func (g *Gatherer) Gather(acc plugins.Accumulator) error {
|
|
if g.Address == "" || g.Disabled {
|
|
return nil
|
|
}
|
|
|
|
if g.c == nil {
|
|
c, err := net.Dial("tcp", g.Address)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
g.c = c
|
|
}
|
|
|
|
g.c.Write([]byte("info\n"))
|
|
|
|
r := bufio.NewReader(g.c)
|
|
|
|
line, err := r.ReadString('\n')
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
if line[0] != '$' {
|
|
return fmt.Errorf("bad line start: %s", ErrProtocolError)
|
|
}
|
|
|
|
line = strings.TrimSpace(line)
|
|
|
|
szStr := line[1:]
|
|
|
|
sz, err := strconv.Atoi(szStr)
|
|
if err != nil {
|
|
return fmt.Errorf("bad size string <<%s>>: %s", szStr, ErrProtocolError)
|
|
}
|
|
|
|
var read int
|
|
|
|
for read < sz {
|
|
line, err := r.ReadString('\n')
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
read += len(line)
|
|
|
|
if len(line) == 1 || line[0] == '#' {
|
|
continue
|
|
}
|
|
|
|
parts := strings.SplitN(line, ":", 2)
|
|
|
|
name := string(parts[0])
|
|
|
|
metric, ok := Tracking[name]
|
|
if !ok {
|
|
continue
|
|
}
|
|
|
|
val := strings.TrimSpace(parts[1])
|
|
|
|
ival, err := strconv.ParseUint(val, 10, 64)
|
|
if err == nil {
|
|
acc.Add(metric, ival, nil)
|
|
continue
|
|
}
|
|
|
|
fval, err := strconv.ParseFloat(val, 64)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
acc.Add(metric, fval, nil)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|
|
func init() {
|
|
plugins.Add("redis", func() plugins.Plugin {
|
|
return &Gatherer{}
|
|
})
|
|
}
|