Add new pdb to ClusterStatus

zalando · Jan 3, 2025 · ac0634d · ac0634d
1 parent 7ee70e3
commit ac0634d
Show file tree

Hide file tree

Showing 3 changed files with 26 additions and 19 deletions.
diff --git a/pkg/cluster/cluster.go b/pkg/cluster/cluster.go
@@ -1706,16 +1706,17 @@ func (c *Cluster) GetCurrentProcess() Process {
 // GetStatus provides status of the cluster
 func (c *Cluster) GetStatus() *ClusterStatus {
 	status := &ClusterStatus{
-		Cluster:                    c.Name,
-		Namespace:                  c.Namespace,
-		Team:                       c.Spec.TeamID,
-		Status:                     c.Status,
-		Spec:                       c.Spec,
-		MasterService:              c.GetServiceMaster(),
-		ReplicaService:             c.GetServiceReplica(),
-		StatefulSet:                c.GetStatefulSet(),
-		GeneralPodDisruptionBudget: c.GetGeneralPodDisruptionBudget(),
-		CurrentProcess:             c.GetCurrentProcess(),
+		Cluster:                       c.Name,
+		Namespace:                     c.Namespace,
+		Team:                          c.Spec.TeamID,
+		Status:                        c.Status,
+		Spec:                          c.Spec,
+		MasterService:                 c.GetServiceMaster(),
+		ReplicaService:                c.GetServiceReplica(),
+		StatefulSet:                   c.GetStatefulSet(),
+		GeneralPodDisruptionBudget:    c.GetGeneralPodDisruptionBudget(),
+		CriticalOpPodDisruptionBudget: c.GetCriticalOpPodDisruptionBudget(),
+		CurrentProcess:                c.GetCurrentProcess(),
 
 		Error: fmt.Errorf("error: %s", c.Error),
 	}

diff --git a/pkg/cluster/resources.go b/pkg/cluster/resources.go
@@ -831,3 +831,8 @@ func (c *Cluster) GetStatefulSet() *appsv1.StatefulSet {
 func (c *Cluster) GetGeneralPodDisruptionBudget() *policyv1.PodDisruptionBudget {
 	return c.GeneralPodDisruptionBudget
 }
+
+// GetPodDisruptionBudget returns cluster's kubernetes PodDisruptionBudget for critical operations
+func (c *Cluster) GetCriticalOpPodDisruptionBudget() *policyv1.PodDisruptionBudget {
+	return c.CriticalOpPodDisruptionBudget
+}
diff --git a/pkg/cluster/types.go b/pkg/cluster/types.go
@@ -58,15 +58,16 @@ type WorkerStatus struct {
 
 // ClusterStatus describes status of the cluster
 type ClusterStatus struct {
-	Team                       string
-	Cluster                    string
-	Namespace                  string
-	MasterService              *v1.Service
-	ReplicaService             *v1.Service
-	MasterEndpoint             *v1.Endpoints
-	ReplicaEndpoint            *v1.Endpoints
-	StatefulSet                *appsv1.StatefulSet
-	GeneralPodDisruptionBudget *policyv1.PodDisruptionBudget
+	Team                          string
+	Cluster                       string
+	Namespace                     string
+	MasterService                 *v1.Service
+	ReplicaService                *v1.Service
+	MasterEndpoint                *v1.Endpoints
+	ReplicaEndpoint               *v1.Endpoints
+	StatefulSet                   *appsv1.StatefulSet
+	GeneralPodDisruptionBudget    *policyv1.PodDisruptionBudget
+	CriticalOpPodDisruptionBudget *policyv1.PodDisruptionBudget
 
 	CurrentProcess Process
 	Worker         uint32