adding collect argument to TblSqlSource execute_query method

pkg-r/R/TblSqlSource.R

@@ -117,10 +117,18 @@ TblSqlSource <- R6::R6Class(
     #' Execute a SQL query and return results
     #'
     #' @param query SQL query string to execute
-    #' @return A data frame containing query results
-    execute_query = function(query) {
+    #' @param collect If `TRUE`, collects the results into a local data frame
+    #'   using [dplyr::collect()]. If `FALSE` (default), returns a lazy SQL
+    #'   tibble.
+    #' @return A data frame (if `collect = TRUE`) or a lazy SQL tibble (if
+    #'   `collect = FALSE`)
+    execute_query = function(query, collect = FALSE) {
       sql_query <- self$prep_query(query)
-      dplyr::tbl(private$conn, dplyr::sql(sql_query))
+      result <- dplyr::tbl(private$conn, dplyr::sql(sql_query))
+      if (collect) {
+        result <- dplyr::collect(result)
+      }
+      result
     },
 
     #' @description

pkg-r/tests/testthat/test-TblSqlSource.R

@@ -35,6 +35,31 @@ describe("TblSqlSource$new()", {
     expect_equal(collected$value, c(30, 40, 50))
   })
 
+  it("returns lazy tibble from execute_query() when collect = FALSE", {
+    source <- local_tbl_sql_source()
+
+    result <- source$execute_query(
+      "SELECT * FROM test_table WHERE value > 25",
+      collect = FALSE
+    )
+    expect_s3_class(result, "tbl_sql")
+    expect_s3_class(result, "tbl_lazy")
+  })
+
+  it("returns data frame from execute_query() when collect = TRUE", {
+    source <- local_tbl_sql_source()
+
+    result <- source$execute_query(
+      "SELECT * FROM test_table WHERE value > 25",
+      collect = TRUE
+    )
+    expect_s3_class(result, "data.frame")
+    expect_false(inherits(result, "tbl_sql"))
+    expect_false(inherits(result, "tbl_lazy"))
+    expect_equal(nrow(result), 3)
+    expect_equal(result$value, c(30, 40, 50))
+  })
+
   it("returns data frame from test_query()", {
     source <- local_tbl_sql_source()