R/dbListTables.R

In RPresto: DBI Connector to Presto

# Copyright (c) Meta Platforms, Inc. and affiliates.
# All rights reserved.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.

#' @include PrestoConnection.R
NULL

#' @rdname PrestoConnection-class
#' @param pattern optional SQL pattern for filtering table names, e.g. '%test%'
#' @importMethodsFrom DBI dbListTables
#' @export
setMethod(
  "dbListTables",
  "PrestoConnection",
  function(conn, pattern, ...) {
    if (!missing(pattern)) {
      statement <- paste("SHOW TABLES LIKE", DBI::dbQuoteString(conn, pattern))
    } else {
      statement <- "SHOW TABLES"
    }
    rv <- dbGetQuery(conn, statement)
    if (NROW(rv)) {
      return(rv[["Table"]])
    } else {
      return(character(0))
    }
  }
)