R/old-dataset.R

In wilpoole-essence/bigrquery: An Interface to Google's 'BigQuery' 'API'

#' Dataset API [deprecated]
#'
#' Please use [api-dataset] instead.
#'
#' @keywords internal
#' @name dataset-dep
NULL

#' @export
#' @rdname dataset-dep
get_dataset <- function(project, dataset) {
  bq_get(bq_path(project, dataset))
}

#' @export
#' @rdname dataset-dep
exists_dataset <- function(project, dataset) {
  tryCatch(
    !is.null(get_dataset(project = project, dataset = dataset)),
    bigrquery_notFound = function(e) FALSE
  )
}

#' @export
#' @rdname dataset-dep
delete_dataset <- function(project, dataset, deleteContents = FALSE) {
  assert_that(is.string(project), is.string(dataset))

  url <- sprintf("projects/%s/datasets/%s", project, dataset)
  bq_delete(url, query = list(deleteContents = deleteContents))
}

#' @export
#' @rdname dataset-dep
insert_dataset <- function(project, dataset, ...) {
  assert_that(is.string(project), is.string(dataset))

  url <- bq_path(project, "")
  body <- list(
    datasetReference = list(
      projectId = project,
      datasetId = dataset
    )
  )

  bq_post(url, body = bq_body(body, ...))
}

#' @export
#' @rdname dataset-dep
update_dataset <- function(project, dataset, ...) {
  url <- bq_path(project, dataset)
  body <- list(
    datasetReference = list(
      projectId = project,
      datasetId = dataset
    )
  )

  bq_patch(url, body = bq_body(body, ...))
}

#' @export
#' @rdname dataset-dep
list_tables <- function(project, dataset, page_size = 50, max_pages = Inf) {
  data <- bq_get_paginated(
    bq_path(project, dataset, ""),
    page_size = page_size,
    max_pages = max_pages,
    warn = FALSE
  )

  tables <- unlist(lapply(data, function(x) x$tables), recursive = FALSE)
  vapply(tables, function(x) x$tableReference$tableId, character(1L))
}