source: DCWoRMS/branches/coolemall/src/example/globalplugin/GridFCFSRandomPlugin.java @ 1562

package example.globalplugin;

import java.util.Comparator;
import java.util.List;
import java.util.Random;

import schedframe.events.scheduling.SchedulingEvent;
import schedframe.scheduling.SchedulerDescription;
import schedframe.scheduling.manager.resources.ResourceManager;
import schedframe.scheduling.manager.tasks.JobRegistry;
import schedframe.scheduling.plan.SchedulingPlanInterface;
import schedframe.scheduling.plan.impl.Allocation;
import schedframe.scheduling.plan.impl.ScheduledTask;
import schedframe.scheduling.plan.impl.SchedulingPlan;
import schedframe.scheduling.plugin.Module;
import schedframe.scheduling.plugin.ModuleList;
import schedframe.scheduling.plugin.grid.ResourceDiscovery;
import schedframe.scheduling.queue.TaskQueue;
import schedframe.scheduling.queue.TaskQueueList;
import schedframe.scheduling.tasks.TaskInterface;

public class GridFCFSRandomPlugin extends BaseGlobalPlugin {

        private Random rand;
        
        public GridFCFSRandomPlugin() {
                rand = new Random(5);
        }
        
        public SchedulingPlanInterface<?> schedule(SchedulingEvent event,
                        TaskQueueList queues, 
                        JobRegistry jobRegistry,
                        ResourceManager resourceManager, ModuleList modules)  {
                
                ResourceDiscovery resources = null;
                for(int i = 0; i < modules.size(); i++){
                        Module m = modules.get(i);
                        switch(m.getType()){
                                case RESOURCE_DISCOVERY: resources = (ResourceDiscovery) m;
                                        break;
                        }
                }
                
                SchedulingPlan plan = new SchedulingPlan();
                
                TaskQueue q = queues.get(0);
                int size = q.size();
                
                // order of the resources on this list is not determined
                List<SchedulerDescription> availableResources = resources.getResources(null);
                availableResources.sort(new SchedulerDescriptionComparator());
                int resourceIdx = -1;
                
                for(int i = 0; i < size; i++) {

                        TaskInterface<?> task = q.remove(0);
                        
                        resourceIdx = rand.nextInt(availableResources.size());
                        SchedulerDescription sd = availableResources.get(resourceIdx);
                        Allocation allocation = new Allocation();
                        allocation.setProcessesCount(1);
                        allocation.setProviderName(sd.getProvider().getProviderId());
                        
                        ScheduledTask scheduledTask = new ScheduledTask(task);
                        scheduledTask.setTaskId(task.getId());
                        scheduledTask.setJobId(task.getJobId());
                        scheduledTask.addAllocation(allocation);                
                        
                        plan.addTask(scheduledTask);

                }
                return plan;
                
        }

}

class SchedulerDescriptionComparator implements Comparator<SchedulerDescription>{

        @Override
        public int compare(SchedulerDescription o1, SchedulerDescription o2) {
                return(o1.getProvider().getProviderId().compareTo(o2.getProvider().getProviderId()));
        }
}