feat: kind-filter

cmd/main.go

@@ -22,6 +22,7 @@ func main() {
 	interval := flag.String("interval", "5s", "Interval between status checks")
 	retryLimit := flag.Int64("retry-limit", 6, "Retry limit for deployments and statefulsets, default 6, -1 to disable")
 	pendingDeadLineSeconds := flag.Int("pending-deadline", 180, "Pending deadLine in seconds, default 180, -1 to disable")
+	kindFilter := flag.String("kind-filter", "", "Kind filter, one of deployment, job or statefulset")
 
 	ignoreSecretNotFound := flag.Bool("ignore-secret-not-found", false, "Ignore secret not found error")
 
@@ -39,6 +40,7 @@ func main() {
 		IgnoreSecretNotFound:   *ignoreSecretNotFound,
 		RetryLimit:             int32(*retryLimit),
 		PendingDeadLineSeconds: *pendingDeadLineSeconds,
+		KindFilter:             config.KindFilter(*kindFilter),
 	}
 
 	clientset := makeClientset(*kubeconfig, *kubecontext)

pkg/config/options.go

@@ -1,7 +1,17 @@
 package config
 
+type KindFilter string
+
+const (
+	NoKindFilter          KindFilter = ""
+	DeploymentKindFilter  KindFilter = "deployment"
+	JobKindFilter         KindFilter = "job"
+	StatefulsetKindFilter KindFilter = "statefulset"
+)
+
 type Options struct {
 	IgnoreSecretNotFound   bool
 	RetryLimit             int32
 	PendingDeadLineSeconds int
+	KindFilter             KindFilter
 }

pkg/status/all.go

@@ -3,22 +3,35 @@ package status
 import (
 	"github.com/SocialGouv/rollout-status/pkg/client"
 	"github.com/SocialGouv/rollout-status/pkg/config"
+	v1 "k8s.io/api/apps/v1"
+	batchv1 "k8s.io/api/batch/v1"
 )
 
 func TestRollout(wrapper client.Kubernetes, namespace, selector string, options *config.Options) RolloutStatus {
-	deployments, err := wrapper.ListAppsV1Deployments(namespace, selector)
-	if err != nil {
-		return RolloutFatal(err)
+	var err error
+
+	var deployments *v1.DeploymentList
+	if options.KindFilter == config.NoKindFilter || options.KindFilter == config.DeploymentKindFilter {
+		deployments, err = wrapper.ListAppsV1Deployments(namespace, selector)
+		if err != nil {
+			return RolloutFatal(err)
+		}
 	}
 
-	statefulsets, err := wrapper.ListAppsV1StatefulSets(namespace, selector)
-	if err != nil {
-		return RolloutFatal(err)
+	var statefulsets *v1.StatefulSetList
+	if options.KindFilter == config.NoKindFilter || options.KindFilter == config.StatefulsetKindFilter {
+		statefulsets, err = wrapper.ListAppsV1StatefulSets(namespace, selector)
+		if err != nil {
+			return RolloutFatal(err)
+		}
 	}
 
-	jobs, err := wrapper.ListBatchV1Jobs(namespace, selector)
-	if err != nil {
-		return RolloutFatal(err)
+	var jobs *batchv1.JobList
+	if options.KindFilter == config.NoKindFilter || options.KindFilter == config.JobKindFilter {
+		jobs, err = wrapper.ListBatchV1Jobs(namespace, selector)
+		if err != nil {
+			return RolloutFatal(err)
+		}
 	}
 
 	if (deployments == nil || len(deployments.Items) == 0) && (statefulsets == nil || len(statefulsets.Items) == 0) && (jobs == nil || len(jobs.Items) == 0) {