package com.trendmicro.grid.acl.ds.jpa.entities;
   
   import com.trendmicro.grid.acl.Limits;
   import com.trendmicro.grid.acl.ds.datatypes.SharedJob;
   import com.trendmicro.grid.acl.metadata.Metadata;
   import org.hibernate.annotations.Index;
   import org.hibernate.annotations.Type;
   
   import javax.persistence.*;
  import java.sql.Timestamp;
  import java.util.Date;
  import java.util.List;
  import java.util.UUID;
  
  import static com.trendmicro.grid.acl.ds.jpa.util.JpaUtils.fromBytes;
  import static com.trendmicro.grid.acl.ds.jpa.util.JpaUtils.toBytes;
  
  /**
   * Binds Job to the table "JOBS".
   *
   * @author juergen_kellerer, 2010-06-01
   * @version 1.0
   */
  @NamedQueries({
  		@NamedQuery(name = "Job.SelectById", query = "" +
  				"SELECT j FROM JOBS j " +
  				"WHERE 	j.publicGUID = :publicGUID"),
  
  		@NamedQuery(name = "Job.SelectJobStateById", query = "" +
  				"SELECT j.state FROM JOBS j " +
  				"WHERE 	j.publicGUID = :publicGUID"),
  
  		@NamedQuery(name = "Job.SelectJobIdsByState", query = "" +
  				"SELECT j.publicGUID FROM JOBS j " +
  				"WHERE 	j.state = :state"),
  
  		@NamedQuery(name = "Job.SelectJobIdsByStateInRange", query = "" +
  				"SELECT j.publicGUID FROM JOBS j " +
  				"WHERE 	j.state = :state AND j.lastUpdated BETWEEN :fromDate AND :toDate"),
  
  		@NamedQuery(name = "Job.SelectJobId", query = "" +
  				"SELECT j.state FROM JOBS j " +
  				"WHERE 	j.publicGUID = :publicGUID")
  })
  @Cacheable
  @Entity(name = "JOBS")
  public class JpaJob extends SharedJob {
  
  	private static final long serialVersionUID = 4621907144968629303L;
  
  	protected int primaryKey;
  	protected JpaJob parentJob;
  	protected List<JpaJob> subJobs;
  
  	/**
  	 * Used by JPA to construct an empty instance.
  	 */
  	public JpaJob() {
  	}
  
  	/**
  	 * Constructs a new JpaJob instance for the given parent job.
  	 *
  	 * @param jobId	 the job id of the job to create.
  	 * @param parentJob the parent job to link this job against.
  	 */
  	public JpaJob(UUID jobId, JpaJob parentJob) {
  		super(jobId, parentJob == null ? null : parentJob.getJobId(), State.PREPARED,
  				0, new Date(), null, null, null);
  		this.parentJob = parentJob;
  	}
  
  	@Id
  	@GeneratedValue
  	@Column(name = "JOB_ID", nullable = false)
  	public int getPrimaryKey() {
  		return primaryKey;
  	}
  
  	public void setPrimaryKey(int primaryKey) {
  		this.primaryKey = primaryKey;
  	}
  
  	@ManyToOne(fetch = FetchType.LAZY)
  	@JoinColumn(name = "PARENT_JOB_ID", nullable = true, updatable = false)
  	public JpaJob getParentJob() {
  		return parentJob;
  	}
  
  	public void setParentJob(JpaJob parentJob) {
  		this.parentJob = parentJob;
  	}
  
  	@OneToMany(mappedBy = "parentJob", fetch = FetchType.LAZY)
  	public List<JpaJob> getSubJobs() {
  		return subJobs;
  	}
  
  	public void setSubJobs(List<JpaJob> subJobs) {
 		this.subJobs = subJobs;
 	}
 
 	@Override
 	@Index(name = "STATE_INDEX")
 	@Column(name = "STATE", nullable = false)
 	@Enumerated(EnumType.ORDINAL)
 	public State getState() {
 		return state;
 	}
 
 	@Override
 	public void setState(State state) {
 		this.state = state;
 	}
 
 	@Override
 	@Column(name = "STAGE")
 	public int getStagesPassed() {
 		return stagesPassed;
 	}
 
 	@Override
 	public void setStagesPassed(int stagesPassed) { //NOSONAR - Override is required as hook for JPA.
 		super.setStagesPassed(stagesPassed);
 	}
 
 	@Transient
 	public void setParentJobId(UUID parentJobId) {
 		this.parentJobId = parentJobId;
 	}
 
 	@Index(name = "IX_JOBS__PUBLIC_GUID_INDEX")
 	@Column(name = "PUBLIC_GUID", nullable = false, length = 16)
 	public byte[] getPublicGUID() {
 		return toBytes(jobId);
 	}
 
 	public void setPublicGUID(byte[] publicGUID) {
 		jobId = fromBytes(publicGUID);
 	}
 
 	@Override
 	@Column(name = "CREATED", nullable = false)
 	@Temporal(TemporalType.TIMESTAMP)
 	public Date getCreated() {
 		return created;
 	}
 
 	public void setCreated(Date created) {
 		// Note: Converting Timestamp to Date to avoid that "equals()" fails to compare 2 instances.
 		this.created = created instanceof Timestamp ? new Date(created.getTime()) : created;
 	}
 
 	@Override
 	@Column(name = "LAST_UPDATED")
 	@Temporal(TemporalType.TIMESTAMP)
 	public Date getLastUpdated() {
 		return lastUpdated;
 	}
 
 	@Override
 	public void setLastUpdated(Date lastUpdated) { //NOSONAR - Override is required as hook for JPA.
 		// Note: Converting Timestamp to Date to avoid that "equals()" fails to compare 2 instances.
 		lastUpdated = lastUpdated instanceof Timestamp ? new Date(lastUpdated.getTime()) : lastUpdated;
 
 		super.setLastUpdated(lastUpdated);
 	}
 
 	@Override
 	@Column(name = "FINISHED")
 	@Temporal(TemporalType.TIMESTAMP)
 	public Date getFinished() {
 		return finished;
 	}
 
 	public void setFinished(Date finished) {
 		if (this.finished == null) {
 			// Note: Converting Timestamp to Date to avoid that "equals()" fails to compare 2 instances.
 			this.finished = finished instanceof Timestamp ? new Date(finished.getTime()) : finished;
 		}
 	}
 
 	@Override
 	@Type(type = "metadata")
 	@Basic(fetch = FetchType.EAGER)
 	@Column(name = "META_DATA", length = Limits.MAX_SERIALIZED_METADATA_LENGTH)
 	public Metadata getMetadata() {
 		return super.getMetadata();
 	}
 
 	@Override
 	public void setMetadata(Metadata metadata) { //NOSONAR - Override is required as hook for JPA.
 		super.setMetadata(metadata == null ? null : metadata.clone());
 	}
 
 	@Override
 	public String toString() {
 		return "JpaJob{" +
 				"primaryKey=" + primaryKey +
 				", jobId=" + jobId +
 				", parentJobId=" + parentJobId +
 				", state=" + state +
 				", stagesPassed=" + stagesPassed +
 				", parentJobId=" + parentJobId +
 				", created=" + created +
 				", lastUpdated=" + lastUpdated +
 				", finished=" + finished +
 				", metadata=" + metadata +
 				", parentJob=" + parentJob +
 				", subJobs=" + subJobs +
 				'}';
 	}
 }