mirror of https://git.sr.ht/~hrbrmstr/metis
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
71 lines
1.7 KiB
71 lines
1.7 KiB
4 years ago
|
list(
|
||
|
|
"-7" = as.logical, # BIT
|
||
|
|
"-6" = as.integer, # TINYINT
|
||
|
|
"-5" = bit64::as.integer64, # BIGINT
|
||
|
|
"-4" = as.character, # LONGVARBINARY
|
||
|
|
"-3" = as.character, # VARBINARY
|
||
|
|
"-2" = as.character, # BINARY
|
||
|
|
"-1" = as.character, # LONGVARCHAR
|
||
|
|
"0" = as.character, # NULL
|
||
|
|
"1" = as.character, # CHAR
|
||
|
|
"2" = as.double, # NUMERIC
|
||
|
|
"3" = as.double, # DECIMAL
|
||
|
|
"4" = as.integer, # INTEGER
|
||
|
|
"5" = as.integer, # SMALLINT
|
||
|
|
"6" = as.double, # FLOAT
|
||
|
|
"7" = as.double, # REAL
|
||
|
|
"8" = as.double, # DOUBLE
|
||
|
|
"12" = as.character, # VARCHAR
|
||
|
|
"16" = as.logical, # BOOLEAN
|
||
|
|
"91" = as_date, # DATE
|
||
|
|
"92" = as.character, # TIME
|
||
|
|
"93" = as_posixct, # TIMESTAMP
|
||
|
|
"1111" = as.character # OTHER
|
||
|
|
) -> .jdbc_converters
|
||
|
|
|
||
|
|
#' AthenaJDBC
|
||
|
|
#'
|
||
|
|
#' @param conn Athena connection
|
||
|
|
#' @param statement SQL statement
|
||
|
|
#' @param ... unused
|
||
|
|
#' @importFrom rJava .jcall
|
||
|
|
#' @export
|
||
|
|
setMethod(
|
||
|
|
|
||
|
|
"dbGetQuery",
|
||
|
|
signature(conn="AthenaConnection", statement="character"),
|
||
|
|
|
||
|
|
definition = function(conn, statement, type_convert=FALSE, ...) {
|
||
|
|
|
||
|
|
r <- dbSendQuery(conn, statement, ...)
|
||
|
|
|
||
|
|
on.exit(.jcall(r@stat, "V", "close"))
|
||
|
|
|
||
|
|
#message("dbGetQuery()")
|
||
|
|
|
||
|
|
nms <- c()
|
||
|
|
athena_type_convert <- list()
|
||
|
|
|
||
|
|
cols <- .jcall(r@md, "I", "getColumnCount")
|
||
|
|
|
||
|
|
for (i in 1:cols) {
|
||
|
|
ct <- as.character(.jcall(r@md, "I", "getColumnType", i))
|
||
|
|
athena_type_convert[[i]] <- .jdbc_converters[[ct]]
|
||
|
|
nms <- c(nms, .jcall(r@md, "S", "getColumnLabel", i))
|
||
|
|
}
|
||
|
|
|
||
|
|
athena_type_convert <- set_names(athena_type_convert, nms)
|
||
|
|
|
||
|
|
res <- fetch(r, -1, block = 1000)
|
||
|
|
|
||
|
|
for (nm in names(athena_type_convert)) {
|
||
|
|
res[[nm]] <- athena_type_convert[[nm]](res[[nm]])
|
||
|
|
}
|
||
|
|
|
||
|
|
class(res) <- c("tbl_df", "tbl", "data.frame")
|
||
|
|
|
||
|
|
res
|
||
|
|
|
||
|
|
}
|
||
|
|
|
||
|
|
)
|