From 00e73833cce88d5f77eaadd741a9a4046a4f17a3 Mon Sep 17 00:00:00 2001
From: "Joseph E. Gonzalez" <joseph.e.gonzalez@gmail.com>
Date: Sat, 26 Oct 2013 15:10:30 -0700
Subject: [PATCH] Fixing a bug in reverse edge direction.
---
.../main/scala/org/apache/spark/graph/EdgeDirection.scala | 8 ++++++--
1 file changed, 6 insertions(+), 2 deletions(-)
diff --git a/graph/src/main/scala/org/apache/spark/graph/EdgeDirection.scala b/graph/src/main/scala/org/apache/spark/graph/EdgeDirection.scala
index 99af2d5458..a1468a152b 100644
--- a/graph/src/main/scala/org/apache/spark/graph/EdgeDirection.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/EdgeDirection.scala
@@ -6,9 +6,13 @@ package org.apache.spark.graph
* the set of adjacent neighbors when running a neighborhood query.
*/
sealed abstract class EdgeDirection {
+ /**
+ * Reverse the direction of an edge. An in becomes out,
+ * out becomes in and both remains both.
+ */
def reverse: EdgeDirection = this match {
- case EdgeDirection.In => EdgeDirection.In
- case EdgeDirection.Out => EdgeDirection.Out
+ case EdgeDirection.In => EdgeDirection.Out
+ case EdgeDirection.Out => EdgeDirection.In
case EdgeDirection.Both => EdgeDirection.Both
}
}
--
GitLab