01: /*
02: * (c) Copyright 2007 by Volker Bergmann. All rights reserved.
03: *
04: * Redistribution and use in source and binary forms, with or without
05: * modification, is permitted under the terms of the
06: * GNU General Public License.
07: *
08: * For redistributing this software or a derivative work under a license other
09: * than the GPL-compatible Free Software License as defined by the Free
10: * Software Foundation or approved by OSI, you must first obtain a commercial
11: * license to this software product from Volker Bergmann.
12: *
13: * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
14: * WITHOUT A WARRANTY OF ANY KIND. ALL EXPRESS OR IMPLIED CONDITIONS,
15: * REPRESENTATIONS AND WARRANTIES, INCLUDING ANY IMPLIED WARRANTY OF
16: * MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR NON-INFRINGEMENT, ARE
17: * HEREBY EXCLUDED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE
18: * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
19: * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
20: * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
21: * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
22: * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
23: * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
24: * POSSIBILITY OF SUCH DAMAGE.
25: */
26:
27: package org.databene.benerator.wrapper;
28:
29: import org.databene.benerator.GeneratorProxy;
30: import org.databene.benerator.Generator;
31: import org.databene.benerator.IllegalGeneratorStateException;
32: import org.apache.commons.logging.Log;
33: import org.apache.commons.logging.LogFactory;
34:
35: /**
36: * Proxy that forwards a limited number of products of another Generator.<br/>
37: * <br/>
38: * Created: 01.09.2007 10:54:01
39: */
40: public class NShotGeneratorProxy<E> extends GeneratorProxy<E> {
41:
42: private static final Log logger = LogFactory
43: .getLog(NShotGeneratorProxy.class);
44:
45: private long shots;
46:
47: private long remainingShots;
48:
49: public NShotGeneratorProxy(Generator<E> source, long shots) {
50: super (source);
51: this .shots = shots;
52: this .remainingShots = shots;
53: }
54:
55: public boolean available() {
56: if (remainingShots <= 0) {
57: logger.debug("requested count reached for " + source);
58: return false;
59: }
60: return super .available();
61: }
62:
63: public E generate() {
64: if (remainingShots <= 0)
65: throw new IllegalGeneratorStateException(
66: "Generator not available.");
67: this .remainingShots--;
68: return super .generate();
69: }
70:
71: public void reset() {
72: super .reset();
73: remainingShots = shots;
74: }
75:
76: public void close() {
77: super .close();
78: remainingShots = 0;
79: }
80: }
|