1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
|
#!/usr/bin/env python3
# -*- coding: utf-8 -*-
#
# Copyright (c) 2017 Cisco and/or its affiliates.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at:
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
import re
from itertools import cycle
from netmodel.model.type import BaseType
from vicn.core.resource import Resource
from vicn.core.attribute import Attribute, Multiplicity
from vicn.core.task import BashTask
from vicn.resource.node import Node
PATTERN_LSCPU_NUMA = 'NUMA node[0-9]+ CPU\(s\)'
CMD_LSCPU = 'lscpu'
class CycleType(BaseType, cycle):
"""
Type: CycleType
"""
pass
#------------------------------------------------------------------------------
def parse_lscpu_line(line):
#Format: NUMA node0 CPU(s): 0-17,36-53
line = line.split(':')[1]
#line = 0-17,36,53
def limits_to_list(string):
limits = string.split('-')
lower_limit = int(limits[0])
#Removes core 0 as it is used the most often by the kernl
if lower_limit is 0 : lower_limit = 1
return cycle(range(lower_limit, int(limits[1])))
return cycle(map(limits_to_list, line.split(',')))
def parse_lscpu_rv(rv):
ret = []
for line in rv.stdout.splitlines():
if re.search(PATTERN_LSCPU_NUMA, line):
ret.append(parse_lscpu_line(line))
return ret
#------------------------------------------------------------------------------
class NumaManager(Resource):
"""
Resource: NumaManager
"""
node = Attribute(Node,
mandatory = True,
multiplicity = Multiplicity.OneToOne,
reverse_auto = True,
reverse_name = 'numa_mgr')
numa_repartitor = Attribute(CycleType,
description = 'Tool to separate cores/CPUs/sockets',
multiplicity = Multiplicity.OneToMany,
ro = True)
#--------------------------------------------------------------------------
# Resource lifecycle
#--------------------------------------------------------------------------
__create__ = None
__delete__ = None
#--------------------------------------------------------------------------
# Constructor and Accessors
#--------------------------------------------------------------------------
def __init__(self, *args, **kwargs):
super().__init__(*args, **kwargs)
self.current_numa_node = 0
#--------------------------------------------------------------------------
# Attributes
#--------------------------------------------------------------------------
def _get_numa_repartitor(self):
return BashTask(self.node, CMD_LSCPU, parse=parse_lscpu_rv)
#--------------------------------------------------------------------------
# Public API
#--------------------------------------------------------------------------
def get_numa_core(self, numa_node=None):
if numa_node is None:
numa_node = self.current_numa_node
self.current_numa_node = (self.current_numa_node+1) % \
len(self.numa_repartitor)
numa_list = self.numa_repartitor[numa_node]
socket = next(numa_list)
return numa_node, next(socket)
def get_number_of_numa(self):
return len(self.numa_repartitor)
|
