Add support for cgclassify to Cgroup class.

Signed-off-by: Tom Hromatka <tom.hroma...@oracle.com>
---
 ftests/cgroup.py | 21 +++++++++++++++++++++
 1 file changed, 21 insertions(+)

diff --git a/ftests/cgroup.py b/ftests/cgroup.py
index 6f7042d61efc..ec5c6821c9d1 100644
--- a/ftests/cgroup.py
+++ b/ftests/cgroup.py
@@ -194,3 +194,24 @@ class Cgroup(Enum):
                                 return Cgroup.CGROUP_V2
 
         return Cgroup.CGROUP_UNK
+
+    @staticmethod
+    def classify(config, controller, cgname, pid_list, sticky=False,
+                 cancel_sticky=False, in_container=True):
+        cmd = list()
+        cmd.append(Cgroup.build_cmd_path(in_container, 'cgclassify'))
+        cmd.append('-g')
+        cmd.append('{}:{}'.format(controller, cgname))
+
+        if isinstance(pid_list, str):
+            cmd.append(pid_list)
+        elif isinstance(pid_list, int):
+            cmd.append(str(pid_list))
+        elif isinstance(pid_list, list):
+            for pid in pid_list:
+                cmd.append(pid)
+
+        if in_container:
+            config.container.run(cmd)
+        else:
+            Run.run(cmd)
-- 
2.25.4



_______________________________________________
Libcg-devel mailing list
Libcg-devel@lists.sourceforge.net
https://lists.sourceforge.net/lists/listinfo/libcg-devel

Reply via email to