package redis import ( "bufio" "errors" "fmt" "net" "strconv" "strings" "sync" "github.com/influxdb/tivan/plugins" ) type Redis struct { Disabled bool Address string Servers []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") // Reads stats from all configured servers accumulates stats. // Returns one of the errors encountered while gather stats (if any). func (g *Redis) Gather(acc plugins.Accumulator) error { if g.Disabled { return nil } var servers []string if g.Address != "" { servers = append(servers, g.Address) } servers = append(servers, g.Servers...) if len(servers) == 0 { g.gatherServer(":6379", acc) return nil } var wg sync.WaitGroup var outerr error for _, serv := range servers { wg.Add(1) go func(serv string) { defer wg.Done() outerr = g.gatherServer(serv, acc) }(serv) } wg.Wait() return outerr } func (g *Redis) gatherServer(addr string, acc plugins.Accumulator) error { if g.c == nil { c, err := net.Dial("tcp", addr) 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 &Redis{} }) }