Update ZookeeperCluster resource status as needed (#37)

* Update the staus size field Signed-off-by: Olivier Bourgeois <[email protected]> * Update the current size of cluster based on replica set status Signed-off-by: Olivier Bourgeois <[email protected]>
pravega · Nov 8, 2018 · b4c1f5d · b4c1f5d
1 parent 73660ba
commit b4c1f5d
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/pkg/zk/sync.go b/pkg/zk/sync.go
@@ -35,12 +35,19 @@ func syncClusterSize(zk *v1beta1.ZookeeperCluster) (err error) {
 		return fmt.Errorf("failed to get stateful-set (%s): %v", sts.Name, err)
 	}
 
+	zk.Status.Size = int(sts.Status.Replicas)
+	err = sdk.Update(zk)
+	if err != nil {
+		return fmt.Errorf("failed to update project: %v", err)
+	}
+
 	if *sts.Spec.Replicas != zk.Spec.Size {
 		sts.Spec.Replicas = &(zk.Spec.Size)
 		err = sdk.Update(sts)
 		if err != nil {
 			return fmt.Errorf("failed to update size of stateful-set (%s): %v", sts.Name, err)
 		}
 	}
+
 	return nil
 }