Wrap more dbplyr generics (#175)

Fixes #171. Fixes #172.

And wrap `sql_query_fields` to simplify `tbl.pool`

NEWS.md

@@ -1,5 +1,8 @@
 # pool (development version)
 
+* Add wrappers for dbplyr generics `db_col_types()` (#171) and 
+  `db_copy_to()` (#172).
+
 * Pool no longer generates spurious messages about needing to use 
   `in_schema()` or avoiding the use of `ident_q()`.
 

R/dbplyr.R

@@ -23,13 +23,6 @@
 #'
 #' poolClose(pool)
 tbl.Pool <- function(src, from, ..., vars = NULL) {
-  con <- poolCheckout(src)
-  on.exit(poolReturn(con))
-
-  if (is.null(vars)) {
-    vars <- dplyr::db_query_fields(con, from)
-  }
-
   dbplyr::tbl_sql("Pool", dbplyr::src_dbi(src), from, ..., vars = vars)
 }
 
@@ -93,10 +86,13 @@ dbplyr_register_methods <- function() {
     dbplyr_s3_register("db_collect")
     dbplyr_s3_register("db_compute")
     dbplyr_s3_register("db_connection_describe")
+    dbplyr_s3_register("db_copy_to")
+    dbplyr_s3_register("db_col_types")
     dbplyr_s3_register("db_sql_render")
     dbplyr_s3_register("sql_translation")
     dbplyr_s3_register("sql_join_suffix")
     dbplyr_s3_register("sql_query_explain")
+    dbplyr_s3_register("sql_query_fields")
   })
 }
 

tests/testthat/test-dbplyr.R

@@ -13,6 +13,14 @@ test_that("can copy and collect", {
 
   # But copy_to must not be temporary
   expect_snapshot(dplyr::copy_to(pool, df), error = TRUE)
+
+  db2 <- dplyr::auto_copy(
+    db,
+    tibble::tibble(x = 11),
+    copy = TRUE,
+    temporary = FALSE
+  )
+  expect_s3_class(dbplyr::remote_con(db2), "Pool")
 })
 
 test_that("can use one-table verbs", {