blob: 70e611e52245c4fb47aa1e85952e03a8675c984c (
plain)
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
|
#
# -------------------------------------------------------------------------
# Copyright (c) 2019 AT&T Intellectual Property
#
# 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.
#
# -------------------------------------------------------------------------
#
#!/usr/bin/env python3
from valet.engine.app_manager.group import LEVEL
class AvailResources(object):
"""Container to keep hosting resources and candidate resources
of each level (host or rack) for search.
"""
def __init__(self, _level):
self.level = _level
self.avail_hosts = {}
self.candidates = {}
def set_next_level(self):
"""Get the next level to search."""
current_level_index = LEVEL.index(self.level)
next_level_index = current_level_index - 1
if next_level_index < 0:
self.level = LEVEL[0]
else:
self.level = LEVEL[next_level_index]
def set_next_avail_hosts(self, _avail_hosts, _resource_of_level):
"""Set the next level of available hosting resources."""
for hk, h in _avail_hosts.items():
if self.level == "rack":
if h.rack_name == _resource_of_level:
self.avail_hosts[hk] = h
elif self.level == "host":
if h.host_name == _resource_of_level:
self.avail_hosts[hk] = h
def set_candidates(self):
if self.level == "rack":
for _, h in self.avail_hosts.items():
self.candidates[h.rack_name] = h
elif self.level == "host":
self.candidates = self.avail_hosts
def get_candidate(self, _resource):
candidate = None
if self.level == "rack":
for _, h in self.avail_hosts.items():
if h.rack_name == _resource.rack_name:
candidate = h
elif self.level == "host":
if _resource.host_name in self.avail_hosts.keys():
candidate = self.avail_hosts[_resource.host_name]
return candidate
|
