package example.localplugin; import gridsim.gssim.DCWormsTags; import java.util.ArrayList; import java.util.Collections; import java.util.Comparator; import java.util.HashMap; import java.util.List; import java.util.Map; import schedframe.events.scheduling.SchedulingEvent; import schedframe.resources.ResourceStatus; import schedframe.resources.computing.ComputingNode; import schedframe.resources.computing.ComputingResource; import schedframe.resources.computing.profiles.energy.power.StandardPowerStateName; import schedframe.resources.units.ProcessingElements; import schedframe.resources.units.ResourceUnit; import schedframe.resources.units.ResourceUnitName; import schedframe.resources.units.StandardResourceUnitName; import schedframe.scheduling.manager.resources.ClusterResourceManager; import schedframe.scheduling.manager.resources.ResourceManager; import schedframe.scheduling.manager.tasks.JobRegistry; import schedframe.scheduling.plan.SchedulingPlanInterface; import schedframe.scheduling.plan.impl.SchedulingPlan; import schedframe.scheduling.plugin.grid.ModuleList; import schedframe.scheduling.queue.TaskQueue; import schedframe.scheduling.queue.TaskQueueList; import schedframe.scheduling.tasks.TaskInterface; import schedframe.scheduling.tasks.WorkloadUnit; public class FCFSNodePowerManagementClusterLocalPlugin extends BaseLocalSchedulingPlugin { public FCFSNodePowerManagementClusterLocalPlugin () { } public SchedulingPlanInterface schedule(SchedulingEvent event, TaskQueueList queues, JobRegistry jobRegistry, ResourceManager resManager, ModuleList modules) { ClusterResourceManager resourceManager = (ClusterResourceManager) resManager; SchedulingPlan plan = new SchedulingPlan(); // chose the events types to serve. // Different actions for different events are possible. switch (event.getType()) { case START_TASK_EXECUTION: case TASK_FINISHED: // our tasks are placed only in first queue (see // BaseLocalPlugin.placeJobsInQueues() method) TaskQueue q = queues.get(0); // check all tasks in queue for (int i = 0; i < q.size(); i++) { WorkloadUnit job = q.get(i); TaskInterface task = (TaskInterface) job; // if status of the tasks in READY if (task.getStatus() == DCWormsTags.READY) { Map choosenResources = chooseResourcesForExecution(resourceManager, task); if (choosenResources != null) { addToSchedulingPlan(plan, task, choosenResources); } else { if(harnessIdleNodesToWork(task, resourceManager.getComputingNodes())) i--; } } } turnOffIdleNodes(resourceManager.getComputingNodes()); break; } return plan; } private Map chooseResourcesForExecution(ClusterResourceManager resourceManager, TaskInterface task) { List nodes = resourceManager.getComputingNodes(); nodes = findSuitableNodes(task, nodes); Collections.sort(nodes, new Comparator(){ public int compare(ComputingNode node1, ComputingNode node2){ return node1.getResourceCharacteristic().getParameters().get("category").get(0).getContent().compareTo(node2.getResourceCharacteristic().getParameters().get("category").get(0).getContent()); } }); if(nodes.size() > 0) { Map map = new HashMap(); List choosenResources = new ArrayList(); int cpuRequest; try { cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue(); } catch (NoSuchFieldException e) { cpuRequest = 1; } for (int i = 0; i < nodes.get(0).getProcessors().size() && cpuRequest > 0; i++) { if (nodes.get(0).getProcessors().get(i).getStatus() == ResourceStatus.FREE) { choosenResources.add(nodes.get(0).getProcessors().get(i)); cpuRequest--; } } ProcessingElements result = new ProcessingElements(nodes.get(0).getName()); result.addAll(choosenResources); map.put(StandardResourceUnitName.PE, result); return map; } else return null; } private List findSuitableNodes(TaskInterface task, List nodes){ int cpuRequest; try { cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue(); } catch (NoSuchFieldException e) { cpuRequest = 1; } List avNodes = new ArrayList(); for(ComputingNode node: nodes){ if(node.getFreeProcessorsNumber() >= cpuRequest){ avNodes.add(node); } } return avNodes; } private void turnOffIdleNodes(List nodes){ for(ComputingNode node : nodes){ if(node.getFreeProcessorsNumber() == node.getProcessorsNumber()){ node.getPowerInterface().setPowerState(StandardPowerStateName.OFF); } } } private boolean harnessIdleNodesToWork(TaskInterface task, List nodes){ int cpuRequest; try { cpuRequest = Double.valueOf(task.getCpuCntRequest()).intValue(); } catch (NoSuchFieldException e) { cpuRequest = 1; } Collections.sort(nodes, new Comparator(){ public int compare(ComputingNode node1, ComputingNode node2){ return node1.getResourceCharacteristic().getParameters().get("category").get(0).getContent().compareTo(node2.getResourceCharacteristic().getParameters().get("category").get(0).getContent()); } }); for (int i = 0; i < nodes.size() && cpuRequest > 0; i++) { ComputingNode node = nodes.get(i); if(node.getPowerInterface().getPowerState() == StandardPowerStateName.OFF){ node.getPowerInterface().setPowerState(StandardPowerStateName.ON); cpuRequest -= node.getProcessorsNumber(); } } return cpuRequest > 0 ? false : true; } public String getName() { return getClass().getName(); } }