R/db_name.R

In skgrange/databaser: Tools to Keep Database Interaction Easy and Consistent

#' Function to get database name.
#'
#' @author Stuart K. Grange
#' 
#' @param con Database connection. 
#' 
#' @param extension For SQLite databases, should the database name include the
#' file name extension? 
#'
#' @export
db_name <- function(con, extension = TRUE) {
  
  if (db.class(con) == "postgres") {
    x <- db_get(con, "SELECT CURRENT_DATABASE()")[, 1, drop = TRUE]
  } else if (db.class(con) %in% c("mysql", "mariadb")) {
    x <- db_get(con, "SELECT DATABASE()")[, 1, drop = TRUE]
  } else if (db.class(con) == "sqlite") {
    # Get file name
    x <- basename(con@dbname)
    # Drop file extension, could be unreliable
    if (!extension) x <- stringr::str_split_fixed(x, "\\.", 2)[, 1]
  } else {
    stop("Database connection not supported.", call. = FALSE)
  }
  
  return(x)
  
}