在使用dbplyr创建函数时,使用return()以整洁的样式

时间:2017-09-08 15:55:51

标签: r dplyr tidyverse dbplyr

我想弄清楚如何将tidy style guide应用于我正在创建的某些功能。具体来说,我试图限制我使用return() - 根据风格指南建议仅用于早期返回的情况 - 在dbplyr函数的上下文中。在阅读样式指南之前,我会这样做(你需要dplyrdbplyr来运行它):

cars_return <- function(){
  con <- DBI::dbConnect(RSQLite::SQLite(), ":memory:")
  dplyr::copy_to(con, mtcars)

  mtcars2 <- dplyr::tbl(con, "mtcars")

  mtcars2 <- dplyr::collect(mtcars2)

  return(mtcars2)

  DBI::dbDisconnect(con)
}

哪会归还这个:

> cars_return()
# A tibble: 32 x 11
     mpg   cyl  disp    hp  drat    wt  qsec    vs    am  gear  carb
   <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl> <dbl>
 1  21.0     6 160.0   110  3.90 2.620 16.46     0     1     4     4
 2  21.0     6 160.0   110  3.90 2.875 17.02     0     1     4     4
 3  22.8     4 108.0    93  3.85 2.320 18.61     1     1     4     1
 4  21.4     6 258.0   110  3.08 3.215 19.44     1     0     3     1
 5  18.7     8 360.0   175  3.15 3.440 17.02     0     0     3     2
 6  18.1     6 225.0   105  2.76 3.460 20.22     1     0     3     1
 7  14.3     8 360.0   245  3.21 3.570 15.84     0     0     3     4
 8  24.4     4 146.7    62  3.69 3.190 20.00     1     0     4     2
 9  22.8     4 140.8    95  3.92 3.150 22.90     1     0     4     2
10  19.2     6 167.6   123  3.92 3.440 18.30     1     0     4     4
# ... with 22 more rows

但是,如果我尝试省略return()这样的话:

cars_no_return <- function(){
  con <- DBI::dbConnect(RSQLite::SQLite(), ":memory:")
  dplyr::copy_to(con, mtcars)

  mtcars2 <- dplyr::tbl(con, "mtcars")

  dplyr::collect(mtcars2)

  DBI::dbDisconnect(con)
}

然后,当我调用该函数时,我什么都没有回复:

> cars_no_return()
>

所以我的问题是,如何在尊重tidy style guide的同时使用collect()处理需要从数据库返回data.frame的函数。

1 个答案:

答案 0 :(得分:2)

您应该查看on.exit。例如,请参阅How and when should I use on.exit?

cars_return <- function(){
  con <- DBI::dbConnect(RSQLite::SQLite(), ":memory:")

  on.exit(DBI::dbDisconnect(con))

  dplyr::copy_to(con, mtcars)

  mtcars2 <- dplyr::tbl(con, "mtcars")

  mtcars2 <- dplyr::collect(mtcars2)

  mtcars2
}