source: DCWoRMS/trunk/src/test/article/recs/plugins/scheduling/old/RecsExclusivenessDFSSP.java @ 709

Revision 709, 5.0 KB checked in by wojtekp, 12 years ago (diff)
  • Property svn:mime-type set to text/plain
Line 
1package test.article.recs.plugins.scheduling.old;
2
3import gridsim.dcworms.DCWormsTags;
4
5import java.util.ArrayList;
6import java.util.HashMap;
7import java.util.List;
8import java.util.Map;
9import java.util.Random;
10
11import schedframe.events.scheduling.SchedulingEvent;
12import schedframe.resources.ResourceStatus;
13import schedframe.resources.computing.ComputingNode;
14import schedframe.resources.computing.ComputingResource;
15import schedframe.resources.computing.Core;
16import schedframe.resources.computing.Processor;
17import schedframe.resources.units.ProcessingElements;
18import schedframe.resources.units.ResourceUnit;
19import schedframe.resources.units.ResourceUnitName;
20import schedframe.resources.units.StandardResourceUnitName;
21import schedframe.scheduling.manager.resources.ClusterResourceManager;
22import schedframe.scheduling.manager.resources.ResourceManager;
23import schedframe.scheduling.manager.tasks.JobRegistry;
24import schedframe.scheduling.plan.SchedulingPlanInterface;
25import schedframe.scheduling.plan.impl.SchedulingPlan;
26import schedframe.scheduling.plugin.grid.ModuleList;
27import schedframe.scheduling.queue.TaskQueue;
28import schedframe.scheduling.queue.TaskQueueList;
29import schedframe.scheduling.tasks.TaskInterface;
30import test.article.recs.plugins.scheduling.RecsSP;
31import test.article.recs.utils.RecsProcessorPowerInterface;
32
33public class RecsExclusivenessDFSSP extends RecsSP {
34
35        private Random rand = new Random(5);
36       
37        public SchedulingPlanInterface<?> schedule(SchedulingEvent event, TaskQueueList queues, JobRegistry jobRegistry,
38                        ResourceManager resManager, ModuleList modules) {
39
40                ClusterResourceManager resourceManager = (ClusterResourceManager) resManager;
41                SchedulingPlan plan = new SchedulingPlan();
42                // our tasks are placed only in first queue (see
43                // BaseLocalSchedulingPlugin.placeJobsInQueues() method)
44                TaskQueue q = queues.get(0);
45                // choose the events types to serve.
46                // Different actions for different events are possible.
47                switch (event.getType()) {
48               
49                case START_TASK_EXECUTION:
50                case TASK_FINISHED:
51                        // check all tasks in queue
52                        for (int i = 0; i < q.size(); i++) {
53                                TaskInterface<?> task = q.get(i);
54                                initApplicationType(task);
55                               
56                                // if status of the tasks in READY
57                                if (task.getStatus() == DCWormsTags.READY) {
58
59                                        Map<ResourceUnitName, ResourceUnit> choosenResources = chooseResourcesForExecution(resourceManager, task);
60                                        if (choosenResources  != null) {
61                                                addToSchedulingPlan(plan, task, choosenResources);
62                                        }
63                                }
64                        }
65                        adjustFrequency(resourceManager.getProcessors());
66                }
67                return plan;
68        }
69       
70        private Map<ResourceUnitName, ResourceUnit> chooseResourcesForExecution(
71                        ClusterResourceManager resourceManager, TaskInterface<?> task) {
72
73                Map<ResourceUnitName, ResourceUnit> map = new HashMap<ResourceUnitName, ResourceUnit>();
74               
75                List<ComputingNode> nodes = resourceManager.getComputingNodes();
76                List<ComputingNode> avNodes = filterNodes(nodes, task);
77                if(avNodes.size() == 0)
78                        return null;
79                ComputingNode node = randomNode(avNodes);
80
81                int cpuRequest;
82                try {
83                        cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue();
84                } catch (NoSuchFieldException e) {
85                        cpuRequest = 0;
86                }
87
88                if (cpuRequest != 0) {
89
90                        List<Core> cores = node.getProcessors().get(0).getCores();
91
92                        List<ComputingResource> choosenResources = new ArrayList<ComputingResource>();                         
93                        for (int i = 0; i < cores.size() && cpuRequest > 0; i++) {
94                                if (cores.get(i).getStatus() == ResourceStatus.FREE) {
95                                        choosenResources.add(cores.get(i));
96                                        cpuRequest--;
97                                }
98                        }
99                        if (cpuRequest > 0) {
100                                return null;
101                        }
102                        ProcessingElements pe = new ProcessingElements();
103                        pe.addAll(choosenResources);
104                        map.put(StandardResourceUnitName.PE, pe);
105                        return map;
106                }
107                return null;
108        }
109       
110        private List<ComputingNode> filterNodes(List<ComputingNode> nodes, TaskInterface<?> task){
111                List<ComputingNode> filteredNodes = new ArrayList<ComputingNode>();
112                for (ComputingNode node : nodes) {
113                        int cpuRequest;
114                        try {
115                                cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue();
116                        } catch (NoSuchFieldException e) {
117                                cpuRequest = 0;
118                        }
119
120                        if (cpuRequest != 0) {
121
122                                List<Core> cores = node.getProcessors().get(0).getCores();
123                                if (cores.size() < cpuRequest) {
124                                        continue;
125                                }
126
127                                int freeCores = 0;
128                                for(Core core: cores){
129                                        if(core.getStatus() == ResourceStatus.FREE)
130                                                freeCores++;
131                                }
132                               
133                                if(freeCores != cores.size())
134                                        continue;
135                               
136                                filteredNodes.add(node);
137                        }
138                }
139               
140                return filteredNodes;
141        }
142       
143        private ComputingNode randomNode(List<ComputingNode> nodes){
144                return nodes.get(rand.nextInt(nodes.size()));
145        }
146       
147       
148        private void adjustFrequency(List<Processor> processors){
149
150                for(Processor cpu: processors){
151                        RecsProcessorPowerInterface rppi = (RecsProcessorPowerInterface) cpu.getPowerInterface();
152                        int freeCores = 0;
153
154                        for(Core core: cpu.getCores()){
155                                if(core.getStatus() == ResourceStatus.FREE)
156                                        freeCores++;
157                        }
158                        if(freeCores == cpu.getCores().size())
159                                rppi.setPState(rppi.getHighestPState().getName());
160                        else
161                                rppi.setPState(rppi.getLowestPState().getName());
162
163                }
164        }
165
166}
Note: See TracBrowser for help on using the repository browser.