move force statement in sample sql to sql preview api

rstudio · Jun 11, 2018 · 8e40405 · 8e40405
1 parent 77b2d29
commit 8e40405
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/src/cpp/session/modules/SessionDataPreview.R b/src/cpp/session/modules/SessionDataPreview.R
@@ -44,6 +44,9 @@
    # remove comments since some drivers might not support them
    statement <- gsub("--[^\n]*\n+", "", statement)
 
+   # force the connection to let DBI and others initialize S3
+   conn <- force(conn)
+
    # fetch at most 100 records as a preview
    rs <- DBI::dbSendQuery(conn, statement = statement, ...)
    data <- DBI::dbFetch(rs, n = 1000)

diff --git a/src/cpp/session/resources/templates/query.sql b/src/cpp/session/resources/templates/query.sql
@@ -1,3 +1,3 @@
--- !preview conn=force(DBI::dbConnect(RSQLite::SQLite()))
+-- !preview conn=DBI::dbConnect(RSQLite::SQLite())
 
 SELECT 1