inst/metadata/entity/entity_handler_dbi.R

In geoflow: Orchestrate Geospatial (Meta)Data Management Workflows and Manage FAIR Services

#handle_entities_dbi
handle_entities_dbi <- function(handler, source, config, handle = TRUE){
  dbi <- config$software$input$dbi
  dbi_config <- config$software$input$dbi_config
  if(is.null(dbi)){
    stop("There is no database input software configured to handle entities from DB")
  }
  
  out_query = NULL
  
  #db source
  is_default_source = FALSE
  query_all_tables = FALSE
  is_query <- if(!is.null(source)) startsWith(tolower(source), "select ") else FALSE
  if(is_query){
    out_query <- try(DBI::dbGetQuery(dbi, source), silent = TRUE)
    if(is.null(out_query) | (!is.null(out_query) & is(out_query,"try-error"))){
      errMsg <- sprintf("Error while trying to execute DB query '%s'.", source)
      config$logger$ERROR(errMsg)
      stop(errMsg)
    }
  }else{
    if(is.null(source) | (!is.null(source) && source == "")){
      source = if(!is.null(dbi_config$properties$metadata_table) && nzchar(dbi_config$properties$metadata_table)){
        dbi_config$properties$metadata_table
      }else{
        "public.metadata"
      }
      config$logger$INFO("Source is null or empty. Use default metadata table '%s'", source)
      is_default_source = TRUE
    }
    out_query <- try(DBI::dbGetQuery(dbi, sprintf("select * from %s", source)), silent = TRUE)
    if(is.null(out_query) | (!is.null(out_query) & is(out_query,"try-error"))){
      if(is_default_source){
        query_all_tables <- TRUE
        warnMsg <- sprintf("Error while trying to read DB default table '%s' (table not in DB).", source)
        config$logger$WARN(warnMsg)
      }else{
        errMsg <- sprintf("Error while trying to read DB table/view '%s'. Check if it exists in DB.", source)
        config$logger$ERROR(errMsg)
        stop(errMsg)
      }
    }
  }
  if(!handle) return(source)
  
  #apply handlers
  entities <- if(query_all_tables){
    config$logger$INFO("Switch to DBI tables entity handler")
    #use a generic DBI geometry columns 
    source = dbi_config$parameters$dbname
    handle_entities_dbi_geometry_columns <-  source(system.file("metadata/entity", "entity_handler_dbi_geometry_columns.R", package = "geoflow"))$value
    handle_entities_dbi_geometry_columns(handler, source, config)
  }else{
    config$logger$INFO("Use default tabular entity handler")
    #use the df entity handler based on the SQL query/table specified as source (with DBI data enricher)
    handle_entities_dbi_df <- source(system.file("metadata/entity", "entity_handler_dbi_df.R", package = "geoflow"))$value
    handle_entities_dbi_df(handler, source = out_query, config)
  }
  return(entities)
}