org.apache.batchee.container.services.status.DefaultJobStatusManager Maven / Gradle / Ivy
/*
* Copyright 2012 International Business Machines Corp.
*
* See the NOTICE file distributed with this work for additional information
* regarding copyright ownership. 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.apache.batchee.container.services.status;
import org.apache.batchee.container.exception.BatchContainerServiceException;
import org.apache.batchee.container.services.JobStatusManagerService;
import org.apache.batchee.container.services.ServicesManager;
import org.apache.batchee.container.status.JobStatus;
import org.apache.batchee.container.status.StepStatus;
import org.apache.batchee.spi.PersistenceManagerService;
import javax.batch.runtime.BatchStatus;
import java.util.Properties;
public class DefaultJobStatusManager implements JobStatusManagerService {
private PersistenceManagerService persistenceManager;
public DefaultJobStatusManager(final ServicesManager servicesManager) {
this.persistenceManager = servicesManager.service(PersistenceManagerService.class);
}
@Override
public JobStatus createJobStatus(long jobInstanceId) throws BatchContainerServiceException {
return persistenceManager.createJobStatus(jobInstanceId);
}
@Override
public JobStatus getJobStatus(final long jobInstanceId) throws BatchContainerServiceException {
return persistenceManager.getJobStatus(jobInstanceId);
}
@Override
public void updateJobStatus(final JobStatus jobStatus) {
persistJobStatus(jobStatus.getJobInstanceId(), jobStatus);
}
@Override
public JobStatus getJobStatusFromExecutionId(final long executionId) throws BatchContainerServiceException {
return persistenceManager.getJobStatusFromExecution(executionId);
}
@Override
public void updateJobBatchStatus(final long jobInstanceId, final BatchStatus batchStatus) throws BatchContainerServiceException {
final JobStatus js = getJobStatus(jobInstanceId);
if (js == null) {
throw new IllegalStateException("Couldn't find entry to update for id = " + jobInstanceId);
}
js.setBatchStatus(batchStatus);
persistJobStatus(jobInstanceId, js);
}
@Override
public void updateJobExecutionStatus(final long jobInstanceId, final BatchStatus batchStatus, final String exitStatus) throws BatchContainerServiceException {
final JobStatus js = getJobStatus(jobInstanceId);
if (js == null) {
throw new IllegalStateException("Couldn't find entry to update for id = " + jobInstanceId);
}
js.setBatchStatus(batchStatus);
js.setExitStatus(exitStatus);
persistJobStatus(jobInstanceId, js);
}
@Override
public void updateJobCurrentStep(final long jobInstanceId, final String currentStepName) throws BatchContainerServiceException {
final JobStatus js = getJobStatus(jobInstanceId);
if (js == null) {
throw new IllegalStateException("Couldn't find entry to update for id = " + jobInstanceId);
}
js.setCurrentStepId(currentStepName);
persistJobStatus(jobInstanceId, js);
}
@Override
public void updateJobStatusWithNewExecution(final long jobInstanceId, final long newExecutionId) throws BatchContainerServiceException {
final JobStatus js = getJobStatus(jobInstanceId);
if (js == null) {
throw new IllegalStateException("Couldn't find entry to update for id = " + jobInstanceId);
}
js.setRestartOn(null);
js.setLatestExecutionId(newExecutionId);
js.setBatchStatus(BatchStatus.STARTING);
persistJobStatus(jobInstanceId, js);
}
private void persistJobStatus(long jobInstanceId, JobStatus newJobStatus) throws BatchContainerServiceException {
persistenceManager.updateJobStatus(jobInstanceId, newJobStatus);
}
@Override
public StepStatus createStepStatus(final long stepExecutionId) throws BatchContainerServiceException {
return persistenceManager.createStepStatus(stepExecutionId);
}
/*
* @return - StepStatus or null if one is unknown
*/
@Override
public StepStatus getStepStatus(final long jobInstanceId, final String stepId) throws BatchContainerServiceException {
return persistenceManager.getStepStatus(jobInstanceId, stepId);
}
@Override
public void updateStepStatus(final long stepExecutionId, final StepStatus newStepStatus) {
persistenceManager.updateStepStatus(stepExecutionId, newStepStatus);
}
@Override
public void init(final Properties batchConfig) throws BatchContainerServiceException {
// no-op
}
/*
* Inefficient, since we've already updated the status to stopped.. would be better to have a single update.
*/
@Override
public void updateJobStatusFromJSLStop(final long jobInstanceId, final String restartOn) throws BatchContainerServiceException {
final JobStatus js = getJobStatus(jobInstanceId);
if (js == null) {
throw new IllegalStateException("Couldn't find entry to update for id = " + jobInstanceId);
}
js.setRestartOn(restartOn);
persistJobStatus(jobInstanceId, js);
}
@Override
public String toString() {
return getClass().getName();
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy