Add new query format count and

[aai/aai-common.git] / aai-core / src / main / java / org / onap / aai / serialization / queryformats / Resource.java

diff --git a/aai-core/src/main/java/org/onap/aai/serialization/queryformats/Resource.java b/aai-core/src/main/java/org/onap/aai/serialization/queryformats/Resource.java
index 649971b..429530c 100644 (file)
--- a/aai-core/src/main/java/org/onap/aai/serialization/queryformats/Resource.java
+++ b/aai-core/src/main/java/org/onap/aai/serialization/queryformats/Resource.java
@@ -40,7 +40,7 @@ import org.onap.aai.serialization.queryformats.utils.UrlBuilder;
 import com.google.gson.JsonObject;
 import com.google.gson.JsonParser;
 
-public class Resource implements FormatMapper {
+public class Resource extends MultiFormatMapper {
 
        private final Loader loader;
        private final DBSerializer serializer;
@@ -60,8 +60,8 @@ public class Resource implements FormatMapper {
        }
 
        @Override
-       public JsonObject formatObject(Object input) throws AAIFormatVertexException {
-               Vertex v = (Vertex)input;
+       protected JsonObject getJsonFromVertex(Vertex v) throws AAIFormatVertexException {
+
                JsonObject json = new JsonObject();
                
                if (this.includeUrl) {