src/main/java/org/onap/crud/dao/champion/ChampionEdgeSerializer.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.onap.crud.dao.champion;
  25
  26 import com.google.gson.JsonElement;
  27 import com.google.gson.JsonObject;
  28 import com.google.gson.JsonSerializationContext;
  29 import com.google.gson.JsonSerializer;
  30
  31 import org.onap.crud.entity.Edge;
  32 import org.onap.crud.entity.Vertex;
  33
  34 import java.lang.reflect.Type;
  35
  36 public class ChampionEdgeSerializer implements JsonSerializer<Edge> {
  37   @Override
  38   public JsonElement serialize(Edge edge, Type type, JsonSerializationContext jsonSerializationContext) {
  39     final JsonObject edgeObj = new JsonObject();
  40     if (edge.getId().isPresent()) {
  41       edgeObj.add("key", jsonSerializationContext.serialize(edge.getId().get()));
  42     }
  43     edgeObj.add("type", jsonSerializationContext.serialize(edge.getType()));
  44     edgeObj.add("properties", jsonSerializationContext.serialize(edge.getProperties()));
  45     edgeObj.add("source", jsonSerializationContext.serialize(edge.getSource()));
  46     edgeObj.add("target", jsonSerializationContext.serialize(edge.getTarget()));
  47     return edgeObj;
  48   }
  49 }