-
Notifications
You must be signed in to change notification settings - Fork 1
/
Framework.cpp
84 lines (68 loc) · 2.01 KB
/
Framework.cpp
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
//
// Created by Hans Henrik Sande on 11/04/2020.
//
#include <iostream>
#include "Framework.h"
Resource sumOfTasks(std::vector<Task> req){
Resource r;
int tempDisk = 0;
int tempCpu = 0;
int tempMem = 0;
for(Task t : req){
tempDisk += t.required.getDisk();
tempCpu += t.required.getCpus();
tempMem += t.required.getMem();
}
r.setCpus(tempCpu);
r.setMem(tempMem);
r.setDisk(tempDisk);
return r;
}
bool Framework::enoughResourceForTask(Resource res, Task task) {
return res.getCpus() >= task.required.getCpus() && res.getMem() >= task.required.getMem();
}
Framework::Framework(int id, std::vector<Task> req, ApplicationType applicationType) {
this->id = id;
this->required = sumOfTasks(req);
this->numberOfTasks = req;
this->applicationType = applicationType;
this->finished = false;
}
Framework::Framework(){
}
bool Framework::sendOffer(Resource res, std::vector<std::pair<int, Resource>> bundles) {
bool temp = false;
for(int i = 0; i < this->numberOfTasks.size(); i++){
if(!numberOfTasks[i].finished){
if(enoughResourceForTask(res, numberOfTasks[i])){
//std::cout << "TASK RUNNING" << std::endl;
temp = true;
}
}
}
return temp;
}
int Framework::getTasksInQueue() {
int temp = 0;
for(int i = 0; i < numberOfTasks.size(); i++){
if(!numberOfTasks[i].finished){
temp++;
}
}
return temp;
}
void Framework::setSlaves(std::vector<std::pair<int, Resource>> slaves) {
//std::cout << "SLAVES ADDED TO FRAMEWORK " << id << " " << std::endl;
this->slavesServingFramework = slaves;
}
void Framework::run() {
for(int i = 0; i < numberOfTasks.size(); i++){
if(this->numberOfTasks[i].finished == true){
this->numberOfTasks.erase(numberOfTasks.begin()+i);
}
}
if(numberOfTasks.size() == 0){
//std::cout << "Framework finished" << std::endl;
this->finished = true;
}
}