001/* 002 * Copyright 2006-2007 the original author or authors. 003 * 004 * Licensed under the Apache License, Version 2.0 (the "License"); 005 * you may not use this file except in compliance with the License. 006 * You may obtain a copy of the License at 007 * 008 * https://www.apache.org/licenses/LICENSE-2.0 009 * 010 * Unless required by applicable law or agreed to in writing, software 011 * distributed under the License is distributed on an "AS IS" BASIS, 012 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 013 * See the License for the specific language governing permissions and 014 * limitations under the License. 015 */ 016package org.springframework.batch.sample.loop; 017 018import org.springframework.batch.core.JobExecution; 019import org.springframework.batch.core.StepExecution; 020import org.springframework.batch.core.job.flow.FlowExecutionStatus; 021import org.springframework.batch.core.job.flow.JobExecutionDecider; 022 023/** 024 * This decider will return "CONTINUE" until the limit it reached, at which 025 * point it will return "COMPLETE". 026 * 027 * @author Dan Garrette 028 * @since 2.0 029 */ 030public class LimitDecider implements JobExecutionDecider { 031 032 private int count = 0; 033 034 private int limit = 1; 035 036 @Override 037 public FlowExecutionStatus decide(JobExecution jobExecution, StepExecution stepExecution) { 038 if (++count >= limit) { 039 return new FlowExecutionStatus("COMPLETED"); 040 } 041 else { 042 return new FlowExecutionStatus("CONTINUE"); 043 } 044 } 045 046 /** 047 * @param limit number of times to return "CONTINUE" 048 */ 049 public void setLimit(int limit) { 050 this.limit = limit; 051 } 052}