Adds optional usage of dplyr::bind_rows() or data.table::rbindlist().

Signed-off-by: Aaron Jacobs <[email protected]>

DESCRIPTION

@@ -26,6 +26,8 @@ Roxygen: list(markdown = TRUE)
 Suggests:
     spelling,
     nycflights13,
-    ggplot2
+    ggplot2,
+    data.table,
+    dplyr
 Language: en-GB
 Encoding: UTF-8

R/stream.R

@@ -61,10 +61,16 @@ mongo_stream_in <- function(cur, handler = NULL, pagesize = 1000, verbose = TRUE
     out <- as.list(out, sorted = FALSE)
     out <- out[order(as.numeric(names(out)))]
     if (flat) {
-      out <- lapply(out, as.data.frame, stringsAsFactors = FALSE)
-      out <- do.call("rbind", out)
-      # Alternatively:
-      # out <- data.table::rbindlist(out)
+      if (requireNamespace("dplyr", quietly = TRUE)) {
+        out <- dplyr::bind_rows(out)
+      } else if (requireNamespace("data.table", quietly = TRUE)) {
+        out <- data.table::rbindlist(out, fill = TRUE)
+      } else {
+        # TODO: This is much, much slower. We might consider bundling an
+        # rbindlist implementation suited to the inputs instead.
+        out <- lapply(out, as.data.frame, stringsAsFactors = FALSE)
+        out <- do.call("rbind", out)
+      }
       # For compatibility with jsonlite:::simplifyDataFrame():
       attr(out, "row.names") <- seq_len(nrow(out))
       out