Implement dbExecute() method

NAMESPACE

@@ -122,8 +122,10 @@ exportMethods(dbClearResult)
 exportMethods(dbColumnInfo)
 exportMethods(dbCommit)
 exportMethods(dbConnect)
+exportMethods(dbCreateTable)
 exportMethods(dbDataType)
 exportMethods(dbDisconnect)
+exportMethods(dbExecute)
 exportMethods(dbExistsTable)
 exportMethods(dbFetch)
 exportMethods(dbGetInfo)

NEWS.md

@@ -1,6 +1,7 @@
 # bigrquery (development version)
 
-* Now supports `dbAppendTable()` (#539) and `dbCreateTable()` (#483).
+* Now supports `dbAppendTable()` (#539), `dbCreateTable()` (#483),
+  and `dbExecute` (#502).
 
 * `dbWriteTable()` now correct uses the `billing` value set in the 
   connection (#486).

R/dbi-connection.R

@@ -77,6 +77,26 @@ setMethod(
     BigQueryResult(conn, statement, ...)
   })
 
+
+#' @rdname DBI
+#' @inheritParams DBI::dbSendQuery
+#' @export
+setMethod("dbExecute", c("BigQueryConnection", "character"), function(conn, statement, ...) {
+  ds <- if (!is.null(conn@dataset)) as_bq_dataset(conn)
+
+  job <- bq_perform_query(statement,
+    billing = conn@billing,
+    default_dataset = ds,
+    quiet = conn@quiet,
+    ...
+  )
+  bq_job_wait(job, quiet = conn@quiet)
+
+  meta <- bq_job_meta(job, "statistics(query(numDmlAffectedRows))")
+  as.numeric(meta$statistics$query$numDmlAffectedRows %||% 0)
+})
+
+
 #' @rdname DBI
 #' @inheritParams DBI::dbQuoteString
 #' @export

tests/testthat/test-dbi-connection.R

@@ -70,6 +70,18 @@ test_that("can roundtrip a data frame", {
   expect_equal(ncol(df), 11)
 })
 
+test_that("can execute a query", {
+  tb <- bq_test_table()
+  con <- DBI::dbConnect(bq_dataset(tb$project, tb$dataset))
+
+  DBI::dbWriteTable(con, tb$table, data.frame(x = 1:4))
+  out <- dbExecute(con, glue("UPDATE {tb$table} SET x = x + 1 WHERE true"))
+  expect_equal(out, 4)
+
+  out <- dbExecute(con, glue("DELETE {tb$table} WHERE x <= 3"))
+  expect_equal(out, 2)
+})
+
 test_that("can use DBI::Id()", {
   ds <- bq_test_dataset()
   con <- DBI::dbConnect(ds)