R/driver-hive.R

In rstats-db/odbc: Connect to ODBC Compatible Databases (using the DBI Interface)

#' @export
#' @rdname DBI-classes
setClass("Hive", contains = "OdbcConnection")

# only need to override dbQuteString when x is character.
# DBI:::quote_string just returns x when it is of class SQL, so no need to override that.
#' @rdname DBI-methods
setMethod("dbQuoteString", c("Hive", "character"),
  function(conn, x, ...) {
    if (is(x, "SQL")) {
      return(x)
    }
    x <- gsub("'", "\\\\'", enc2utf8(x))
    if (length(x) == 0L) {
      DBI::SQL(character())
    } else {
      str <- paste0("'", x, "'")
      str[is.na(x)] <- "NULL"
      DBI::SQL(str)
    }
  }
)


#' @export
#' @rdname odbcDataType
#' @usage NULL
setMethod("odbcDataType", "Hive",
  function(con, obj, ...) {
    switch_type(
      obj,
      factor = "STRING",
      datetime = "TIMESTAMP",
      date = "DATE",
      binary = "BINARY",
      integer = "INT",
      int64 = "INT",
      double = "DOUBLE",
      character = "STRING",
      logical = "BOOLEAN",
      list = "STRING",
      time = ,
      stop("Unsupported type", call. = FALSE)
    )
  }
)