R/dplyr-single.R

In hadley/multidplyr: A Multi-Process 'dplyr' Backend

#' @importFrom dplyr arrange
#' @export
arrange.multidplyr_party_df <- function(.data, ..., .by_group = FALSE) {
  shard_call(.data, "arrange", enquos(...), .by_group = .by_group)
}

#' @importFrom dplyr filter
# exported on load
filter.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "filter", enquos(...))
}

#' @importFrom dplyr group_by
#' @export
group_by.multidplyr_party_df <- function(.data, ..., .add = FALSE) {
  shard_call(.data, "group_by", enquos(...), .add = .add)
}

#' @importFrom dplyr ungroup
#' @export
ungroup.multidplyr_party_df <- function(x, ...) {
  shard_call(x, "ungroup", enquos(...))
}

#' @importFrom dplyr mutate
#' @export
mutate.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "mutate", enquos(...))
}

#' @importFrom dplyr rename
#' @export
rename.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "rename", enquos(...))
}

#' @importFrom dplyr select
#' @export
select.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "select", enquos(...))
}

#' @importFrom dplyr slice
#' @export
slice.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "slice", enquos(...))
}

#' @importFrom dplyr summarise
#' @export
summarise.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "summarise", enquos(...))
}

#' @importFrom dplyr do
#' @export
do.multidplyr_party_df <- function(.data, ...) {
  shard_call(.data, "do", enquos(...))
}


# helpers -----------------------------------------------------------------

shard_call <- function(.data, .verb, dots, ...) {
  call <- call2(.verb, .data$name, !!!dots, ..., .ns = "dplyr")

  new_name <- table_name()
  cluster_send(.data$cluster, !!call2("<-", new_name, call))
  new_party_df(.data$cluster, new_name, auto_rm = TRUE)
}