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.engine;
28
29 import com.rapiddweller.benerator.Generator;
30 import com.rapiddweller.benerator.util.WrapperProvider;
31 import com.rapiddweller.benerator.wrapper.ProductWrapper;
32
33 /**
34 * Uses a {@link Generator} to create the currently processed object.<br/><br/>
35 * Created: 01.09.2011 19:03:38
36 *
37 * @author Volker Bergmann
38 * @since 0.7.0
39 */
40 public class CurrentProductGeneration implements Statement, LifeCycleHolder {
41
42 private final String instanceName;
43 private final Generator<Object> source;
44 private final WrapperProvider<Object> provider;
45
46 /**
47 * Instantiates a new Current product generation.
48 *
49 * @param instanceName the instance name
50 * @param source the source
51 */
52 @SuppressWarnings({"unchecked", "rawtypes"})
53 public CurrentProductGeneration(String instanceName, Generator<?> source) {
54 this.instanceName = instanceName;
55 this.source = (Generator) source;
56 this.provider = new WrapperProvider<>();
57 }
58
59 @Override
60 public void init(BeneratorContext context) {
61 source.init(context);
62 }
63
64 @Override
65 public boolean execute(BeneratorContext context) {
66 ProductWrapper<Object> wrapper = source.generate(provider.get());
67 context.setCurrentProduct(wrapper);
68 if (instanceName != null && wrapper != null) {
69 BeneratorContext parent = ((BeneratorSubContext) context).getParent();
70 parent.set(instanceName, wrapper.unwrap());
71 }
72 return (wrapper != null);
73 }
74
75 @Override
76 public void reset() {
77 source.reset();
78 }
79
80 @Override
81 public void close() {
82 source.close();
83 }
84
85 @Override
86 public String toString() {
87 return instanceName + ':' + source;
88 }
89
90 }