Feat: Add option to filter processes by both containers and nodes (#27)

* Feat: Add option to filter processes by both containers and nodes - Part of #2110 in testsuite Signed-off-by: svteb <[email protected]> Co-authored-by: Konstantin <[email protected]> --------- Signed-off-by: svteb <[email protected]> Co-authored-by: Martin Matyas <[email protected]> Co-authored-by: Konstantin <[email protected]>
cnf-testsuite · Aug 2, 2024 · 0b8f04b · 0b8f04b
1 parent 71f9497
commit 0b8f04b
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/cluster_tools.cr b/cluster_tools.cr
@@ -163,7 +163,7 @@ module ClusterTools
     pid 
   end
   #each_container_by_resource(resource, namespace) do | container_id, container_pid_on_node, node, container_proctree_statuses, container_status|
-  def self.all_containers_by_resource?(resource, namespace, &block) 
+  def self.all_containers_by_resource?(resource, namespace, only_container_pids : Bool = false, &block) 
 		kind = resource["kind"].downcase
 		case kind
 		when  "deployment","statefulset","pod","replicaset", "daemonset"
@@ -210,8 +210,12 @@ module ClusterTools
 
 						node_name = node.dig("metadata", "name").as_s
 						Log.info { "node name : #{node_name}" }
-						pids = KernelIntrospection::K8s::Node.pids(node)
-						Log.info { "proctree_by_pid pids: #{pids}" }
+						if only_container_pids 
+                                                  pids = KernelIntrospection::K8s::Node.pids_by_container(container_id, node)
+                                                else
+                                                  pids = KernelIntrospection::K8s::Node.pids(node)
+                                                end
+						Log.info { "parsed pids: #{pids}" }
 						proc_statuses = KernelIntrospection::K8s::Node.all_statuses_by_pids(pids, node)
 
 						container_proctree_statuses = KernelIntrospection::K8s::Node.proctree_by_pid(container_pid_on_node, node, proc_statuses)