catalog-dao/src/main/java/org/openecomp/sdc/be/resources/data/ServiceMetadataData.java

   1 /*-
   2  * ============LICENSE_START=======================================================
   3  * SDC
   4  * ================================================================================
   5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
   6  * ================================================================================
   7  * Licensed under the Apache License, Version 2.0 (the "License");
   8  * you may not use this file except in compliance with the License.
   9  * You may obtain a copy of the License at
  10  *
  11  *      http://www.apache.org/licenses/LICENSE-2.0
  12  *
  13  * Unless required by applicable law or agreed to in writing, software
  14  * distributed under the License is distributed on an "AS IS" BASIS,
  15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  16  * See the License for the specific language governing permissions and
  17  * limitations under the License.
  18  * ============LICENSE_END=========================================================
  19  */
  20
  21 package org.openecomp.sdc.be.resources.data;
  22
  23 import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionary;
  24 import org.openecomp.sdc.be.dao.neo4j.GraphPropertiesDictionaryExtractor;
  25 import org.openecomp.sdc.be.datatypes.components.ServiceMetadataDataDefinition;
  26 import org.openecomp.sdc.be.datatypes.enums.NodeTypeEnum;
  27
  28 import java.util.Map;
  29
  30 public class ServiceMetadataData extends ComponentMetadataData {
  31
  32         public ServiceMetadataData() {
  33                 super(NodeTypeEnum.Service, new ServiceMetadataDataDefinition());
  34         }
  35
  36         public ServiceMetadataData(ServiceMetadataDataDefinition serviceMetadataDataDefinition) {
  37                 super(NodeTypeEnum.Service, serviceMetadataDataDefinition);
  38         }
  39
  40         public ServiceMetadataData(GraphPropertiesDictionaryExtractor extractor) {
  41                 super(NodeTypeEnum.Service, new ServiceMetadataDataDefinition(), extractor);
  42                  metadataDataDefinition.setProjectCode(extractor.getProjectCode());
  43                 ((ServiceMetadataDataDefinition) metadataDataDefinition).setDistributionStatus(extractor.getDistributionStatus());
  44         }
  45
  46         @Override
  47         public String getUniqueIdKey() {
  48                 return GraphPropertiesDictionary.UNIQUE_ID.getProperty();
  49         }
  50
  51         @Override
  52         public Map<String, Object> toGraphMap() {
  53                 Map<String, Object> graphMap = super.toGraphMap();
  54                 addIfExists(graphMap, GraphPropertiesDictionary.PROJECT_CODE,
  55                                 ((ServiceMetadataDataDefinition) metadataDataDefinition).getProjectCode());
  56                 addIfExists(graphMap, GraphPropertiesDictionary.DISTRIBUTION_STATUS,
  57                                 ((ServiceMetadataDataDefinition) metadataDataDefinition).getDistributionStatus());
  58                 return graphMap;
  59         }
  60
  61 }