/*
    * Copyright 2006-2013 the original author or authors.
    *
    * Licensed under the Apache License, Version 2.0 (the "License");
    * you may not use this file except in compliance with the License.
    * You may obtain a copy of the License at
    *
    *      http://www.apache.org/licenses/LICENSE-2.0
    *
   * Unless required by applicable law or agreed to in writing, software
   * distributed under the License is distributed on an "AS IS" BASIS,
   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   * See the License for the specific language governing permissions and
   * limitations under the License.
   */
  
  package org.springframework.batch.core.explore.support;
  
  import java.util.List;
  import java.util.Set;
  
  import org.springframework.batch.core.JobExecution;
  import org.springframework.batch.core.JobInstance;
  import org.springframework.batch.core.StepExecution;
  import org.springframework.batch.core.explore.JobExplorer;
  import org.springframework.batch.core.repository.dao.ExecutionContextDao;
  import org.springframework.batch.core.repository.dao.JobExecutionDao;
  import org.springframework.batch.core.repository.dao.JobInstanceDao;
  import org.springframework.batch.core.repository.dao.StepExecutionDao;
  
  /**
   * Implementation of {@link JobExplorer} using the injected DAOs.
   *
   * @author Dave Syer
   * @author Lucas Ward
   *
   * @see JobExplorer
   * @see JobInstanceDao
   * @see JobExecutionDao
   * @see StepExecutionDao
   * @since 2.0
   */
  public class SimpleJobExplorer implements JobExplorer {
  
  	private JobInstanceDao jobInstanceDao;
  
  	private JobExecutionDao jobExecutionDao;
  
  	private StepExecutionDao stepExecutionDao;
  
  	private ExecutionContextDao ecDao;
  
  	/**
  	 * Provide default constructor with low visibility in case user wants to use
  	 * use aop:proxy-target-class="true" for AOP interceptor.
  	 */
  	SimpleJobExplorer() {
  	}
  
  	public SimpleJobExplorer(JobInstanceDao jobInstanceDao, JobExecutionDao jobExecutionDao,
  			StepExecutionDao stepExecutionDao, ExecutionContextDao ecDao) {
  		super();
  		this.jobInstanceDao = jobInstanceDao;
  		this.jobExecutionDao = jobExecutionDao;
  		this.stepExecutionDao = stepExecutionDao;
  		this.ecDao = ecDao;
  	}
  
  	/*
  	 * (non-Javadoc)
  	 *
  	 * @see
  	 * org.springframework.batch.core.explore.JobExplorer#findJobExecutions(
  	 * org.springframework.batch.core.JobInstance)
  	 */
  	@Override
  	public List<JobExecution> getJobExecutions(JobInstance jobInstance) {
  		List<JobExecution> executions = jobExecutionDao.findJobExecutions(jobInstance);
  		for (JobExecution jobExecution : executions) {
  			getJobExecutionDependencies(jobExecution);
  			for (StepExecution stepExecution : jobExecution.getStepExecutions()) {
  				getStepExecutionDependencies(stepExecution);
  			}
  		}
  		return executions;
  	}
  
  	/*
  	 * (non-Javadoc)
  	 *
  	 * @see
  	 * org.springframework.batch.core.explore.JobExplorer#findRunningJobExecutions
  	 * (java.lang.String)
  	 */
  	@Override
  	public Set<JobExecution> findRunningJobExecutions(String jobName) {
  		Set<JobExecution> executions = jobExecutionDao.findRunningJobExecutions(jobName);
  		for (JobExecution jobExecution : executions) {
  			getJobExecutionDependencies(jobExecution);
 			for (StepExecution stepExecution : jobExecution.getStepExecutions()) {
 				getStepExecutionDependencies(stepExecution);
 			}
 		}
 		return executions;
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.springframework.batch.core.explore.JobExplorer#getJobExecution(java
 	 * .lang.Long)
 	 */
 	@Override
 	public JobExecution getJobExecution(Long executionId) {
 		if (executionId == null) {
 			return null;
 		}
 		JobExecution jobExecution = jobExecutionDao.getJobExecution(executionId);
 		if (jobExecution == null) {
 			return null;
 		}
 		getJobExecutionDependencies(jobExecution);
 		for (StepExecution stepExecution : jobExecution.getStepExecutions()) {
 			getStepExecutionDependencies(stepExecution);
 		}
 		return jobExecution;
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.springframework.batch.core.explore.JobExplorer#getStepExecution(java
 	 * .lang.Long)
 	 */
 	@Override
 	public StepExecution getStepExecution(Long jobExecutionId, Long executionId) {
 		JobExecution jobExecution = jobExecutionDao.getJobExecution(jobExecutionId);
 		if (jobExecution == null) {
 			return null;
 		}
 		StepExecution stepExecution = stepExecutionDao.getStepExecution(jobExecution, executionId);
 		getStepExecutionDependencies(stepExecution);
 		return stepExecution;
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.springframework.batch.core.explore.JobExplorer#getJobInstance(java
 	 * .lang.Long)
 	 */
 	@Override
 	public JobInstance getJobInstance(Long instanceId) {
 		return jobInstanceDao.getJobInstance(instanceId);
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see
 	 * org.springframework.batch.core.explore.JobExplorer#getLastJobInstances
 	 * (java.lang.String, int)
 	 */
 	@Override
 	public List<JobInstance> getJobInstances(String jobName, int start, int count) {
 		return jobInstanceDao.getJobInstances(jobName, start, count);
 	}
 
 	/*
 	 * (non-Javadoc)
 	 *
 	 * @see org.springframework.batch.core.explore.JobExplorer#getJobNames()
 	 */
 	@Override
 	public List<String> getJobNames() {
 		return jobInstanceDao.getJobNames();
 	}
 
 	/*
 	 * Find all dependencies for a JobExecution, including JobInstance (which
 	 * requires JobParameters) plus StepExecutions
 	 */
 	private void getJobExecutionDependencies(JobExecution jobExecution) {
 
 		JobInstance jobInstance = jobInstanceDao.getJobInstance(jobExecution);
 		stepExecutionDao.addStepExecutions(jobExecution);
 		jobExecution.setJobInstance(jobInstance);
 		jobExecution.setExecutionContext(ecDao.getExecutionContext(jobExecution));
 
 	}
 
 	private void getStepExecutionDependencies(StepExecution stepExecution) {
 		if (stepExecution != null) {
 			stepExecution.setExecutionContext(ecDao.getExecutionContext(stepExecution));
 		}
 	}
 
 }