1 package org.onap.aai.sparky.synchronizer;
3 import static java.util.concurrent.CompletableFuture.supplyAsync;
5 import java.io.IOException;
6 import java.util.ArrayList;
7 import java.util.Collection;
9 import java.util.Map.Entry;
10 import java.util.TreeMap;
11 import java.util.concurrent.CountDownLatch;
12 import java.util.concurrent.ExecutorService;
13 import java.util.function.Supplier;
15 import org.onap.aai.cl.api.Logger;
16 import org.onap.aai.cl.eelf.LoggerFactory;
17 import org.onap.aai.restclient.client.OperationResult;
18 import org.onap.aai.restclient.enums.RestAuthenticationMode;
19 import org.onap.aai.setup.Version;
20 import org.onap.aai.sparky.config.oxm.OxmModelLoader;
21 import org.onap.aai.sparky.dal.GizmoAdapter;
22 import org.onap.aai.sparky.dal.exception.ElasticSearchOperationException;
23 import org.onap.aai.sparky.dal.rest.RestClientConstructionException;
24 import org.onap.aai.sparky.dal.rest.config.RestEndpointConfig;
25 import org.onap.aai.sparky.logging.AaiUiMsgs;
26 import org.onap.aai.sparky.util.NodeUtils;
27 import org.onap.aai.sparky.util.OxmModelAndProcessorHelper;
29 import com.fasterxml.jackson.databind.JsonNode;
30 import com.fasterxml.jackson.databind.ObjectMapper;
31 import com.fasterxml.jackson.databind.node.ArrayNode;
33 public class GizmoEntitySummarizer {
35 protected ObjectMapper mapper;
36 protected OxmModelLoader oxmModelLoader;
37 private static final Logger logger = LoggerFactory.getInstance().getLogger(GizmoEntitySummarizer.class);
38 protected ExecutorService gizmoExecutor;
39 protected GizmoAdapter gizmoAdapter;
40 protected OxmModelAndProcessorHelper oxmHelper;
43 * We need to add another concept to the OxmModelLoader which is to generate
44 * a list of entity containers from the OXM JaxbContext
47 public GizmoEntitySummarizer()
48 throws ElasticSearchOperationException, IOException, RestClientConstructionException {
50 OxmModelAndProcessorHelper.API_VERSION_OVERRIDE = Version.V11;
52 this.gizmoExecutor = NodeUtils.createNamedExecutor("GIZMO-WORKER", 5, logger);
54 oxmHelper = OxmModelAndProcessorHelper.getInstance();
55 this.oxmModelLoader = oxmHelper.getModelLoader();
57 this.mapper = new ObjectMapper();
59 RestEndpointConfig gizmoConfig = new RestEndpointConfig();
61 gizmoConfig.setEndpointIpAddress("10.147.138.153");
62 gizmoConfig.setEndpointServerPort("9520");
63 gizmoConfig.setNumRequestRetries(5);
64 gizmoConfig.setRestAuthenticationMode(RestAuthenticationMode.SSL_CERT);
65 gizmoConfig.setConnectTimeoutInMs(60000);
66 gizmoConfig.setReadTimeoutInMs(30000);
67 gizmoConfig.setCertFileName("client-cert-onap.p12");
68 gizmoConfig.setCertPassword("OBF:1y0q1uvc1uum1uvg1pil1pjl1uuq1uvk1uuu1y10");
69 gizmoConfig.setTruststoreFileName("synchronizer.jks");
70 gizmoConfig.setValidateServerCertChain(false);
71 gizmoConfig.setValidateServerHostname(false);
73 gizmoAdapter = new GizmoAdapter(oxmModelLoader, gizmoConfig);
75 gizmoAdapter.setInventoryBasePath("/services/inventory/v12/");
76 gizmoAdapter.setRelationshipsBasePath("/services/inventory/relationships/v12/");
80 private Map<String, Integer> getNumEntitiesPerType() {
82 Collection<String> containerTypes = oxmHelper.getOxmEntityContainerLookup().getEntityContainers();
83 Collection<String> links = new ArrayList<String>();
84 Map<String, Integer> entityTypeCounts = new TreeMap<String, Integer>();
86 final CountDownLatch latch = new CountDownLatch(containerTypes.size());
88 for (String entityType : containerTypes) {
90 supplyAsync(new Supplier<Void>() {
95 OperationResult typeLinksResult = null;
97 typeLinksResult = gizmoAdapter.queryGizmoWithRetries(
98 gizmoAdapter.getFullInventoryUrl(entityType), "application/json", 1);
100 if (typeLinksResult != null) {
102 if (typeLinksResult.wasSuccessful() && typeLinksResult.getResult() != null) {
104 JsonNode rootNode = mapper.readValue(typeLinksResult.getResult(), JsonNode.class);
106 if (rootNode.isArray()) {
107 ArrayNode arrayNode = (ArrayNode) rootNode;
108 entityTypeCounts.put(entityType, new Integer(arrayNode.size()));
110 entityTypeCounts.put(entityType, new Integer(-1));
115 entityTypeCounts.put(entityType, new Integer(-1));
120 } catch (Exception exc) {
121 entityTypeCounts.put(entityType, new Integer(-1));
127 }, gizmoExecutor).whenComplete((result, error) -> {
132 logger.error(AaiUiMsgs.ERROR_GENERIC,
133 "An error occurred getting data from AAI. Error = " + error.getMessage());
140 // System.out.println("self links size = " + selflinks.size());
144 } catch (InterruptedException e) {
148 return entityTypeCounts;
151 private Map<String, Integer> getNumRelationshipsPerType() {
153 Map<String, Integer> entityTypeCounts = new TreeMap<String, Integer>();
155 final CountDownLatch latch = new CountDownLatch(1);
157 supplyAsync(new Supplier<Void>() {
162 OperationResult typeLinksResult = null;
164 typeLinksResult = gizmoAdapter.queryGizmoWithRetries(gizmoAdapter.getFullRelationshipUrl("has"),
165 "application/json", 1);
167 if (typeLinksResult != null) {
169 if (typeLinksResult.wasSuccessful() && typeLinksResult.getResult() != null) {
171 JsonNode rootNode = mapper.readValue(typeLinksResult.getResult(), JsonNode.class);
173 if (rootNode.isArray()) {
174 ArrayNode arrayNode = (ArrayNode) rootNode;
175 entityTypeCounts.put("has", new Integer(arrayNode.size()));
177 entityTypeCounts.put("has", new Integer(-1));
182 entityTypeCounts.put("has", new Integer(-1));
186 entityTypeCounts.put("has", new Integer(-1));
189 } catch (Exception exc) {
190 entityTypeCounts.put("has", new Integer(-1));
196 }, gizmoExecutor).whenComplete((result, error) -> {
201 logger.error(AaiUiMsgs.ERROR_GENERIC,
202 "An error occurred getting data from AAI. Error = " + error.getMessage());
207 // System.out.println("self links size = " + selflinks.size());
211 } catch (InterruptedException e) {
215 return entityTypeCounts;
218 public void shutdown() {
219 this.gizmoExecutor.shutdown();
222 public static void main(String[] args)
223 throws ElasticSearchOperationException, IOException, RestClientConstructionException {
225 System.setProperty("CONFIG_HOME", "X:\\2018_dev\\OSEAAI\\gizmo_integration\\onap_sparky-be\\appconfig-local\\");
226 GizmoEntitySummarizer gizmoSummarizer = new GizmoEntitySummarizer();
228 Map<String, Integer> entityCounts = gizmoSummarizer.getNumEntitiesPerType();
229 Map<String, Integer> relationshipCounts = gizmoSummarizer.getNumRelationshipsPerType();
230 gizmoSummarizer.shutdown();
232 System.out.println("Gizmo Entities:");
234 for (Entry<String, Integer> entry : entityCounts.entrySet()) {
235 String key = entry.getKey();
236 Integer value = entry.getValue();
238 System.out.printf("\t%s : %d\n", key, value);
241 System.out.println("\nGizmo Relationships:");
243 for (Entry<String, Integer> entry : relationshipCounts.entrySet()) {
244 String key = entry.getKey();
245 Integer value = entry.getValue();
247 System.out.printf("\t%s : %d\n", key, value);