handled db name error

jpbnetley · Apr 19, 2020 · 7d8bc4d · 7d8bc4d
1 parent 3b0f622
commit 7d8bc4d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/scala/Util/Database/Database.scala b/src/main/scala/Util/Database/Database.scala
@@ -54,7 +54,7 @@ case object Database {
     }
   }
 
-  private val databaseName: String                  = sys.env("mongo_db_name")
+  private val databaseName: String                  = sys.env.get("mongo_db_name").fold(throw error("environment variables not set for db name"))(identity)
   private val settings: MongoClientSettings.Builder = settingsBuilder().fold(throw error("environment variables not set for db"))(identity)
   private val mongoClient: MongoClient              = MongoClient(settings.build())
   private val database: MongoDatabase               = mongoClient.getDatabase(databaseName)