R/dbHasCompleted__duckdb_result.R

In duckdb: DBI Package for the DuckDB Database Management System

#' @rdname duckdb_result-class
#' @inheritParams DBI::dbHasCompleted
#' @usage NULL
dbHasCompleted__duckdb_result <- function(res, ...) {
  if (!res@env$open) {
    stop("result has already been cleared")
  }

  if (is.null(res@env$resultset)) {
    FALSE
  } else if (res@stmt_lst$type == "SELECT") {
    res@env$rows_fetched == nrow(res@env$resultset)
  } else {
    TRUE
  }
}

#' @rdname duckdb_result-class
#' @export
setMethod("dbHasCompleted", "duckdb_result", dbHasCompleted__duckdb_result)