org.apache.batchee.container.impl.controller.DecisionController 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.impl.controller;
import org.apache.batchee.container.ExecutionElementController;
import org.apache.batchee.container.exception.BatchContainerRuntimeException;
import org.apache.batchee.container.impl.controller.chunk.ExceptionConfig;
import org.apache.batchee.container.impl.jobinstance.RuntimeJobExecution;
import org.apache.batchee.container.jsl.ExecutionElement;
import org.apache.batchee.container.proxy.InjectionReferences;
import org.apache.batchee.container.proxy.ProxyFactory;
import org.apache.batchee.container.services.ServicesManager;
import org.apache.batchee.container.status.ExecutionStatus;
import org.apache.batchee.container.status.ExtendedBatchStatus;
import org.apache.batchee.jaxb.Decision;
import org.apache.batchee.jaxb.Property;
import org.apache.batchee.spi.BatchArtifactFactory;
import org.apache.batchee.spi.PersistenceManagerService;
import javax.batch.api.Decider;
import javax.batch.runtime.StepExecution;
import java.util.List;
public class DecisionController implements ExecutionElementController {
private RuntimeJobExecution jobExecution;
private Decision decision;
private StepExecution[] previousStepExecutions = null;
private final PersistenceManagerService persistenceService;
private final BatchArtifactFactory factory;
public DecisionController(final RuntimeJobExecution jobExecution, final Decision decision, final ServicesManager manager) {
this.jobExecution = jobExecution;
this.decision = decision;
this.persistenceService = manager.service(PersistenceManagerService.class);
this.factory = manager.service(BatchArtifactFactory.class);
}
@Override
public ExecutionStatus execute() {
final String deciderId = decision.getRef();
final List propList = (decision.getProperties() == null) ? null : decision.getProperties().getPropertyList();
//Create a decider proxy and inject the associated properties
/* Set the contexts associated with this scope */
//job context is always in scope
//the parent controller will only pass one valid context to a decision controller
//so two of these contexts will always be null
final InjectionReferences injectionRef = new InjectionReferences(jobExecution.getJobContext(), null, propList);
final Decider deciderProxy = ProxyFactory.createDeciderProxy(factory, deciderId, injectionRef, jobExecution);
String exitStatus = null;
try {
exitStatus = deciderProxy.decide(this.previousStepExecutions);
} catch (Exception e) {
ExceptionConfig.wrapBatchException(e);
}
//Set the value returned from the decider as the job context exit status.
this.jobExecution.getJobContext().setExitStatus(exitStatus);
return new ExecutionStatus(ExtendedBatchStatus.NORMAL_COMPLETION, exitStatus);
}
protected void setPreviousStepExecutions(ExecutionElement previousExecutionElement, ExecutionElementController previousElementController) {
if (previousExecutionElement == null) {
// only job context is available to the decider
} else if (previousExecutionElement instanceof Decision) {
throw new BatchContainerRuntimeException("A decision cannot precede another decision.");
}
List previousStepExecsIds = previousElementController.getLastRunStepExecutions();
StepExecution[] stepExecArray = new StepExecution[previousStepExecsIds.size()];
for (int i = 0; i < stepExecArray.length; i++) {
StepExecution stepExec = persistenceService.getStepExecutionByStepExecutionId(previousStepExecsIds.get(i));
stepExecArray[i] = stepExec;
}
this.previousStepExecutions = stepExecArray;
}
@Override
public void stop() {
// no-op
}
@Override
public List getLastRunStepExecutions() {
// TODO Auto-generated method stub
return null;
}
}
© 2015 - 2024 Weber Informatics LLC | Privacy Policy