[WIP] add remove vertex edge, TODO: tests

maven-projects/info/pom.xml

@@ -92,6 +92,11 @@
                 <version>${spotless-maven-plugin.version}</version>
                 <configuration>
                     <java>
+                        <!--<includes>
+                            <include>
+                                src/test/java/org/apache/graphar/info/GraphInfoTest.java
+                            </include>
+                        </includes>-->
                         <googleJavaFormat>
                             <version>1.7</version>
                             <style>AOSP</style>

maven-projects/info/src/main/java/org/apache/graphar/info/GraphInfo.java

@@ -136,6 +136,45 @@ public Optional<GraphInfo> addVertexAsNew(VertexInfo vertexInfo) {
                         edgeConcat2EdgeInfo));
     }
 
+    public Optional<GraphInfo> removeVertex(VertexInfo vertexInfo) {
+
+        if (vertexInfo == null || !hasVertexInfo(vertexInfo.getType())) {
+            return Optional.empty();
+        }
+        List<EdgeInfo> newCachedEdgeInfoList =
+                edgeInfos.stream()
+                        .filter(
+                                currEdge ->
+                                        !currEdge.getSrcType().equals(vertexInfo.getType())
+                                                && !currEdge.getDstType()
+                                                        .equals(vertexInfo.getType()))
+                        .collect(Collectors.toList());
+
+        final List<VertexInfo> newVertexInfoList =
+                vertexInfos.stream()
+                        .filter(
+                                v ->
+                                        !v.getType().equals(vertexInfo.getType())
+                                                && !v.getPrefix().equals(vertexInfo.getPrefix()))
+                        .collect(Collectors.toList());
+
+        final Map<String, VertexInfo> newVertexInfoMap =
+                vertexType2VertexInfo.entrySet().stream()
+                        .filter(v -> !v.getValue().getType().equals(vertexInfo.getType()))
+                        .collect(
+                                Collectors.toUnmodifiableMap(
+                                        Map.Entry::getKey, Map.Entry::getValue));
+        return Optional.of(
+                new GraphInfo(
+                        name,
+                        newVertexInfoList,
+                        newCachedEdgeInfoList,
+                        baseUri,
+                        version,
+                        newVertexInfoMap,
+                        edgeConcat2EdgeInfo));
+    }
+
     public Optional<GraphInfo> addEdgeAsNew(EdgeInfo edgeInfo) {
         if (edgeInfo == null
                 || hasEdgeInfo(
@@ -162,6 +201,39 @@ public Optional<GraphInfo> addEdgeAsNew(EdgeInfo edgeInfo) {
                         newEdgeConcat2EdgeInfo));
     }
 
+    public Optional<GraphInfo> removeEdge(EdgeInfo edgeInfo) {
+        if (edgeInfo == null
+                || hasEdgeInfo(
+                        edgeInfo.getSrcType(), edgeInfo.getEdgeType(), edgeInfo.getDstType())) {
+            return Optional.empty();
+        }
+        final List<EdgeInfo> newEdgeInfos =
+                edgeInfos.stream()
+                        .filter(
+                                e ->
+                                        !(e.getSrcType().equals(edgeInfo.getSrcType())
+                                                && e.getDstType().equals(edgeInfo.getDstType())
+                                                && e.getEdgeType().equals(edgeInfo.getEdgeType())))
+                        .collect(Collectors.toList());
+
+        final Map<String, EdgeInfo> newEdgeConcat2EdgeInfo =
+                edgeConcat2EdgeInfo.entrySet().stream()
+                        .filter(e -> !e.getKey().equals(edgeInfo.getConcat()))
+                        .filter(e -> !e.getValue().getConcat().equals(edgeInfo.getConcat()))
+                        .collect(
+                                Collectors.toUnmodifiableMap(
+                                        Map.Entry::getKey, Map.Entry::getValue));
+        return Optional.of(
+                new GraphInfo(
+                        name,
+                        vertexInfos,
+                        newEdgeInfos,
+                        baseUri,
+                        version,
+                        vertexType2VertexInfo,
+                        newEdgeConcat2EdgeInfo));
+    }
+
     public boolean hasVertexInfo(String type) {
         return vertexType2VertexInfo.containsKey(type);
     }

maven-projects/info/src/test/java/org/apache/graphar/info/GraphInfoTest.java

@@ -109,6 +109,19 @@ public void testGraphInfoBasics() {
                 illegalArgumentException.getMessage());
         // test version gar/v1
         Assert.assertEquals(1, graphInfo.getVersion().getVersion());
+        // basic tests for addVertex and removeVertex (more advanced ones should include adjancency
+        // list checks)
+        VertexInfo testingVertexInfo =
+                new VertexInfo("", 100, Arrays.asList(TestUtil.pg1), "vertex/person/", "gar/v1");
+        GraphInfo testingGraphInfo =
+                new GraphInfo("graphTest", new ArrayList<>(), new ArrayList<>(), "", "");
+        // add the created vertex on an empty graph
+        Assert.assertEquals(
+                1,
+                testingGraphInfo.addVertexAsNew(testingVertexInfo).get().getVertexInfos().size());
+        // remove the newly created vertex and check again the emptied graph
+        Assert.assertEquals(true, testingGraphInfo.removeVertex(testingVertexInfo).isEmpty());
+        // TODO (same tests as vertices for edges)
     }
 
     @Test