R/db_connect.R

In vwrobel/dataexpr: All Dataexp tools for R

#' Connect to the Mongo Database
#'
#' @param collection_name Name of the collection
#'
#' @return Object of class mongo
#' @export

connect_db <- function(collection_name) {
  env <- Sys.getenv("env")
  if (env == "dev") {
    mongo_url <- sprintf("mongodb://%s",
                         Sys.getenv("MONGO_DEV_DB"))
  } else {
    mongo_url <- sprintf(
      "mongodb://%s:%s@%s",
      Sys.getenv("MLAB_USER"),
      Sys.getenv("MLAB_PWD"),
      Sys.getenv(sprintf("MONGO_%s_DB", toupper(env)))
    )
  }
  db_collection <- mongolite::mongo(collection = collection_name,
                                    url = mongo_url)
  return(db_collection)
}