src/main/java/org/onap/crud/parser/CrudResponseBuilder.java

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * org.onap.aai
   4  * ================================================================================
   5  * Copyright © 2017-2018 AT&T Intellectual Property. All rights reserved.
   6  * Copyright © 2017-2018 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 package org.onap.crud.parser;
  22
  23 import java.util.ArrayList;
  24 import java.util.HashMap;
  25 import java.util.List;
  26 import java.util.Map;
  27 import org.onap.crud.entity.Edge;
  28 import org.onap.crud.entity.Vertex;
  29 import org.onap.crud.exception.CrudException;
  30 import org.onap.crud.service.BulkPayload;
  31 import org.onap.crud.service.EdgePayload;
  32 import org.onap.crud.service.VertexPayload;
  33 import org.onap.schema.EdgeRulesLoader;
  34
  35 import com.google.gson.Gson;
  36 import com.google.gson.GsonBuilder;
  37 import com.google.gson.JsonArray;
  38 import com.google.gson.JsonElement;
  39 import com.google.gson.JsonObject;
  40
  41 public class CrudResponseBuilder {
  42
  43   private static final Gson gson = new GsonBuilder().create();
  44
  45   public static final String SOURCE = "source";
  46   public static final String TARGET = "target";
  47   public static final String URL_BASE = "services/inventory/";
  48
  49   public static String buildUpsertBulkResponse(HashMap<String, Vertex> vertices, HashMap<String, Edge> edges,
  50       String version, BulkPayload incomingPayload) throws CrudException {
  51
  52     for (JsonElement e : incomingPayload.getObjects()) {
  53       List<Map.Entry<String, JsonElement>> entries = new ArrayList<Map.Entry<String, JsonElement>>(
  54           e.getAsJsonObject().entrySet());
  55
  56       Map.Entry<String, JsonElement> item = entries.get(1);
  57
  58       Vertex responseVertex = vertices.get(item.getKey());
  59       if (responseVertex != null) {
  60         JsonObject v = gson.fromJson(buildUpsertVertexResponse(responseVertex, version), JsonObject.class);
  61         item.setValue(v);
  62       } else {
  63         item.setValue(gson.fromJson("{}", JsonObject.class));
  64       }
  65
  66     }
  67     for (JsonElement e : incomingPayload.getRelationships()) {
  68       List<Map.Entry<String, JsonElement>> entries = new ArrayList<Map.Entry<String, JsonElement>>(
  69           e.getAsJsonObject().entrySet());
  70
  71       Map.Entry<String, JsonElement> item = entries.get(1);
  72
  73       Edge responseEdge = edges.get(item.getKey());
  74       if (responseEdge != null) {
  75         JsonObject v = gson.fromJson(buildUpsertEdgeResponse(responseEdge, version), JsonObject.class);
  76         item.setValue(v);
  77       } else {
  78         item.setValue(gson.fromJson("{}", JsonObject.class));
  79       }
  80
  81     }
  82     return incomingPayload.toJson();
  83   }
  84
  85   public static String buildUpsertVertexResponse(Vertex vertex, String version) throws CrudException {
  86     VertexPayload payload = new VertexPayload();
  87     payload.setId(vertex.getId().get());
  88     payload.setType(vertex.getType());
  89     payload.setUrl(URL_BASE + version + "/" + vertex.getType() + "/" + vertex.getId().get());
  90     JsonObject props = new JsonObject();
  91     for (String key : vertex.getProperties().keySet()) {
  92       addJsonProperperty(props, key, vertex.getProperties().get(key));
  93     }
  94     payload.setProperties(props);
  95     return payload.toJson();
  96   }
  97
  98   public static String buildUpsertEdgeResponse(Edge edge, String version) throws CrudException {
  99     return buildGetEdgeResponse(edge, version);
 100   }
 101
 102   public static String buildGetVertexResponse(Vertex vertex, List<Edge> edges, String version) throws CrudException {
 103     VertexPayload vertexPayload = new VertexPayload();
 104     vertexPayload.setId(vertex.getId().get());
 105     vertexPayload.setType(vertex.getType());
 106     vertexPayload.setUrl(URL_BASE + version + "/" + vertex.getType() + "/" + vertex.getId().get());
 107     JsonObject props = new JsonObject();
 108     for (String key : vertex.getProperties().keySet()) {
 109       addJsonProperperty(props, key, vertex.getProperties().get(key));
 110     }
 111     vertexPayload.setProperties(props);
 112     List<EdgePayload> inEdges = new ArrayList<EdgePayload>();
 113     List<EdgePayload> outEdges = new ArrayList<EdgePayload>();
 114     for (Edge e : edges) {
 115       if (e.getTarget().getId().get().equals(vertex.getId().get())) {
 116         EdgePayload inEdge = new EdgePayload();
 117         inEdge.setId(e.getId().get());
 118         inEdge.setType(e.getType());
 119         inEdge.setUrl(URL_BASE + "relationships/" + EdgeRulesLoader.getLatestSchemaVersion() + "/"
 120             + e.getType() + "/" + e.getId().get());
 121         inEdge.setSource(URL_BASE + version + "/" + e.getSource().getType() + "/" + e.getSource().getId().get());
 122
 123         inEdges.add(inEdge);
 124       } else if (e.getSource().getId().get().equals(vertex.getId().get())) {
 125         EdgePayload outEdge = new EdgePayload();
 126         outEdge.setId(e.getId().get());
 127         outEdge.setType(e.getType());
 128         outEdge.setUrl(URL_BASE + "relationships/" + EdgeRulesLoader.getLatestSchemaVersion() + "/"
 129             + e.getType() + "/" + e.getId().get());
 130         outEdge.setTarget(URL_BASE + version + "/" + e.getTarget().getType() + "/" + e.getTarget().getId().get());
 131         outEdges.add(outEdge);
 132       }
 133     }
 134
 135     vertexPayload.setIn(inEdges);
 136     vertexPayload.setOut(outEdges);
 137
 138     return vertexPayload.toJson();
 139   }
 140
 141   public static String buildGetVerticesResponse(List<Vertex> items, String version) throws CrudException {
 142
 143     JsonArray arry = new JsonArray();
 144     for (Vertex v : items) {
 145       JsonObject item = new JsonObject();
 146       item.addProperty("id", v.getId().get());
 147       item.addProperty("type", v.getType());
 148       item.addProperty("url", "services/inventory/" + version + "/" + v.getType() + "/" + v.getId().get());
 149       if (!v.getProperties().isEmpty()) {
 150         JsonObject propertiesObject = new JsonObject();
 151         for (String key : v.getProperties().keySet()) {
 152           propertiesObject.addProperty(key, v.getProperties().get(key).toString());
 153         }
 154         item.add("properties", propertiesObject);
 155       }
 156
 157       arry.add(item);
 158     }
 159
 160     return gson.toJson(arry);
 161   }
 162
 163   public static String buildGetEdgeResponse(Edge edge, String version) throws CrudException {
 164
 165     EdgePayload payload = new EdgePayload();
 166     payload.setId(edge.getId().get());
 167     payload.setType(edge.getType());
 168     payload.setUrl(URL_BASE + "relationships/" + version + "/" + edge.getType() + "/" + edge.getId().get());
 169     payload.setSource(URL_BASE + version + "/" + edge.getSource().getType() + "/" + edge.getSource().getId().get());
 170     payload.setTarget(URL_BASE + version + "/" + edge.getTarget().getType() + "/" + edge.getTarget().getId().get());
 171
 172     JsonObject props = new JsonObject();
 173     for (String key : edge.getProperties().keySet()) {
 174       addJsonProperperty(props, key, edge.getProperties().get(key));
 175     }
 176     payload.setProperties(props);
 177     return payload.toJson();
 178   }
 179
 180   public static String buildGetEdgesResponse(List<Edge> items, String version) throws CrudException {
 181
 182     JsonArray arry = new JsonArray();
 183     for (Edge e : items) {
 184       JsonObject item = new JsonObject();
 185       item.addProperty("id", e.getId().get());
 186       item.addProperty("type", e.getType());
 187       item.addProperty("url", URL_BASE + "relationships/" + version + "/" + e.getType() + "/" + e.getId().get());
 188       item.addProperty(SOURCE,
 189           "services/inventory/" + version + "/" + e.getSource().getType() + "/" + e.getSource().getId().get());
 190       item.addProperty(TARGET,
 191           "services/inventory/" + version + "/" + e.getTarget().getType() + "/" + e.getTarget().getId().get());
 192       arry.add(item);
 193     }
 194
 195     return gson.toJson(arry);
 196   }
 197
 198   private static void addJsonProperperty(JsonObject jsonObj, String key, Object value) {
 199     if (value instanceof Integer) {
 200       jsonObj.addProperty(key, (Integer) value);
 201     } else if (value instanceof Boolean) {
 202       jsonObj.addProperty(key, (Boolean) value);
 203     } else if (value instanceof Double) {
 204       jsonObj.addProperty(key, (Double) value);
 205     } else if (value instanceof String) {
 206       jsonObj.addProperty(key, (String) value);
 207     } else {
 208       jsonObj.addProperty(key, value.toString());
 209     }
 210   }
 211
 212 }