使用 dbplyr 创建函数时以简洁的方式使用 return()

Question

我正在尝试弄清楚如何将 tidy style guide 应用于我正在创建的某些函数。具体来说，我试图限制我对 return() 的使用——根据风格指南的建议，仅在早期 returns 的情况下使用——在 dbplyr 函数的上下文中。在阅读风格指南之前，我会这样做（你需要 dplyr 和 dbplyr 到运行这个）：

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)
}

return编辑这个：

> 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)
}

然后当我调用那个函数时，我什么也得不到 returned:

> cars_no_return()
>

所以我的问题是，如何使用 collect() 处理需要 return 来自数据库的 data.frame 的函数，同时尊重 tidy style guide.

Answer 1

你应该调查一下 on.exit。例如参见 [=12=]

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
}

使用 dbplyr 创建函数时以简洁的方式使用 return()

Use of return() in tidy style when creating functions with dbplyr

r

dplyr

tidyverse

dbplyr