src/main/java/org/onap/aai/champ/model/fluent/relationship/impl/TargetBuildOrPropertiesStepImpl.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.onap.aai.champ.model.fluent.relationship.impl;
  23
  24 import org.onap.aai.champ.model.ChampObject;
  25 import org.onap.aai.champ.model.ChampRelationship;
  26 import org.onap.aai.champ.model.fluent.relationship.RelationshipBuildOrPropertiesStep;
  27 import org.onap.aai.champ.model.fluent.relationship.TargetBuildOrPropertiesStep;
  28
  29 public final class TargetBuildOrPropertiesStepImpl implements TargetBuildOrPropertiesStep {
  30
  31         private final String relationshipType;
  32         private final ChampRelationship relationship;
  33         private final Object relationshipKey;
  34         private final ChampObject source;
  35         private final ChampObject.Builder targetBuilder;
  36
  37         public TargetBuildOrPropertiesStepImpl(String relationshipType, ChampRelationship relationship, Object relationshipKey,
  38                         ChampObject source, ChampObject.Builder targetBuilder) {
  39                 this.relationshipType = relationshipType;
  40                 this.relationship = relationship;
  41                 this.relationshipKey = relationshipKey;
  42                 this.source = source;
  43                 this.targetBuilder = targetBuilder;
  44         }
  45
  46         @Override
  47         public TargetBuildOrPropertiesStep withProperty(String key, Object value) {
  48                 targetBuilder.property(key, value);
  49                 return this;
  50         }
  51
  52         @Override
  53         public RelationshipBuildOrPropertiesStep build() {
  54
  55                 final ChampRelationship.Builder relationshipBuilder;
  56
  57                 if (relationship != null) relationshipBuilder = new ChampRelationship.Builder(relationship);
  58                 else relationshipBuilder = new ChampRelationship.Builder(source, targetBuilder.build(), relationshipType);
  59
  60                 if (relationshipKey != null) relationshipBuilder.key(relationshipKey);
  61
  62                 return new RelationshipBuildOrPropertiesStepImpl(relationshipBuilder);
  63         }
  64
  65 }