1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
|
#' Refer to a table in a schema or a database catalog
#'
#' `in_schema()` can be used in [tbl()] to indicate a table in a specific
#' schema.
#' `in_catalog()` additionally allows specifying the database catalog.
#'
#' @param catalog,schema,table Names of catalog, schema, and table.
#' These will be automatically quoted; use [sql()] to pass a raw name
#' that won't get quoted.
#' @export
#' @examples
#' in_schema("my_schema", "my_table")
#' in_catalog("my_catalog", "my_schema", "my_table")
#' # eliminate quotes
#' in_schema(sql("my_schema"), sql("my_table"))
#'
#' # Example using schemas with SQLite
#' con <- DBI::dbConnect(RSQLite::SQLite(), ":memory:")
#'
#' # Add auxilary schema
#' tmp <- tempfile()
#' DBI::dbExecute(con, paste0("ATTACH '", tmp, "' AS aux"))
#'
#' library(dplyr, warn.conflicts = FALSE)
#' copy_to(con, iris, "df", temporary = FALSE)
#' copy_to(con, mtcars, in_schema("aux", "df"), temporary = FALSE)
#'
#' con %>% tbl("df")
#' con %>% tbl(in_schema("aux", "df"))
in_schema <- function(schema, table) {
structure(
list(
schema = as.sql(schema),
table = as.sql(table)
),
class = "dbplyr_schema"
)
}
#' @rdname in_schema
#' @export
in_catalog <- function(catalog, schema, table) {
structure(
list(
schema = as.sql(schema),
table = as.sql(table),
catalog = as.sql(catalog)
),
class = "dbplyr_catalog"
)
}
#' @export
print.dbplyr_schema <- function(x, ...) {
cat_line("<SCHEMA> ", escape_ansi(x$schema), ".", escape_ansi(x$table))
}
#' @export
print.dbplyr_catalog <- function(x, ...) {
cat_line("<CATALOG> ", escape_ansi(x$catalog), ".", escape_ansi(x$schema), ".", escape_ansi(x$table))
}
#' @export
as.sql.dbplyr_schema <- function(x, con) {
ident_q(paste0(escape(x$schema, con = con), ".", escape(x$table, con = con)))
}
#' @export
as.sql.dbplyr_catalog <- function(x, con) {
ident_q(paste0(
escape(x$catalog, con = con), ".", escape(x$schema, con = con), ".", escape(x$table, con = con)
))
}
is_schema <- function(x) inherits(x, "dbplyr_schema")
is_catalog <- function(x) inherits(x, "dbplyr_catalog")
# Support for DBI::Id() ---------------------------------------------------
#' @export
as.sql.Id <- function(x, con) ident_q(dbQuoteIdentifier(con, x))
# Old dbplyr approach -----------------------------------------------------
#' Declare a identifer as being pre-quoted.
#'
#' No longer needed; please use [sql()] instead.
#'
#' @keywords internal
#' @export
ident_q <- function(...) {
x <- c_character(...)
structure(x, class = c("ident_q", "ident", "character"))
}
#' @export
escape.ident_q <- function(x, parens = FALSE, collapse = ", ", con = NULL) {
sql_vector(names_to_as(x, names2(x), con = con), parens, collapse, con = con)
}
#' @export
dbi_quote.ident_q <- function(x, con) DBI::SQL(as.character(x))
|
