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);
                 }
             }