Add new query format count and

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

diff --git a/aai-core/src/main/java/org/onap/aai/serialization/queryformats/IdURL.java b/aai-core/src/main/java/org/onap/aai/serialization/queryformats/IdURL.java
index 320cd61..4395c8c 100644 (file)
--- a/aai-core/src/main/java/org/onap/aai/serialization/queryformats/IdURL.java
+++ b/aai-core/src/main/java/org/onap/aai/serialization/queryformats/IdURL.java
@@ -32,7 +32,7 @@ import org.onap.aai.introspection.exceptions.AAIUnknownObjectException;
 import org.onap.aai.serialization.queryformats.exceptions.AAIFormatVertexException;
 import org.onap.aai.serialization.queryformats.utils.UrlBuilder;
 
-public class IdURL implements FormatMapper {
+public class IdURL extends MultiFormatMapper {
 
        private final UrlBuilder urlBuilder;
        private final JsonParser parser;
@@ -50,8 +50,8 @@ public class IdURL implements FormatMapper {
        }
 
        @Override
-       public JsonObject formatObject(Object input) throws AAIFormatVertexException {
-               Vertex v = (Vertex)input;
+       protected JsonObject getJsonFromVertex(Vertex v) throws AAIFormatVertexException {
+
                try {
                        final Introspector searchResult = this.loader.introspectorFromName("result-data");