Merge "Added @Override annotation above signature"
[aai/champ.git] / src / main / java / org / openecomp / aai / champ / model / fluent / schema / impl / RelationshipConstraintRequiredOptionalStepImpl.java
1 /**
2  * ============LICENSE_START==========================================
3  * org.onap.aai
4  * ===================================================================
5  * Copyright © 2017 AT&T Intellectual Property. All rights reserved.
6  * Copyright © 2017 Amdocs
7  * ===================================================================
8  * Licensed under the Apache License, Version 2.0 (the "License");
9  * you may not use this file except in compliance with the License.
10  * You may obtain a copy of the License at
11  *
12  *        http://www.apache.org/licenses/LICENSE-2.0
13  *
14  * Unless required by applicable law or agreed to in writing, software
15  * distributed under the License is distributed on an "AS IS" BASIS,
16  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17  * See the License for the specific language governing permissions and
18  * limitations under the License.
19  * ============LICENSE_END============================================
20  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
21  */
22 package org.openecomp.aai.champ.model.fluent.schema.impl;
23
24 import org.openecomp.aai.champ.model.ChampField;
25 import org.openecomp.aai.champ.model.ChampPropertyConstraint;
26 import org.openecomp.aai.champ.model.ChampRelationshipConstraint;
27 import org.openecomp.aai.champ.model.ChampSchema;
28 import org.openecomp.aai.champ.model.ChampSchema.Builder;
29 import org.openecomp.aai.champ.model.fluent.schema.RelationshipConstraintBuildStep;
30 import org.openecomp.aai.champ.model.fluent.schema.RelationshipConstraintRequiredOptionalStep;
31
32 public class RelationshipConstraintRequiredOptionalStepImpl implements RelationshipConstraintRequiredOptionalStep {
33
34         private final ChampSchema.Builder schemaBuilder;
35         private final ChampRelationshipConstraint.Builder relConstraintBuilder;
36         private final ChampField.Builder fieldBuilder;
37
38         public RelationshipConstraintRequiredOptionalStepImpl(Builder schemaBuilder,
39                                                                                                                         ChampRelationshipConstraint.Builder relConstraintBuilder,
40                                                                                                                         ChampField.Builder fieldBuilder) {
41                 this.schemaBuilder = schemaBuilder;
42                 this.relConstraintBuilder = relConstraintBuilder;
43                 this.fieldBuilder = fieldBuilder;
44         }
45
46         @Override
47         public RelationshipConstraintBuildStep required() {
48                 final ChampPropertyConstraint propConstraint = new ChampPropertyConstraint.Builder(fieldBuilder.build())
49                                                                                                                                                                         .required(true)
50                                                                                                                                                                         .build();
51
52                 return new RelationshipConstraintBuildStepImpl(schemaBuilder, relConstraintBuilder.constraint(propConstraint));
53         }
54         
55         @Override
56         public RelationshipConstraintBuildStep optional() {
57                 final ChampPropertyConstraint propConstraint = new ChampPropertyConstraint.Builder(fieldBuilder.build())
58                                                                                                                                                                         .required(false)
59                                                                                                                                                                         .build();
60
61                 return new RelationshipConstraintBuildStepImpl(schemaBuilder, relConstraintBuilder.constraint(propConstraint));
62         }
63
64 }