Sharing the connection string among databases

Author: pierre-leonard

src/main/kotlin/EndpointToCheck.kt

@@ -1 +1 @@
-data class EndpointToCheck(val dbname: String, val endpoint: String)
+data class EndpointToCheck(val dbname: String, val dblabel: String)

src/main/kotlin/MonitorService.kt

@@ -13,7 +13,9 @@ import org.omg.CORBA.Environment
 import sun.misc.ObjectInputFilter
 import java.io.File
 import java.net.URL
+import java.sql.Connection
 import java.sql.DriverManager
+import java.sql.Statement
 import kotlin.properties.Delegates
 
 
@@ -24,8 +26,10 @@ object MonitorService{
 
     fun start(config : Config)= runBlocking {
         try{
-            config.getConfigList("databases").forEach {
-                endpointToCheck ->  checkIfDbIfReachable(endpointToCheck.getString("dbname"), endpointToCheck.getString("endpoint"))
+            DriverManager.getConnection(config.getString("endpoint")).use {
+                config.getConfigList("databases").forEach {
+                        endpointToCheck ->  checkIfDbIfReachable(endpointToCheck.getString("dblabel"), endpointToCheck.getString("dbname"), it)
+                }
             }
         }
         catch (e: Exception){
@@ -47,14 +51,15 @@ object MonitorService{
     /**
      * Throws an exception if can't reach the db
      */
-    private suspend fun checkIfDbIfReachable(dbname : String, url : String){
+    private suspend fun checkIfDbIfReachable(dblabel : String, dbname : String, connection : Connection){
         withContext(Dispatchers.IO) {
             try {
-                val connection = DriverManager.getConnection(url)
+                val stmt: Statement = connection.createStatement()
+                stmt.execute("use [$dbname]")
+                stmt.close()
                 assert(connection.isValid(5000))
-                connection.close()
             } catch (e: Exception) {
-                throw Exception("Connection to DB $dbname failed")
+                throw Exception("Connection to DB $dblabel failed")
             }
         }
     }

src/main/resources/environment.conf

@@ -1,13 +1,17 @@
 slackbridge = ${?SLACK_BRIDGE}
 errormessage = "Alert from db monitor:"
-
+endpoint = ${?TRANSITDATA_PUBTRANS_CONN_STRING}
 databases = [
     {
-      dbname = "OMM DB"
-      endpoint = ${?OMM_DB_ENDPOINT}
+      dblabel = "OMM DB"
+      dbname = "OMM_Community"
     },
     {
-      dbname = "LIJ DB"
-      endpoint = ${?LIJ_DB_ENDPOINT}
+      dblabel = "ROI DB"
+      dbname = "ptROI_Community"
+    },
+    {
+      dblabel = "DOI DB"
+      dbname = "ptDOI4_Community"
     }
 ]