Add pod counter to provide index attribute for pods in a deployment R…

…eplicaSet

contents/pods-resource-model.py

@@ -37,7 +37,7 @@ def get(self, path, default=None):
 log = logging.getLogger('kubernetes-model-source')
 
 
-def nodeCollectData(pod, container, defaults, taglist, mappingList, boEmoticon):
+def nodeCollectData(pod, container, defaults, taglist, mappingList, boEmoticon, index):
     tags = []
     tags.extend(taglist.split(','))
 
@@ -105,6 +105,9 @@ def nodeCollectData(pod, container, defaults, taglist, mappingList, boEmoticon):
     mappings = []
     custom_attributes = {}
 
+    # Count pods within their owner
+    custom_attributes['index'] = index
+
     # custom mapping attributes
     if mappingList:
         log.debug('Mapping: %s', mappingList)
@@ -206,6 +209,8 @@ def main():
         label_selector = os.environ.get('RD_CONFIG_LABEL_SELECTOR')
 
     node_set = []
+    # Used to count child pods, particularly of a (possibly autoscaling) ReplicaSet.
+    parents = {}
     v1 = client.CoreV1Api()
 
     log.debug(label_selector)
@@ -244,12 +249,22 @@ def main():
                       i.metadata.name,
                       container.name)
 
+            # For scalable pods in a deployment, the ReplicaSet of a pod is the pod name with the last dash-separated
+            # token stripped off. If we have seen the ReplicaSet already, add one. Otherwise, initialize this as the
+            # first pod in the ReplicaSet.
+            parent_name = '-'.join(i.metadata.name.split('-')[0:-1])
+            if parent_name in parents:
+                parents[parent_name] += 1
+            else:
+                parents[parent_name] = 1
+
             node_data = nodeCollectData(i,
                                         container,
                                         defaults,
                                         tags,
                                         mappingList,
-                                        boEmoticon
+                                        boEmoticon,
+                                        parents[parent_name]
                                         )
 
             if running is False: