add option for local connection

bcgov · Oct 16, 2024 · 1d0b335 · 1d0b335
1 parent 3b886e4
commit 1d0b335
Showing 1 changed file with 43 additions and 31 deletions.
diff --git a/R/database.R b/R/database.R
@@ -6,37 +6,49 @@
 #' @importFrom RPostgres Postgres
 #'
 #' @export
-data_connect <- function() {
-  pool <- tryCatch(
-    {
-      dbPool(
-        drv = Postgres(),
-        dbname = "climr",
-        host = "146.190.244.244",
-        port = 5432,
-        user = "climr_client",
-        password = "PowerOfBEC2023"
-      )
-    },
-    error = function(e) {
-      tryCatch(
-        {
-          dbPool(
-            drv = Postgres(),
-            dbname = "climr",
-            host = "146.190.244.244",
-            port = 5432,
-            user = "climr_client",
-            password = "PowerOfBEC2023"
-          )
-        },
-        error = function(f) {
-          warning("Could not connect to database. Will try using cached data.")
-          NULL
-        }
-      )
-    }
-  )
+data_connect <- function(local = FALSE) {
+  if(local){
+    pool <- dbPool(
+      drv = Postgres(),
+      dbname = "climr",
+      host = "localhost",
+      port = 5432,
+      user = "postgres",
+      password = "climrserver"
+    )
+  }else{
+    pool <- tryCatch(
+      {
+        dbPool(
+          drv = Postgres(),
+          dbname = "climr",
+          host = "146.190.244.244",
+          port = 5432,
+          user = "climr_client",
+          password = "PowerOfBEC2023"
+        )
+      },
+      error = function(e) {
+        tryCatch(
+          {
+            dbPool(
+              drv = Postgres(),
+              dbname = "climr",
+              host = "146.190.244.244",
+              port = 5432,
+              user = "climr_client",
+              password = "PowerOfBEC2023"
+            )
+          },
+          error = function(f) {
+            warning("Could not connect to database. Will try using cached data.")
+            NULL
+          }
+        )
+      }
+    )
+  }
+
   return(pool)
 }