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