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.distribution;
28
29 import com.rapiddweller.benerator.Generator;
30 import com.rapiddweller.benerator.GeneratorContext;
31 import com.rapiddweller.benerator.wrapper.GeneratorProxy;
32
33 /**
34 * General purpose generator proxy which is supposed to work with any distribution.
35 * The behavior on a reset is up to the generator created by the distribution.<br/>
36 * <br/>
37 * Created: 22.03.2010 10:45:48
38 *
39 * @param <E> the type parameter
40 * @author Volker Bergmann
41 * @since 0.6.0
42 */
43 public class DistributingGenerator<E> extends GeneratorProxy<E> {
44
45 private final Generator<E> dataProvider;
46 private final Distribution distribution;
47 private final boolean unique;
48
49 /**
50 * Instantiates a new Distributing generator.
51 *
52 * @param dataProvider the data provider
53 * @param distribution the distribution
54 * @param unique the unique
55 */
56 public DistributingGenerator(Generator<E> dataProvider, Distribution distribution, boolean unique) {
57 super(dataProvider.getGeneratedType());
58 this.dataProvider = dataProvider;
59 this.distribution = distribution;
60 this.unique = unique;
61 }
62
63 @Override
64 public void init(GeneratorContext context) {
65 dataProvider.init(context);
66 setSource(distribution.applyTo(dataProvider, unique));
67 super.init(context);
68 }
69
70 }