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}