From 0a61cafba8262b3d45330f5bb18e999d67553fd6 Mon Sep 17 00:00:00 2001
From: Dan Crankshaw <dscrankshaw@gmail.com>
Date: Thu, 31 Oct 2013 19:54:06 -0700
Subject: [PATCH] Added logging to Graph, GraphLab, and Pregel.
---
graph/src/main/scala/org/apache/spark/graph/Graph.scala | 3 ++-
graph/src/main/scala/org/apache/spark/graph/GraphLab.scala | 3 ++-
graph/src/main/scala/org/apache/spark/graph/Pregel.scala | 3 ++-
3 files changed, 6 insertions(+), 3 deletions(-)
diff --git a/graph/src/main/scala/org/apache/spark/graph/Graph.scala b/graph/src/main/scala/org/apache/spark/graph/Graph.scala
index 89e1b4ea01..f0e96896de 100644
--- a/graph/src/main/scala/org/apache/spark/graph/Graph.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/Graph.scala
@@ -3,6 +3,7 @@ package org.apache.spark.graph
import org.apache.spark.rdd.RDD
import org.apache.spark.util.ClosureCleaner
+import org.apache.spark.Logging
@@ -24,7 +25,7 @@ import org.apache.spark.util.ClosureCleaner
* @tparam VD the vertex attribute type
* @tparam ED the edge attribute type
*/
-abstract class Graph[VD: ClassManifest, ED: ClassManifest] {
+abstract class Graph[VD: ClassManifest, ED: ClassManifest] extends Logging {
/**
* Get the vertices and their data.
diff --git a/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala b/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala
index b8503ab7fd..39dc33acf0 100644
--- a/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/GraphLab.scala
@@ -2,11 +2,12 @@ package org.apache.spark.graph
import scala.collection.JavaConversions._
import org.apache.spark.rdd.RDD
+import org.apache.spark.Logging
/**
* This object implements the GraphLab gather-apply-scatter api.
*/
-object GraphLab {
+object GraphLab extends Logging {
/**
* Execute the GraphLab Gather-Apply-Scatter API
diff --git a/graph/src/main/scala/org/apache/spark/graph/Pregel.scala b/graph/src/main/scala/org/apache/spark/graph/Pregel.scala
index 1750b7f8dc..c6b5324624 100644
--- a/graph/src/main/scala/org/apache/spark/graph/Pregel.scala
+++ b/graph/src/main/scala/org/apache/spark/graph/Pregel.scala
@@ -1,6 +1,7 @@
package org.apache.spark.graph
import org.apache.spark.rdd.RDD
+import org.apache.spark.Logging
/**
@@ -41,7 +42,7 @@ import org.apache.spark.rdd.RDD
* }}}
*
*/
-object Pregel {
+object Pregel extends Logging {
/**
--
GitLab