Index: graph/src/main/java/org/apache/hama/graph/Edge.java =================================================================== --- graph/src/main/java/org/apache/hama/graph/Edge.java (revision 1296975) +++ graph/src/main/java/org/apache/hama/graph/Edge.java (working copy) @@ -17,26 +17,33 @@ */ package org.apache.hama.graph; +/** + * The edge class + */ public class Edge { - private String name; - private String target; + private String sourceVertexID; + private String destVertexID; private int cost; - public Edge(String name,String target, int cost) { - this.name = name; - this.target = target; + public Edge(String sourceVertexID, String destVertexID, int cost) { + this.sourceVertexID = sourceVertexID; + this.destVertexID = destVertexID; this.cost = cost; } public String getName() { - return name; + return sourceVertexID; } - + + public String getDestVertexID() { + return destVertexID; + } + public int getCost() { return cost; } - public String getTarget() { - return target; + public String toString() { + return this.getDestVertexID() + ":" + this.getCost(); } } Index: graph/src/main/java/org/apache/hama/graph/Vertex.java =================================================================== --- graph/src/main/java/org/apache/hama/graph/Vertex.java (revision 1296975) +++ graph/src/main/java/org/apache/hama/graph/Vertex.java (working copy) @@ -47,7 +47,7 @@ MapWritable message = new MapWritable(); message.put(new Text(e.getName()), msg); - peer.send(e.getTarget(), message); + peer.send(e.getDestVertexID(), message); } @Override