1 /*
2 * (c) Copyright 2006-2020 by rapiddweller GmbH & Volker Bergmann. All rights reserved.
3 *
4 * Redistribution and use in source and binary forms, with or without
5 * modification, is permitted under the terms of the
6 * GNU General Public License.
7 *
8 * For redistributing this software or a derivative work under a license other
9 * 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 rapiddweller GmbH & 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 com.rapiddweller.benerator.wrapper;
28
29 import com.rapiddweller.benerator.Generator;
30 import com.rapiddweller.benerator.GeneratorContext;
31 import com.rapiddweller.benerator.InvalidGeneratorSetupException;
32
33 /**
34 * {@link Generator} implementation which wraps a source {@link Generator}
35 * but generates a null value before forwarding the products of the source.<br/><br/>
36 * Created: 04.07.2011 13:18:03
37 *
38 * @param <E> the type parameter
39 * @author Volker Bergmann
40 * @since 0.7.0
41 */
42 public class NullStartingGenerator<E> extends GeneratorProxy<E> {
43
44 private boolean nullUsed;
45
46 /**
47 * Instantiates a new Null starting generator.
48 *
49 * @param source the source
50 */
51 public NullStartingGenerator(Generator<E> source) {
52 super(source);
53 this.nullUsed = false;
54 }
55
56 @Override
57 public void init(GeneratorContext context) throws InvalidGeneratorSetupException {
58 super.init(context);
59 }
60
61 @Override
62 public ProductWrapper<E> generate(ProductWrapper<E> wrapper) {
63 if (!nullUsed) {
64 this.nullUsed = true;
65 return wrapper.wrap(null);
66 }
67 return super.generate(wrapper);
68 }
69
70 @Override
71 public void reset() {
72 nullUsed = false;
73 super.reset();
74 }
75 }