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 */ 016 017package org.springframework.batch.sample.domain.trade.internal; 018 019import java.math.BigDecimal; 020 021import org.springframework.batch.item.ItemReader; 022import org.springframework.batch.sample.domain.trade.Trade; 023 024/** 025 * Generates configurable number of {@link Trade} items. 026 * 027 * @author Robert Kasanicky 028 */ 029public class GeneratingTradeItemReader implements ItemReader<Trade> { 030 031 private int limit = 1; 032 033 private int counter = 0; 034 035 @Override 036 public Trade read() throws Exception { 037 if (counter < limit) { 038 counter++; 039 return new Trade( 040 "isin" + counter, 041 counter, 042 new BigDecimal(counter), 043 "customer" + counter); 044 } 045 return null; 046 } 047 048 /** 049 * @param limit number of items that will be generated 050 * (null returned on consecutive calls). 051 */ 052 public void setLimit(int limit) { 053 this.limit = limit; 054 } 055 056 public int getCounter() { 057 return counter; 058 } 059 060 public int getLimit() { 061 return limit; 062 } 063 064 public void resetCounter() 065 { 066 this.counter = 0; 067 } 068}