当连接到Oracle数据库时，如何使用dbplyr进行EXTRACT操作

Question

请看这个查询：

SELECT EXTRACT(month FROM order_date) "Month"
  FROM orders

（来自官方oracle文档的简化示例）

如何在dbplyr链中集成上述EXTRACT操作？

我可以接受任何其他解决方法（即使是丑陋/昂贵的），以从服务器端提取月份。

- moodymudskipper

---

mutate(Month = sql('extract(month from order_date)') 对我很有效。 - pentandrous

Answer 1

更加优雅：

tbl(con, "orders") %>% mutate(Month = extract(NULL %month from% order_date))

这导致以下 SQL（ANSI SQL）的结果：

EXTRACT( MONTH FROM "order_date")

这个技巧是行得通的，因为操作符的名称（在百分号之间）被直接翻译为 SQL。 NULL 会消失（不像 NA）。

- Fabianius

---

这将非常有用，谢谢。该功能的文档不是非常明确，但可以在?dbplyr::translate_sql的示例中看到。 - moodymudskipper

Answer 2

与此同时，我想到了一些东西。

通过执行以下代码，可以获得给定示例的预期输出：

con <- ROracle::dbConnect(drv, username, password, dbname) # your connection parameters
dplyr::tbl(con,"orders") %>%
  extract_o("Month","order_date",append = FALSE,force_upper_case = FALSE)

这是函数的代码，我包含了一些参数来强制大写列（默认）并将新列附加到现有列（默认）。新列的名称可以定义，或者默认情况下将命名为您想要提取的值的类型。

#' use Oracle EXTRACT function
#' 
#' Will add a column to the table, containing extracted value,
#' optionally returns only this column
#' @param data tbl_lazy object
#' @param what type of data to extract
#' @param from column to extract from
#' @param new_col name of new column
#' @param append keep existing columns,
#' FALSE ditches them and keep only extracted column
#' @param force_upper_case make new column name uppercase
extract_o <-function(data, what, from, new_col = what,
                     append = TRUE,force_upper_case = TRUE) {
  allowed <- c("day","month","year","hour","minute","second",
                     "timezone_hour","timezone_minute",
                     "timezone_region","timezone_abbr")
  assertthat::assert_that(
    tolower(what) %in% allowed,
    msg=paste("Choose 'what' among",
              paste0("'",allowed,"'",collapse=", ")))
  if(force_upper_case) new_col <- toupper(new_col)
  tbl_query <- as.character(dbplyr::sql_render(data)) # previous query
  append_sql <- if(append)
    paste0(paste(colnames(data),collapse=", "),", ") else ""
  query <- paste0("SELECT ", append_sql,                         # initial cols or none
                  "EXTRACT(",what," FROM ",from,") \"",new_col,  # new col
                  "\" FROM (",tbl_query,")")                     # previous query
  dplyr::tbl(data$src$con,sql(query))
}

- moodymudskipper