bserp-core/src/main/java/org/turro/erp/task/WorkloadSet.java

Go to the documentation of this file.

/*
 * Turró i Cutiller Foundation. License notice.
 * Copyright (C) 2012 Lluis Turró Cutiller <http://www.turro.org/>
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Affero General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Affero General Public License for more details.
 *
 * You should have received a copy of the GNU Affero General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */
 
package org.turro.erp.task;
 
import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;
import java.util.List;
import java.util.TreeSet;
import org.turro.elephant.db.WhereClause;
import org.turro.erp.db.ErpPU;
import org.turro.erp.entity.HumanResource;
import org.turro.erp.entity.OrderItem;
import org.turro.erp.entity.RequiredUsage;
import org.turro.erp.entity.Resource;
import org.turro.erp.entity.Task;
import org.turro.erp.entity.TaskStatus;
import org.turro.jpa.Dao;
 
public class WorkloadSet extends TreeSet<Workload> {
  
  private HumanResource humanResource;
  private Resource resource;
  private boolean assigned;
 
  public WorkloadSet(HumanResource humanResource, boolean assigned) {
    this.assigned = assigned;
    setHumanResource(humanResource);
  }
 
  public WorkloadSet(Resource resource, boolean assigned) {
    this.assigned = assigned;
    setResource(resource);
  }
 
  public HumanResource getHumanResource() {
    return humanResource;
  }
 
  public void setHumanResource(HumanResource humanResource) {
    this.humanResource = humanResource;
    this.resource = null;
    generateList();
  }
 
  public Resource getResource() {
    return resource;
  }
 
  public void setResource(Resource resource) {
    this.resource = resource;
    this.humanResource = null;
    generateList();
  }
  
  public Collection<RequiredUsage> getRequiredUsages() {
    ArrayList<RequiredUsage> list = new ArrayList<RequiredUsage>();
    for(Workload wl : this) {
      list.add(wl.getRequiredUsage());
    }
    return list;
  }
 
  public Collection<OrderItem> getOrderItems() {
    ArrayList<OrderItem> list = new ArrayList<OrderItem>();
    for(Workload wl : this) {
      list.add(wl.getOrderItem());
    }
    return list;
  }
 
  private void generateList() {
    
    clear();
    
    if(assigned) {
      WhereClause wc = new WhereClause();
      wc.addClause("select distinct oi from OrderItem as oi");
      wc.addClause("where oi.requiredUsage.task.status <> :status1");
      wc.addClause("and oi.requiredUsage.task.status <> :status2");
      if(humanResource != null) {
        wc.addClause("and oi.humanResource = :humanRes1");
        wc.addNamedValue("humanRes1", humanResource);
      } else if(resource != null) {
        wc.addClause("and oi.related.resource = :resource1");
        wc.addNamedValue("resource1", resource);
      }
      wc.addNamedValue("status1", TaskStatus.TASK_PLANNING);
      wc.addNamedValue("status2", TaskStatus.TASK_FINISHED);
      Date now = new Date();
      Dao dao = new ErpPU();
      for(OrderItem oi : (List<OrderItem>) dao.getResultList(wc)) {
        Task task = oi.getRequiredUsage().getTask();
        task.checkStatus(now);
        if(TaskStatus.isOnWork(task.getStatus())) {
          if(resource != null) {
            add(new Workload(oi.getRelated()));
          } else {
            add(new Workload(oi));
          }
        }
      }
    } else {
      WhereClause wc = new WhereClause();
      wc.addClause("select distinct ru from RequiredUsage as ru");
      wc.addClause("where ru.task.status <> :status1");
      wc.addClause("and ru.task.status <> :status2");
      wc.addNamedValue("status1", TaskStatus.TASK_PLANNING);
      wc.addNamedValue("status2", TaskStatus.TASK_FINISHED);
      if(humanResource != null) {
        wc.addClause("and (ru.humanResource = :humanRes1");
        wc.addNamedValue("humanRes1", humanResource);
        wc.addClause("or ru.canChange = TRUE");
        wc.addClause(")");
      } else if(resource != null) {
        wc.addClause("and ru.resource = :resource1");
        wc.addNamedValue("resource1", resource);
      } else {
        wc.addClause("and (ru.humanResource is not null or ru.resource is not null)");
      }
 
      Date now = new Date();
      Dao dao = new ErpPU();
      for(RequiredUsage ru : (List<RequiredUsage>) dao.getResultList(wc)) {
        if(ru.getEstimatedUnits() > ru.getExpectedUnits()) {
          Task task = ru.getTask();
          task.checkStatus(now);
          if(TaskStatus.isOnWork(task.getStatus()) && (humanResource == null || ru.fits(humanResource))) {
            add(new Workload(ru));
          }
        }
      }
    }
 
  }
  
}