X-Git-Url: http://git.onelab.eu/?a=blobdiff_plain;f=cgroups.py;h=a1c38b455dbb5e2ad7e58a7e56586c0ba2b925c9;hb=570d234b7d7b40416b296de71225f28c826ee991;hp=7ccbf0bdb7a005b8dcb723a3e191f4b73cce721e;hpb=430195a2b1382182d28dbcd16b21864ed758f52d;p=nodemanager.git diff --git a/cgroups.py b/cgroups.py index 7ccbf0b..a1c38b4 100644 --- a/cgroups.py +++ b/cgroups.py @@ -3,52 +3,12 @@ # # Xavi Leon -import os +import os, os.path import pyinotify import logger # Base dir for libvirt BASE_DIR = '/sys/fs/cgroup' -SUB_SYSTEMS = ['blkio', 'freezer', 'devices', 'memory', 'cpu,cpuacct', 'cpuset'] -VIRT_TECHS = ['lxc'] - -# Global cgroup mapping. -CGROUPS = dict() - -class CgroupWatch(pyinotify.ProcessEvent): - - def process_IN_CREATE(self, event): - path = os.path.join(event.path, event.name) - CGROUPS[event.name] = path - logger.verbose("Cgroup Notify: Created cgroup %s on %s" % \ - (event.name, event.path)) - - def process_IN_DELETE(self, event): - try: - del CGROUPS[event.name] - except: - logger.verbose("Cgroup Notify: Cgroup %s does not exist, continuing..."%event.name) - logger.verbose("Cgroup Notify: Deleted cgroup %s on %s" % \ - (event.name, event.path)) - - -#logger.verbose("Cgroups: Recognizing already existing cgroups...") -#for virt in VIRT_TECHS: -# filenames = os.listdir(os.path.join(BASE_DIR, virt)) -# for filename in filenames: -# path = os.path.join(BASE_DIR, virt, filename) -# if os.path.isdir(path): -# CGROUPS[filename] = path - -#logger.verbose("Cgroups: Initializing watchers...") -#wm = pyinotify.WatchManager() -#notifier = pyinotify.ThreadedNotifier(wm, CgroupWatch()) -#for virt in VIRT_TECHS: -# wdd = wm.add_watch(os.path.join(BASE_DIR, virt), -# pyinotify.IN_DELETE | pyinotify.IN_CREATE, -# rec=False) -#notifier.daemon = True -#notifier.start() def get_cgroup_paths(subsystem="cpuset"): subsystem_bases = [ @@ -98,12 +58,14 @@ def write(name, key, value, subsystem="cpuset"): base_path = get_cgroup_path(name, subsystem) with open(os.path.join(base_path, key), 'w') as f: print >>f, value + logger.verbose("cgroups.write: overwrote {}".format(base_path)) 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, subsystem) with open(os.path.join(base_path, key), 'a') as f: print >>f, value + logger.verbose("cgroups.append: appended {}".format(base_path)) if __name__ == '__main__':