-
Notifications
You must be signed in to change notification settings - Fork 496
Expand file tree
/
Copy pathSimpleResourceManager.cxx
More file actions
78 lines (72 loc) · 2.54 KB
/
SimpleResourceManager.cxx
File metadata and controls
78 lines (72 loc) · 2.54 KB
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
// Copyright CERN and copyright holders of ALICE O2. This software is
// distributed under the terms of the GNU General Public License v3 (GPL
// Version 3), copied verbatim in the file "COPYING".
//
// See http://alice-o2.web.cern.ch/license for full licensing information.
//
// In applying this license CERN does not waive the privileges and immunities
// granted to it by virtue of its status as an Intergovernmental Organization
// or submit itself to any jurisdiction.
#include "SimpleResourceManager.h"
#include "Framework/ComputingResource.h"
#include <fmt/format.h>
#include <exception>
#include <stdexcept>
namespace o2::framework
{
/// The simplest implementation of this allocates mMaxPorts ports starting from
/// the mInitialPort. For now we still consider everything running on a single
/// machine.
std::vector<ComputingOffer> SimpleResourceManager::getAvailableOffers()
{
std::vector<ComputingOffer> result;
for (auto& resource : mResources) {
if (resource.cpu < 0.01) {
continue;
}
if (resource.memory < 0.01) {
continue;
}
if (resource.usedPorts == (resource.lastPort - resource.startPort + 1)) {
continue;
}
ComputingOffer offer;
offer.cpu = resource.cpu;
offer.memory = resource.memory;
offer.hostname = resource.hostname;
offer.startPort = resource.startPort + resource.usedPorts;
offer.rangeSize = (resource.lastPort - resource.startPort) - resource.usedPorts;
result.push_back(offer);
}
return result;
}
void SimpleResourceManager::notifyAcceptedOffer(ComputingOffer const& offer)
{
bool resourceFound = false;
for (auto& resource : mResources) {
if (resource.hostname != offer.hostname) {
continue;
}
if (resource.startPort > offer.startPort) {
continue;
}
if (resource.lastPort < offer.startPort + offer.rangeSize) {
continue;
}
resourceFound = true;
resource.cpu -= offer.cpu;
resource.memory -= offer.memory;
resource.usedPorts += offer.rangeSize;
break;
}
if (resourceFound == false) {
std::string resources = "Available resources:\n";
for (auto& resource : mResources) {
resources += fmt::format("- ({}, {}, {}, {})\n", resource.hostname.c_str(), resource.cpu, resource.memory, resource.startPort);
}
throw std::runtime_error(fmt::format("Could not match offer (host:{}, cpu:{}, mem:{}, ports:{}) to original resource.\n",
offer.hostname, offer.cpu, offer.memory, offer.rangeSize) +
resources);
}
}
} // namespace o2::framework