Mercurial > hg > Database > Christie
diff src/main/java/christie/topology/manager/FileParser.java @ 94:87a203c99177
update HostMessage refactor
author | akahori |
---|---|
date | Sat, 15 Sep 2018 18:59:12 +0900 |
parents | 7abfe041b75c |
children | ce96f375f369 |
line wrap: on
line diff
--- a/src/main/java/christie/topology/manager/FileParser.java Sat Sep 15 17:41:14 2018 +0900 +++ b/src/main/java/christie/topology/manager/FileParser.java Sat Sep 15 18:59:12 2018 +0900 @@ -11,6 +11,7 @@ import java.util.HashMap; import java.util.LinkedList; +import christie.topology.HostMessage; import com.alexmerz.graphviz.ParseException; import com.alexmerz.graphviz.Parser; import com.alexmerz.graphviz.objects.*; @@ -23,7 +24,7 @@ @Override protected void run(CodeGearManager cgm) { LinkedList<String> nodeNames = new LinkedList<>(); - HashMap<String, LinkedList<NodeInfo>> resultParse = new HashMap<>(); + HashMap<String, LinkedList<HostMessage>> resultParse = new HashMap<>(); int nodeNum = 0; try { @@ -47,20 +48,13 @@ ArrayList<Edge> edges = digraph.getEdges(); - String connection; - String source; - String target; - - NodeInfo nodeInfo; + for (Edge edge : edges) { + HostMessage nodeInfo = new HostMessage(); - for (Edge edge : edges) { - connection = edge.getAttribute("label"); - source = edge.getSource().getNode().getId().getId(); - target = edge.getTarget().getNode().getId().getId(); - nodeInfo = new NodeInfo(source, connection, target); - - resultParse.get(source).add(nodeInfo); - + nodeInfo.setNodeInfo(edge.getSource().getNode().getId().getId(), + edge.getAttribute("label"), + edge.getTarget().getNode().getId().getId()); + resultParse.get(nodeInfo.getNodeName()).add(nodeInfo); } }