CSIT Fix for SDC-2585
[sdc.git] / catalog-dao / src / main / java / org / openecomp / sdc / be / dao / neo4j / GraphNeighbourTable.java
1 /*-
2  * ============LICENSE_START=======================================================
3  * SDC
4  * ================================================================================
5  * Copyright (C) 2017 AT&T Intellectual Property. All rights reserved.
6  * ================================================================================
7  * Licensed under the Apache License, Version 2.0 (the "License");
8  * you may not use this file except in compliance with the License.
9  * You may obtain a copy of the License at
10  * 
11  *      http://www.apache.org/licenses/LICENSE-2.0
12  * 
13  * Unless required by applicable law or agreed to in writing, software
14  * distributed under the License is distributed on an "AS IS" BASIS,
15  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
16  * See the License for the specific language governing permissions and
17  * limitations under the License.
18  * ============LICENSE_END=========================================================
19  */
20
21 package org.openecomp.sdc.be.dao.neo4j;
22
23 import org.openecomp.sdc.be.dao.graph.datatype.GraphNode;
24
25 import java.util.ArrayList;
26 import java.util.List;
27
28 public class GraphNeighbourTable {
29
30         List<GraphNode> nodes = new ArrayList<>();
31
32         List<NodeRelation> directedEdges = new ArrayList<>();
33
34         public List<GraphNode> getNodes() {
35                 return nodes;
36         }
37
38         public void setNodes(List<GraphNode> nodes) {
39                 this.nodes = nodes;
40         }
41
42         public List<NodeRelation> getDirectedEdges() {
43                 return directedEdges;
44         }
45
46         public void setDirectedEdges(List<NodeRelation> directedEdges) {
47                 this.directedEdges = directedEdges;
48         }
49
50         public int addNode(GraphNode node) {
51                 this.nodes.add(node);
52                 return this.nodes.size() - 1;
53         }
54
55         public void addEdge(NodeRelation directedEdge) {
56                 this.directedEdges.add(directedEdge);
57         }
58
59         @Override
60         public String toString() {
61                 return "GraphNeighbourTable [nodes=" + nodes + ", directedEdges=" + directedEdges + "]";
62         }
63
64 }