add replicas info item to ReplicaSet node

Signed-off-by: Zach Hammer <[email protected]>
seatgeek · May 6, 2024 · ea085a8 · ea085a8
1 parent a2a0360
commit ea085a8
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 0 deletions.
diff --git a/controller/cache/info.go b/controller/cache/info.go
@@ -54,6 +54,8 @@ func populateNodeInfo(un *unstructured.Unstructured, res *ResourceInfo, customLa
 			populatePodInfo(un, res)
 		case kube.ServiceKind:
 			populateServiceInfo(un, res)
+		case kube.ReplicaSetKind:
+			populateReplicaSetInfo(un, res)
 		case "Node":
 			populateHostNodeInfo(un, res)
 		}
@@ -103,6 +105,14 @@ func populateServiceInfo(un *unstructured.Unstructured, res *ResourceInfo) {
 	res.NetworkingInfo = &v1alpha1.ResourceNetworkingInfo{TargetLabels: targetLabels, Ingress: ingress, ExternalURLs: urls}
 }
 
+func populateReplicaSetInfo(un *unstructured.Unstructured, res *ResourceInfo) {
+	// due to https://github.com/kubernetes-sigs/yaml/issues/45, replicas is parsed as a float
+	replicas, ok, err := unstructured.NestedFloat64(un.Object, "spec", "replicas")
+	if err == nil && ok {
+		res.Info = append(res.Info, v1alpha1.InfoItem{Name: "Replicas", Value: fmt.Sprintf("Replicas:%d", int(replicas))})
+	}
+}
+
 func getServiceName(backend map[string]interface{}, gvk schema.GroupVersionKind) (string, error) {
 	switch gvk.Group {
 	case "extensions":

diff --git a/controller/cache/info_test.go b/controller/cache/info_test.go
@@ -754,3 +754,19 @@ func TestManifestHash(t *testing.T) {
 	assert.Equal(t, expected, hash)
 	assert.Nil(t, err)
 }
+
+func TestReplicaSet(t *testing.T) {
+	replicaSet := strToUnstructured(`
+  apiVersion: v1
+  kind: ReplicaSet
+  metadata:
+    name: helm-guestbook
+  spec:
+    replicas: 10
+`)
+
+	info := &ResourceInfo{}
+	populateNodeInfo(replicaSet, info, []string{})
+
+	assert.Contains(t, info.Info, v1alpha1.InfoItem{Name: "Replicas", Value: "Replicas:10"})
+}