source: DCWoRMS/trunk/build/classes/example/localplugin/BaseLocalPlugin.java @ 477

package example.localplugin;

import java.util.Map;

import schedframe.Parameters;
import schedframe.PluginConfiguration;
import schedframe.events.scheduling.SchedulingEvent;
import schedframe.events.scheduling.SchedulingEventType;
import schedframe.events.scheduling.SchedulingResponseType;
import schedframe.resources.units.ResourceUnit;
import schedframe.resources.units.ResourceUnitName;
import schedframe.scheduling.WorkloadUnitListImpl;
import schedframe.scheduling.manager.resources.ResourceManager;
import schedframe.scheduling.manager.tasks.JobRegistryImpl;
import schedframe.scheduling.plan.impl.Allocation;
import schedframe.scheduling.plan.impl.ScheduledTask;
import schedframe.scheduling.plan.impl.SchedulingPlan;
import schedframe.scheduling.plugin.SchedulingPluginConfiguration;
import schedframe.scheduling.plugin.grid.ModuleList;
import schedframe.scheduling.plugin.local.LocalSchedulingPlugin;
import schedframe.scheduling.queue.TaskQueue;
import schedframe.scheduling.queue.TaskQueueList;
import schedframe.scheduling.tasks.TaskInterface;
import schedframe.scheduling.tasks.WorkloadUnit;
import schemas.StringValueWithUnit;

public abstract class BaseLocalPlugin implements LocalSchedulingPlugin {

        SchedulingPluginConfiguration plugConf;
        
        public PluginConfiguration getConfiguration() {
                return plugConf;
        }
        
        public SchedulingResponseType handleResourceAllocationViolation(SchedulingEvent event,
                        TaskQueueList queues, 
                        JobRegistryImpl jobRegistry, 
                                        ResourceManager resourceManager, ModuleList modules){
                SchedulingResponseType timeEvent = null;
                switch(event.getType()){
                        case TASK_REQUESTED_TIME_EXPIRED: 
                                timeEvent = SchedulingResponseType.KILL_TASK;
                                break;
                }
                return timeEvent;
        }
        
        public int placeJobsInQueues(WorkloadUnitListImpl newJobs,
                        TaskQueueList queues, 
                        ResourceManager resourceManager, ModuleList moduleList) {
                
                // get the first queue from all available queues.
                TaskQueue queue = queues.get(0);

                for(int i = 0; i < newJobs.size(); i++){
                        WorkloadUnit<?> task = newJobs.get(i);
                        queue.add(task);
                }
                
                return 0;
        }
        
        public void addToSchedulingPlan(SchedulingPlan plan, TaskInterface<?> task, Map<ResourceUnitName, ResourceUnit> choosenResources ){
                
                Allocation allocation = new Allocation();
                allocation.setProcessesCount(1);
                allocation.setSpecificResources(choosenResources);
                
                ScheduledTask scheduledTask = new ScheduledTask(task);
                scheduledTask.setTaskId(task.getId());
                scheduledTask.setJobId(task.getJobId());
                scheduledTask.addAllocation(allocation);                
                
                plan.addTask(scheduledTask);
        }
        
        public void addToSchedulingPlan(SchedulingPlan plan, TaskInterface<?> task, String providerName){
                
                Allocation allocation = new Allocation();
                allocation.setProcessesCount(1);
                allocation.setProviderName(providerName);
                
                ScheduledTask scheduledTask = new ScheduledTask(task);
                scheduledTask.setTaskId(task.getId());
                scheduledTask.setJobId(task.getJobId());
                scheduledTask.addAllocation(allocation);                
                
                plan.addTask(scheduledTask);
        }
        
        public void addToSchedulingPlan(SchedulingPlan plan, TaskInterface<?> task){
                
                Allocation allocation = new Allocation();
                allocation.setProcessesCount(1);
                allocation.setProviderName(null);
                
                ScheduledTask scheduledTask = new ScheduledTask(task);
                scheduledTask.setTaskId(task.getId());
                scheduledTask.setJobId(task.getJobId());
                scheduledTask.addAllocation(allocation);                
                
                plan.addTask(scheduledTask);
        }

        public void init(Parameters parameters) {
                plugConf = new SchedulingPluginConfiguration();
                try{
                        StringValueWithUnit freq = parameters.get("frequency").get(0);
                        plugConf.addServedEvent(SchedulingEventType.TIMER, Double.valueOf(freq.getContent()).intValue());
                } catch(Exception e){
                        
                }
        }
}