1 /*
2 * Copyright 2006-2007 the original author or authors.
3 *
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
7 *
8 * http://www.apache.org/licenses/LICENSE-2.0
9 *
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
15 */
16 package org.springframework.batch.sample.loop;
17
18 import org.springframework.batch.core.JobExecution;
19 import org.springframework.batch.core.StepExecution;
20 import org.springframework.batch.core.job.flow.FlowExecutionStatus;
21 import org.springframework.batch.core.job.flow.JobExecutionDecider;
22
23 /**
24 * This decider will return "CONTINUE" until the limit it reached, at which
25 * point it will return "COMPLETE".
26 *
27 * @author Dan Garrette
28 * @since 2.0
29 */
30 public class LimitDecider implements JobExecutionDecider {
31
32 private int count = 0;
33
34 private int limit = 1;
35
36 public FlowExecutionStatus decide(JobExecution jobExecution, StepExecution stepExecution) {
37 if (++count >= limit) {
38 return new FlowExecutionStatus("COMPLETED");
39 }
40 else {
41 return new FlowExecutionStatus("CONTINUE");
42 }
43 }
44
45 /**
46 * @param limit
47 */
48 public void setLimit(int limit) {
49 this.limit = limit;
50 }
51 }