add a util method for changing the log level while running

apache · Sep 17, 2014 · 44d6577 · 44d6577
1 parent 86d253e
commit 44d6577
Show file tree

Hide file tree

Showing 2 changed files with 19 additions and 1 deletion.
diff --git a/core/src/main/scala/org/apache/spark/util/Utils.scala b/core/src/main/scala/org/apache/spark/util/Utils.scala
@@ -1452,6 +1452,14 @@ private[spark] object Utils extends Logging {
     }
   }
 
+  /**
+   * configure a new log4j level
+   */
+  def setLogLevel(l: org.apache.log4j.Level) {
+    org.apache.log4j.Logger.getRootLogger().setLevel(l)
+  }
+
+
   /**
    * config a log4j properties used for testsuite
    */

diff --git a/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala b/core/src/test/scala/org/apache/spark/util/UtilsSuite.scala
@@ -27,7 +27,9 @@ import com.google.common.base.Charsets
 import com.google.common.io.Files
 import org.scalatest.FunSuite
 
-class UtilsSuite extends FunSuite {
+import org.apache.spark.Logging
+
+class UtilsSuite extends FunSuite with Logging {
 
   test("bytesToString") {
     assert(Utils.bytesToString(10) === "10.0 B")
@@ -297,4 +299,12 @@ class UtilsSuite extends FunSuite {
     }
   }
 
+  // Test for using the util function to change our log levels.
+  test("log4j log level change") {
+    Utils.setLogLevel(org.apache.log4j.Level.ALL)
+    assert(log.isInfoEnabled())
+    Utils.setLogLevel(org.apache.log4j.Level.ERROR)
+    assert(!log.isInfoEnabled())
+    assert(log.isErrorEnabled())
+  }
 }