Changeset 1412 for DCWoRMS/branches

Timestamp:

07/02/14 14:07:24 (11 years ago)

Author:

wojtekp

Message:

 

File:

• DCWoRMS/branches/coolemall/src/example/localplugin/coolemall/computebox2/CB2_FCFS_LoadBalancing_SP.java (modified) (3 diffs)

DCWoRMS/branches/coolemall/src/example/localplugin/coolemall/computebox2/CB2_FCFS_LoadBalancing_SP.java

@@ -1 +1 @@
 package example.localplugin.coolemall.computebox2;
 
+import java.util.ArrayList;
 import java.util.List;
 
 import schedframe.events.scheduling.SchedulingEvent;
+import schedframe.resources.ResourceStatus;
+import schedframe.resources.computing.Core;
 import schedframe.resources.computing.Node;
 import schedframe.scheduling.manager.resources.ClusterResourceManager;
@@ -43 +46 @@
                                 if (task.getStatus() == DCWormsTags.READY) {
 
-                                        String nodeName = chooseProvider(resourceManager);
+                                        String nodeName = chooseProvider(resourceManager, task);
                                         if (nodeName != null) {
                                                 addToSchedulingPlan(plan, task, nodeName);
@@ -54 +57 @@
         }
 
-        private String chooseProvider(ClusterResourceManager resourceManager) {
+        private String chooseProvider(ClusterResourceManager resourceManager, TaskInterface<?> task) {
                 List<Node> nodes = resourceManager.getNodes();
-                int nodeIdx = findLeastLoadedResourceIdx(nodes);
-                return nodes.get(nodeIdx).getFullName();
+                List<Node> avNodes = filterNodes(nodes, task);
+                int nodeIdx = findLeastLoadedResourceIdx(avNodes);
+                return nodeIdx != -1 ? avNodes.get(nodeIdx).getFullName() : null;
+        }
+        
+        private List<Node> filterNodes(List<Node> nodes, TaskInterface<?> task){
+                List<Node> filteredNodes = new ArrayList<Node>();
+                int cpuRequest;
+                try {
+                        cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue();
+                } catch (NoSuchFieldException e) {
+                        cpuRequest = 0;
+                }
+                for (Node node : nodes) {
+
+                        if (cpuRequest != 0) {
+
+                                List<Core> cores = node.getCores();
+                                if (cores.size() < cpuRequest) {
+                                        if(cores.size() == 0){
+                                                if(node.getProcessors().size() < cpuRequest)
+                                                        continue;
+                                        }
+                                }
+
+                                int freeCores = 0;
+                                for(Core core: cores){
+                                        if(core.getStatus() == ResourceStatus.FREE)
+                                                freeCores++;
+                                }
+                                
+                                if(freeCores < cpuRequest)
+                                        continue;
+                                
+                                filteredNodes.add(node);
+                        }
+                }
+                
+                return filteredNodes;
         }