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

   1 /**
   2  * ============LICENSE_START=======================================================
   3  * Gizmo
   4  * ================================================================================
   5  * Copyright © 2017 AT&T Intellectual Property.
   6  * Copyright © 2017 Amdocs
   7  * All rights reserved.
   8  * ================================================================================
   9  * Licensed under the Apache License, Version 2.0 (the "License");
  10  * you may not use this file except in compliance with the License.
  11  * You may obtain a copy of the License at
  12  *
  13  *    http://www.apache.org/licenses/LICENSE-2.0
  14  *
  15  * Unless required by applicable law or agreed to in writing, software
  16  * distributed under the License is distributed on an "AS IS" BASIS,
  17  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
  18  * See the License for the specific language governing permissions and
  19  * limitations under the License.
  20  * ============LICENSE_END=========================================================
  21  *
  22  * ECOMP is a trademark and service mark of AT&T Intellectual Property.
  23  */
  24 package org.openecomp.crud.parser;
  25
  26 import java.util.ArrayList;
  27 import java.util.HashMap;
  28 import java.util.List;
  29 import java.util.Map;
  30 import java.util.Set;
  31
  32 import javax.ws.rs.core.Response.Status;
  33
  34 import org.openecomp.crud.entity.Edge;
  35 import org.openecomp.crud.entity.Vertex;
  36 import org.openecomp.crud.exception.CrudException;
  37 import org.openecomp.crud.service.BulkPayload;
  38 import org.openecomp.crud.service.EdgePayload;
  39 import org.openecomp.crud.service.VertexPayload;
  40 import org.openecomp.schema.RelationshipSchemaLoader;
  41
  42 import com.google.gson.Gson;
  43 import com.google.gson.GsonBuilder;
  44 import com.google.gson.JsonArray;
  45 import com.google.gson.JsonElement;
  46 import com.google.gson.JsonObject;
  47
  48 public class CrudResponseBuilder {
  49
  50   private static final Gson gson = new GsonBuilder().create();
  51
  52   public static final String SOURCE = "source";
  53   public static final String TARGET = "target";
  54   public static final String URL_BASE = "services/inventory/";
  55
  56   public static String buildUpsertBulkResponse(HashMap<String,Vertex> vertices, HashMap<String,Edge> edges , String version,BulkPayload incomingPayload)
  57               throws CrudException {
  58
  59                 for (JsonElement e : incomingPayload.getObjects()) {
  60                         List<Map.Entry<String, JsonElement>> entries = new ArrayList<Map.Entry<String, JsonElement>>(e.getAsJsonObject().entrySet());
  61
  62                         Map.Entry<String, JsonElement> item = entries.get(1);
  63
  64                         Vertex responseVertex = vertices.get(item.getKey());
  65                         if(responseVertex != null){
  66                         JsonObject v = gson.fromJson(buildUpsertVertexResponse(responseVertex,version), JsonObject.class);
  67                         item.setValue(v);
  68                         }else{
  69                                 item.setValue(gson.fromJson("{}", JsonObject.class));
  70                         }
  71
  72                 }
  73                 for (JsonElement e : incomingPayload.getRelationships()) {
  74                         List<Map.Entry<String, JsonElement>> entries = new ArrayList<Map.Entry<String, JsonElement>>(e.getAsJsonObject().entrySet());
  75
  76                         Map.Entry<String, JsonElement> item = entries.get(1);
  77
  78                         Edge responseEdge = edges.get(item.getKey());
  79                         if(responseEdge != null){
  80                         JsonObject v = gson.fromJson(buildUpsertEdgeResponse(responseEdge,version), JsonObject.class);
  81                         item.setValue(v);
  82                         }else{
  83                                 item.setValue(gson.fromJson("{}", JsonObject.class));
  84                         }
  85
  86                 }
  87             return incomingPayload.toJson();
  88           }
  89
  90
  91
  92   public static String buildUpsertVertexResponse(Vertex vertex, String version)
  93       throws CrudException {
  94     VertexPayload payload = new VertexPayload();
  95     payload.setId(vertex.getId().get());
  96     payload.setType(vertex.getType());
  97     payload.setUrl(URL_BASE + version + "/" + vertex.getType() + "/" + vertex.getId().get());
  98     JsonObject props = new JsonObject();
  99     for (String key : vertex.getProperties().keySet()) {
 100       addJsonProperperty(props, key, vertex.getProperties().get(key));
 101     }
 102     payload.setProperties(props);
 103     return payload.toJson();
 104   }
 105
 106   public static String buildUpsertEdgeResponse(Edge edge, String version) throws CrudException {
 107     return buildGetEdgeResponse(edge, version);
 108   }
 109
 110   public static String buildGetVertexResponse(Vertex vertex, List<Edge> edges, String version)
 111       throws CrudException {
 112     VertexPayload vertexPayload = new VertexPayload();
 113     vertexPayload.setId(vertex.getId().get());
 114     vertexPayload.setType(vertex.getType());
 115     vertexPayload.setUrl(URL_BASE + version + "/" + vertex.getType() + "/" + vertex.getId().get());
 116     JsonObject props = new JsonObject();
 117     for (String key : vertex.getProperties().keySet()) {
 118       addJsonProperperty(props, key, vertex.getProperties().get(key));
 119     }
 120     vertexPayload.setProperties(props);
 121     List<EdgePayload> inEdges = new ArrayList<EdgePayload>();
 122     List<EdgePayload> outEdges = new ArrayList<EdgePayload>();
 123     for (Edge e : edges) {
 124       if (e.getTarget().getId().get().equals(vertex.getId().get())) {
 125         EdgePayload inEdge = new EdgePayload();
 126         inEdge.setId(e.getId().get());
 127         inEdge.setType(e.getType());
 128         inEdge.setUrl(URL_BASE + "relationships/"
 129             + RelationshipSchemaLoader.getLatestSchemaVersion()
 130             + "/" + e.getType() + "/" + e.getId().get());
 131         inEdge.setSource(
 132             URL_BASE + version + "/" + e.getSource().getType() + "/" + e.getSource().getId().get());
 133
 134         inEdges.add(inEdge);
 135       } else if (e.getSource().getId().get().equals(vertex.getId().get())) {
 136         EdgePayload outEdge = new EdgePayload();
 137         outEdge.setId(e.getId().get());
 138         outEdge.setType(e.getType());
 139         outEdge.setUrl(URL_BASE + "relationships/"
 140             + RelationshipSchemaLoader.getLatestSchemaVersion()
 141             + "/" + e.getType() + "/" + e.getId().get());
 142         outEdge.setTarget(
 143             URL_BASE + version + "/" + e.getTarget().getType() + "/" + e.getTarget().getId().get());
 144         outEdges.add(outEdge);
 145       }
 146     }
 147
 148
 149     vertexPayload.setIn(inEdges);
 150     vertexPayload.setOut(outEdges);
 151
 152     return vertexPayload.toJson();
 153   }
 154
 155   public static String buildGetVerticesResponse(List<Vertex> items, String version)
 156       throws CrudException {
 157
 158     JsonArray arry = new JsonArray();
 159     for (Vertex v : items) {
 160       JsonObject item = new JsonObject();
 161       item.addProperty("id", v.getId().get());
 162       item.addProperty("type", v.getType());
 163       item.addProperty("url", "services/inventory/" + version + "/"
 164           + v.getType() + "/" + v.getId().get());
 165
 166       arry.add(item);
 167     }
 168
 169     return gson.toJson(arry);
 170   }
 171
 172   public static String buildGetEdgeResponse(Edge edge, String version) throws CrudException {
 173
 174     EdgePayload payload = new EdgePayload();
 175     payload.setId(edge.getId().get());
 176     payload.setType(edge.getType());
 177     payload.setUrl(URL_BASE + "relationships/" + version + "/" + edge.getType()
 178         + "/" + edge.getId().get());
 179     payload.setSource(
 180         URL_BASE + version + "/" + edge.getSource().getType()
 181             + "/" + edge.getSource().getId().get());
 182     payload.setTarget(
 183         URL_BASE + version + "/" + edge.getTarget().getType()
 184             + "/" + edge.getTarget().getId().get());
 185
 186     JsonObject props = new JsonObject();
 187     for (String key : edge.getProperties().keySet()) {
 188       addJsonProperperty(props, key, edge.getProperties().get(key));
 189     }
 190     payload.setProperties(props);
 191     return payload.toJson();
 192   }
 193
 194   public static String buildGetEdgesResponse(List<Edge> items, String version)
 195       throws CrudException {
 196
 197     JsonArray arry = new JsonArray();
 198     for (Edge e : items) {
 199       JsonObject item = new JsonObject();
 200       item.addProperty("id", e.getId().get());
 201       item.addProperty("type", e.getType());
 202       item.addProperty("url", URL_BASE + "relationships/" + version + "/" + e.getType()
 203           + "/" + e.getId().get());
 204       item.addProperty(SOURCE, "services/inventory/" + version + "/" + e.getSource().getType()
 205           + "/" + e.getSource().getId().get());
 206       item.addProperty(TARGET, "services/inventory/" + version + "/" + e.getTarget().getType()
 207           + "/" + e.getTarget().getId().get());
 208       arry.add(item);
 209     }
 210
 211     return gson.toJson(arry);
 212   }
 213
 214   private static void addJsonProperperty(JsonObject jsonObj, String key, Object value) {
 215     if (value instanceof Integer) {
 216       jsonObj.addProperty(key, (Integer) value);
 217     } else if (value instanceof Boolean) {
 218       jsonObj.addProperty(key, (Boolean) value);
 219     } else if (value instanceof Double) {
 220       jsonObj.addProperty(key, (Double) value);
 221     } else if (value instanceof String) {
 222       jsonObj.addProperty(key, (String) value);
 223     } else {
 224       jsonObj.addProperty(key, value.toString());
 225     }
 226   }
 227
 228 }