Add filter to status command

Signed-off-by: Valentin Pichard <[email protected]>
ylecuyer · Jan 10, 2019 · 1b3a4dc · 1b3a4dc
1 parent 8366e67
commit 1b3a4dc
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 0 deletions.
diff --git a/cmd/status.go b/cmd/status.go
@@ -8,6 +8,7 @@ import (
 
 func init() {
 	RootCmd.AddCommand(statusCmd)
+	setLocalFlags()
 }
 
 // statusCmd represents the status command
@@ -26,3 +27,7 @@ var statusCmd = &cobra.Command{
 		return nil
 	},
 }
+
+func setLocalFlags() {
+	statusCmd.Flags().StringVarP(&helper.Filter, "filter", "f", "", "Show only applications who match")
+}
diff --git a/helper/client.go b/helper/client.go
@@ -13,4 +13,5 @@ type Configuration struct {
 
 var (
 	CfgFile Configuration
+	Filter  string
 )
diff --git a/helper/status.go b/helper/status.go
@@ -5,6 +5,7 @@ import (
 	"encoding/json"
 	"fmt"
 	"os/exec"
+	"strings"
 	"time"
 
 	. "github.com/logrusorgru/aurora"
@@ -116,6 +117,10 @@ func printApplicationGroups() error {
 
 	line := 0
 	for appname, key := range CfgFile.Applications {
+		if Filter != "" && !strings.Contains(appname, Filter) {
+			continue
+		}
+
 		ps, err := readPumaStats(key)
 		if err != nil {
 			return err
-Original file line number
+Diff line change
@@ Expand Up / @@ -13,4 +13,5 @@ type Configuration struct { @@
     var (
     	CfgFile Configuration
+    	Filter  string
     )