From 37fb58922dd8b6429ccb4a2b94d4ca907ee83232 Mon Sep 17 00:00:00 2001 From: smbaker Date: Tue, 18 Jun 2013 11:37:41 -0700 Subject: [PATCH] support for writing cgroups in subsystems other than cpuset --- cgroups.py | 16 ++++++++-------- 1 file changed, 8 insertions(+), 8 deletions(-) diff --git a/cgroups.py b/cgroups.py index ebd4b56..67b2040 100644 --- a/cgroups.py +++ b/cgroups.py @@ -50,16 +50,16 @@ class CgroupWatch(pyinotify.ProcessEvent): #notifier.daemon = True #notifier.start() -def get_cgroup_paths(): - cpusetBase = os.path.join(BASE_DIR, 'cpuset', 'libvirt', 'lxc') +def get_cgroup_paths(subsystem="cpuset"): + cpusetBase = os.path.join(BASE_DIR, subsystem, 'libvirt', 'lxc') return filter(os.path.isdir, map(lambda f: os.path.join(cpusetBase, f), os.listdir(cpusetBase))) -def get_cgroup_path(name): +def get_cgroup_path(name, subsystem="cpuset"): """ Returns the base path for the cgroup with a specific name or None.""" return reduce(lambda a, b: b if os.path.basename(b) == name else a, - get_cgroup_paths(), None) + get_cgroup_paths(subsystem), None) def get_base_path(): return BASE_DIR @@ -68,14 +68,14 @@ def get_cgroups(): """ Returns the list of cgroups active at this moment on the node """ return map(os.path.basename, get_cgroup_paths()) -def write(name, key, value): +def write(name, key, value, subsystem="cpuset"): """ Writes a value to the file key with the cgroup with name """ - base_path = get_cgroup_path(name) + base_path = get_cgroup_path(name, subsystem) with open(os.path.join(base_path, key), 'w') as f: print >>f, value -def append(name, key, value): +def append(name, key, value, subsystem="cpuset"): """ Appends a value to the file key with the cgroup with name """ - base_path = get_cgroup_path(name) + base_path = get_cgroup_path(name, subsystem) with open(os.path.join(base_path, key), 'a') as f: print >>f, value -- 2.43.0