vstf/vstf/agent/equalizer/equalizer.py

   1 #!/usr/bin/env python
   2 # coding=utf-8
   3 import os
   4 import re
   5 import subprocess
   6 import logging
   7
   8 log = logging.getLogger(__name__)
   9
  10
  11 def run_cmd(cmd, shell=True):
  12     try:
  13         ret = subprocess.check_output(cmd, shell=shell)
  14     except subprocess.CalledProcessError as e:
  15         raise e
  16     return ret
  17
  18
  19 class Resource(object):
  20     def __init__(self):
  21         super(Resource, self).__init__()
  22         self.sysfs = "/sys/devices/system/node"
  23         self.mapping = {}
  24         for node in self._init_numa():
  25             self.mapping[node] = {}
  26
  27             process_mapping = self._get_process_mapping(node)
  28             for process_index in xrange(0, len(bin(process_mapping)) - 2):
  29                 if process_mapping & 1 << process_index != 0:
  30                     core = self._get_core_id(node, process_index)
  31                     if not self.mapping[node].has_key(core):
  32                         self.mapping[node][core] = []
  33                     self.mapping[node][core].append(process_index)
  34
  35     def _get_process_mapping(self, numa_node):
  36         ret = run_cmd("cat " + self.sysfs + '/' + numa_node + '/cpumap').replace(',', '').lstrip('0')
  37         return int(ret, 16)
  38
  39     def _get_core_id(self, numa_node, process_index):
  40         cmd = "cat " + self.sysfs + '/' + numa_node + '/cpu' + str(process_index) + '/topology/core_id'
  41         return run_cmd(cmd).strip('\n')
  42
  43     def _init_numa(self):
  44         """the node name is node0, node1......"""
  45         try:
  46             node_list = os.listdir(self.sysfs)
  47         except Exception as e:
  48             raise e
  49         ret = []
  50         partten = re.compile("^node[0-9]{,}$")
  51         for node in node_list:
  52             if partten.match(node) is None:
  53                 continue
  54             ret.append(node)
  55         return ret
  56
  57
  58 class Equalizer(Resource):
  59     def __init__(self):
  60         super(Equalizer, self).__init__()
  61
  62     def topology(self):
  63         print self.mapping
  64
  65
  66 e = Equalizer()
  67 e.topology()