refactor word count job config

mkuthan · Apr 12, 2015 · 365eff6 · 365eff6
1 parent dbbe5b9
commit 365eff6
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/src/main/scala/example/WordCountJob.scala b/src/main/scala/example/WordCountJob.scala
@@ -95,11 +95,14 @@ object WordCountJobConfig {
   import net.ceedubs.ficus.Ficus._
   import net.ceedubs.ficus.readers.ArbitraryTypeReader.arbitraryTypeValueReader
 
-  def apply(): WordCountJobConfig = apply(ConfigFactory.load())
+  def apply(applicationConfig: Option[Config] = None): WordCountJobConfig = {
 
-  def apply(applicationConfig: Config): WordCountJobConfig = {
+    val rootConfig = applicationConfig match {
+      case Some(c) => c.withFallback(ConfigFactory.load)
+      case _ => ConfigFactory.load
+    }
 
-    val config = applicationConfig.getConfig("wordCountJob")
+    val config = rootConfig.getConfig("wordCountJob")
 
     new WordCountJobConfig(
       config.as[String]("input.topic"),