+ read_arrow / write_arrow for now

romainfrancois · romainfrancois · commit 8d8cdd1ce184 · 2018-09-24T20:15:02.000+02:00
diff --git a/r/NAMESPACE b/r/NAMESPACE
@@ -29,6 +29,7 @@ export(int64)
 export(int8)
 export(list_of)
 export(null)
+export(read_arrow)
 export(read_record_batch)
 export(record_batch)
 export(schema)
@@ -42,6 +43,7 @@ export(uint32)
 export(uint64)
 export(uint8)
 export(utf8)
+export(write_arrow)
 importFrom(R6,R6Class)
 importFrom(Rcpp,sourceCpp)
 importFrom(assertthat,assert_that)
diff --git a/r/R/array.R b/r/R/array.R
@@ -78,6 +78,15 @@ array <- function(...){
   RecordBatch_to_dataframe(x)
 }
 
+#' @export
+write_arrow <- function(data, path){
+  record_batch(data)$to_file(path)
+}
+
+#' @export
+read_arrow <- function(path){
+  as_tibble(read_record_batch(path))
+}
 
 #' Create an arrow::RecordBatch from a data frame
 #'
diff --git a/r/src/buffer.cpp b/r/src/buffer.cpp
@@ -159,8 +159,6 @@ List RecordBatch_to_dataframe(const std::shared_ptr<arrow::RecordBatch>& batch){
   return tbl;
 }
 
-
-
 // [[Rcpp::export]]
 std::shared_ptr<arrow::Table> dataframe_to_Table(DataFrame tbl){
   auto rb = dataframe_to_RecordBatch(tbl);

Original file line number	Diff line number	Diff line change
`@@ -159,8 +159,6 @@ List RecordBatch_to_dataframe(const std::shared_ptr<arrow::RecordBatch>& batch){`
`159`	`159`	`return tbl;`
`160`	`160`	`}`
`161`	`161`
`162`		`-`
`163`		`-`
`164`	`162`	`// [[Rcpp::export]]`
`165`	`163`	`std::shared_ptr<arrow::Table> dataframe_to_Table(DataFrame tbl){`
`166`	`164`	`auto rb = dataframe_to_RecordBatch(tbl);`