145b23f6ad8d3cf75b1d120506bc7f342f7f51ef
[aai/sparky-be.git] / src / main / java / org / onap / aai / sparky / synchronizer / entity / AggregationSuggestionEntity.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  *
21  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
22  */
23 package org.onap.aai.sparky.synchronizer.entity;
24
25 import com.fasterxml.jackson.databind.ObjectMapper;
26 import com.fasterxml.jackson.databind.node.ObjectNode;
27
28 import java.util.ArrayList;
29 import java.util.List;
30
31 import org.json.JSONArray;
32 import org.json.JSONObject;
33 import org.onap.aai.sparky.util.NodeUtils;
34
35 public class AggregationSuggestionEntity extends IndexableEntity implements IndexDocument {
36
37   private List<String> inputs = new ArrayList<String>();
38   /**
39    * @return the inputs
40    */
41   public List<String> getInputs() {
42     return inputs;
43   }
44
45   /**
46    * @param inputs the inputs to set
47    */
48   public void setInputs(List<String> inputs) {
49     this.inputs = inputs;
50   }
51
52   /**
53    * @return the mapper
54    */
55   public ObjectMapper getMapper() {
56     return mapper;
57   }
58
59   /**
60    * @param mapper the mapper to set
61    */
62   public void setMapper(ObjectMapper mapper) {
63     this.mapper = mapper;
64   }
65
66   /**
67    * @return the outputString
68    */
69   public String getOutputString() {
70     return outputString;
71   }
72
73   private final String outputString = "VNFs";
74   protected ObjectMapper mapper = new ObjectMapper();
75   
76   public AggregationSuggestionEntity() {
77     super();
78     inputs.add("VNFs");
79     inputs.add("generic-vnfs");
80   }
81   
82   @Override
83   public void deriveFields() {
84     this.id = NodeUtils.generateUniqueShaDigest(this.outputString);
85   }
86
87   @Override
88   public String getIndexDocumentJson() {
89     
90     JSONArray inputArray = new JSONArray();
91     for (String input: inputs) {
92       input = input.replace(",","" );
93       input = input.replace("[","" );
94       input = input.replace("]","" );
95       inputArray.put(input);
96     }
97     
98     JSONObject entitySuggest = new JSONObject();
99     entitySuggest.put("input", inputArray);
100     entitySuggest.put("output", this.outputString);
101     entitySuggest.put("weight", 100);
102     
103     JSONObject payloadNode = new JSONObject();
104     entitySuggest.put("payload", payloadNode);
105     
106     JSONObject rootNode = new JSONObject();
107     rootNode.put("entity_suggest", entitySuggest);
108
109     return rootNode.toString();
110   }
111
112   @Override
113   public ObjectNode getBulkImportEntity() {
114     // TODO Auto-generated method stub
115     return null;
116   }
117
118 }