From ee7a22adef9ca95c2752d471f7eb701d2a7b9df5 Mon Sep 17 00:00:00 2001
From: Ken Hibino <ken.hibino7@gmail.com>
Date: Sun, 22 Dec 2019 09:27:22 -0800
Subject: [PATCH] Display redis info in stats command

---
 asynq.go                    |  1 -
 tools/asynqmon/cmd/stats.go | 31 ++++++++++++++++++++++++++++---
 2 files changed, 28 insertions(+), 4 deletions(-)

diff --git a/asynq.go b/asynq.go
index a7821cf..4064f9d 100644
--- a/asynq.go
+++ b/asynq.go
@@ -7,7 +7,6 @@ TODOs:
 - [P0] asynqmon kill <taskID>, asynqmon killall <qname>
 - [P0] Pagination for `asynqmon ls` command
 - [P0] Show elapsed time for InProgress tasks (asynqmon ls inprogress)
-- [P0] Redis Memory Usage, Connection info in stats
 - [P0] Processed, Failed count for today
 - [P0] Go docs + CONTRIBUTION.md + Github issue template + License comment
 - [P0] Redis Sentinel support
diff --git a/tools/asynqmon/cmd/stats.go b/tools/asynqmon/cmd/stats.go
index 5232620..7b2a0c5 100644
--- a/tools/asynqmon/cmd/stats.go
+++ b/tools/asynqmon/cmd/stats.go
@@ -16,6 +16,7 @@ var statsCmd = &cobra.Command{
 	Use:   "stats",
 	Short: "Shows current state of the queues",
 	Long: `Stats (aysnqmon stats) will show the number of tasks in each queue at that instant.
+It also displays basic information about the running redis instance.
 
 To monitor the queues continuously, it's recommended that you run this
 command in conjunction with the watch command.
@@ -51,15 +52,39 @@ func stats(cmd *cobra.Command, args []string) {
 		fmt.Println(err)
 		os.Exit(1)
 	}
+	info, err := r.RedisInfo()
+	if err != nil {
+		fmt.Println(err)
+		os.Exit(1)
+	}
+	fmt.Println("QUEUES")
 	printStats(stats)
 	fmt.Println()
+	fmt.Println("REDIS INFO")
+	printInfo(info)
+	fmt.Println()
 }
 
 func printStats(s *rdb.Stats) {
 	format := strings.Repeat("%v\t", 5) + "\n"
 	tw := new(tabwriter.Writer).Init(os.Stdout, 0, 8, 2, ' ', 0)
-	fmt.Fprintf(tw, format, "Enqueued", "InProgress", "Scheduled", "Retry", "Dead")
-	fmt.Fprintf(tw, format, "--------", "----------", "---------", "-----", "----")
-	fmt.Fprintf(tw, format, s.Enqueued, s.InProgress, s.Scheduled, s.Retry, s.Dead)
+	fmt.Fprintf(tw, format, "InProgress", "Enqueued", "Scheduled", "Retry", "Dead")
+	fmt.Fprintf(tw, format, "----------", "--------", "---------", "-----", "----")
+	fmt.Fprintf(tw, format, s.InProgress, s.Enqueued, s.Scheduled, s.Retry, s.Dead)
+	tw.Flush()
+}
+
+func printInfo(info map[string]string) {
+	format := strings.Repeat("%v\t", 5) + "\n"
+	tw := new(tabwriter.Writer).Init(os.Stdout, 0, 8, 2, ' ', 0)
+	fmt.Fprintf(tw, format, "Version", "Uptime", "Connections", "Memory Usage", "Peak Memory Usage")
+	fmt.Fprintf(tw, format, "-------", "------", "-----------", "------------", "-----------------")
+	fmt.Fprintf(tw, format,
+		info["redis_version"],
+		fmt.Sprintf("%s days", info["uptime_in_days"]),
+		info["connected_clients"],
+		fmt.Sprintf("%sB", info["used_memory_human"]),
+		fmt.Sprintf("%sB", info["used_memory_peak_human"]),
+	)
 	tw.Flush()
 }