Reading db connection string from the secet file

Author: pierre-leonard

src/main/kotlin/fi/hsl/transitdata/dbmonitor/Main.kt

@@ -1,8 +1,25 @@
 package fi.hsl.transitdata.dbmonitor
 
 import fi.hsl.transitdata.dbmonitor.config.ConfigParser
+import java.io.File
+import java.sql.Connection
+import java.sql.DriverManager
+import java.sql.SQLException
+import java.util.*
 
 fun main() {
     val config = ConfigParser.createConfig()
-    MonitorService.start(config)
+    MonitorService.start(config, createPubtransConnection())
+}
+
+@Throws(Exception::class)
+private fun createPubtransConnection(): Connection {
+   //Default path is what works with Docker out-of-the-box. Override with a local file if needed
+    val secretFilePath: String = System.getenv("FILEPATH_CONNECTION_STRING") ?: "/run/secrets/pubtrans_community_conn_string"
+    val connectionString = Scanner(File(secretFilePath))
+        .useDelimiter("\\Z").next()
+    if (connectionString.isEmpty()) {
+        throw Exception("Failed to find Pubtrans connection string, exiting application")
+    }
+    return DriverManager.getConnection(connectionString)
 }

src/main/kotlin/fi/hsl/transitdata/dbmonitor/MonitorService.kt

@@ -18,9 +18,9 @@ object MonitorService{
     private val f = JsonFactory()
     private val mapper: ObjectMapper = ObjectMapper(f).registerModule(KotlinModule())
 
-    fun start(config : Config)= runBlocking {
+    fun start(config : Config, connection : Connection)= runBlocking {
         try{
-            DriverManager.getConnection(config.getString("endpoint")).use {
+            connection.use {
                 config.getConfigList("databases").forEach {
                         endpointToCheck ->  checkIfDbIfReachable(endpointToCheck.getString("dblabel"), endpointToCheck.getString("dbname"), it)
                 }

src/main/resources/environment.conf

@@ -1,6 +1,5 @@
 slackbridge = ${?SLACK_BRIDGE}
 errormessage = "Alert from db monitor:"
-endpoint = ${?TRANSITDATA_PUBTRANS_CONN_STRING}
 databases = [
     {
       dblabel = "OMM DB"