source: DCWoRMS/branches/coolemall/src/schedframe/scheduling/tasks/Job.java @ 1396

package schedframe.scheduling.tasks;


import java.io.StringWriter;
import java.io.Writer;
import java.util.ArrayList;
import java.util.Iterator;
import java.util.List;

import org.joda.time.DateTime;
import org.qcg.broker.schemas.resreqs.ResourceRequirements;

import qcg.shared.constants.BrokerConstants;
import schedframe.scheduling.WorkloadUnitHandler;
import schedframe.scheduling.manager.tasks.JobRegistryImpl;


/**
 * 
 * @author Marcin Krystek
 *
 */
public class Job implements JobInterface<ResourceRequirements> {

        protected List<Task> tasks;
        protected int senderId;
        protected boolean isRegistered;
        
        private int status;
        
        public Job(String id){
                tasks = new ArrayList<Task>();
        }
        
        public Job(ResourceRequirements resourceRequirements) throws Exception{
                org.qcg.broker.schemas.resreqs.Task task[] = resourceRequirements.getTask();
                if(task == null || task.length == 0)
                        throw new NoSuchFieldException("No tasks are defined for job.");
                
                this.tasks = new ArrayList<Task>();
                
                for(int i = 0; i < task.length; i++){
                        this.tasks.add(new Task(task[i]));
                }
                
                isRegistered = true;
        }

        public void add(Task task) {
                this.tasks.add(task);
        }
        
        public String getId() {
                if(this.tasks.size() == 0)
                        throw new RuntimeException("No tasks are defined for job, so it is not possible to obtain job id.");
                
                return this.tasks.get(0).getJobId();
        }
        
        public List<Task> getTask() {
                return this.tasks;
        }
        
        public Task getTask(String taskId) throws NoSuchFieldException {
                if(taskId == null)
                        throw new IllegalArgumentException("TaskId can not be null. Specify appropriate taskId.");
                
                if(this.tasks == null || this.tasks.size() == 0)
                        throw new NoSuchFieldException("No tasks are defined for job.");
                
                Task retTask = null;
                
                Iterator<Task> itr = this.tasks.iterator();
                while(itr.hasNext() && retTask == null){
                        Task task = itr.next();
                        if(taskId.equals(task.getId())){
                                retTask = task;
                        }
                }

                if(retTask == null)
                        throw new NoSuchFieldException("Task "+taskId + " is not available in job " + getId());
                
                return retTask;
        }
        
        public int getTaskCount() {
                return this.tasks.size();
        }

        
        public ResourceRequirements getDescription() {
                
                ResourceRequirements resReq = new ResourceRequirements();
                if(this.tasks == null)
                        return resReq;
                
                Iterator<Task> itr = this.tasks.iterator();
                
                while(itr.hasNext()){
                        Task task = (Task) itr.next();
                        resReq.addTask(task.getDescription());
                }
                
                return resReq;
        }

        public String getDocument() throws Exception {
                ResourceRequirements resReq = getDescription();
                Writer writer = new StringWriter();
                
                resReq.marshal(writer);
                
                return writer.toString();
        }

        public boolean isFinished(){
                
                for(int i = 0; i < tasks.size(); i++){
                        //if(tasks.get(i).getStatus() != BrokerConstants.TASK_STATUS_FINISHED)
                        //      return false;
                        if(!tasks.get(i).isFinished())
                                return false;
                }
                return true;
        }
        
        public DateTime getSubmissionTimeToBroker(){
                return tasks.get(0).getSubmissionTimeToBroker();
        }
        
        public int getStatus(){
                boolean isForAll = true;
                int baseStatus = tasks.get(0).getStatus();
                
                for(int i = 1; i < tasks.size() && isForAll; i++){
                        Task t = tasks.get(i);
                        isForAll = (t.getStatus() == baseStatus);
                        switch(t.getStatus()){
                                case (int) BrokerConstants.TASK_STATUS_RUNNING:{
                                        return (int)BrokerConstants.JOB_STATUS_ACTIVE;
                                }
                                case (int) BrokerConstants.TASK_STATUS_QUEUED:{
                                        return (int)BrokerConstants.JOB_STATUS_ACTIVE;
                                }
                        }
                }
                
                if(isForAll && baseStatus == BrokerConstants.TASK_STATUS_FINISHED){
                        return (int)BrokerConstants.JOB_STATUS_FINISHED;
                } else if( baseStatus == BrokerConstants.TASK_STATUS_RUNNING){
                        return (int)BrokerConstants.JOB_STATUS_ACTIVE;
                } else if( baseStatus == BrokerConstants.TASK_STATUS_QUEUED){
                        return (int)BrokerConstants.JOB_STATUS_ACTIVE;
                }
                
                return status;
        }
        
        public boolean setStatus(String taskId, int status){
                boolean found = false;
                for(int i = 0; i < tasks.size() && !found; i++){
                        Task t = tasks.get(i);
                        if(taskId.equals(t.getId())){
                                try {
                                        t.setStatus(status);
                                } catch (Exception e) {
                                        // TODO Auto-generated catch block
                                        e.printStackTrace();
                                }
                                found = true;
                        }
                }
                return found;
        }
        
        public void setSenderId(int id){
                this.senderId = id;
        }
        
        public int getSenderId(){
                return this.senderId;
        }
        
        public int getUserId(){
                return this.senderId;
        }

        public void setStatus(int status){
                this.status = status;
        }

        public boolean isRegistered() {
                return isRegistered;
        }

        public void register(JobRegistryImpl jobRegistry) {
                isRegistered = jobRegistry.addJob(this);
        }
        
        @Override
        public void accept(WorkloadUnitHandler wuh) {
                wuh.handleJob(this);
        }
        
}