r - 在 Plumber API R 中获取 CSV 作为输入和输出 CSV
问题描述
我想在管道工 API 中获取 CSV 的输入并将其输出为 CSV。
我可以单独做这两个(在下面附加的链接的帮助下)。
但我不能一起做。
使用来自 Github Link 的序列化器:
serializer_csv <- function(){
function(val, req, res, errorHandler){
tryCatch({
res$setHeader("Content-Type", "text/plain")
res$setHeader("Content-Disposition", 'attachment; filename="xxx.csv"')
res$body <- paste0(val, collapse="\n")
return(res$toResponse())
}, error=function(e){
errorHandler(req, res, e)
})
}
}
plumber::addSerializer("csv", serializer_csv)
plumber::plumber$new("plumber-csv.R")$run(port = 8833)
管道工文件:
#* @serializer csv
#* @get /csv
function() {
df <- data.frame(CHAR = letters, NUM = rnorm(length(letters)), stringsAsFactors = F)
csv_file <- tempfile(fileext = ".csv")
on.exit(unlink(csv_file), add = TRUE)
write.csv(df, file = csv_file)
readLines(csv_file)
}
在这里,readLines
我没有直接尝试使用return(csv_file)
. 它下载一个临时文件。但是在我的本地磁盘上,该文件没有出现。
那么输入和输出CSV文件的方法是什么?链接:
解决方案
这项工作在 Plumber 版本 4.6.0 中,但有点 hacky。您可以将write.csv
andread.csv
替换为fwrite
、fread
fromdata.table
或format_csv
from readr
。
library(plumber)
serializer_csv <- function(){
function(val, req, res, errorHandler){
tryCatch({
res$setHeader("Content-Type", "text/csv")
res$setHeader("Content-Disposition", paste0('attachment; filename="', names(val),'.csv"'))
content <- capture.output(write.csv(val[[1]]), type = "output")
res$body <- charToRaw(paste0(content, collapse = "\n"))
return(res$toResponse())
}, error=function(e){
errorHandler(req, res, e)
})
}
}
plumber::addSerializer("csv", serializer_csv)
plumber::plumber$new("plumber-csv.R")$run(port = 8833)
水管工
#* @serializer csv
#* @post /csv
function(req) {
req$rook.input$rewind()
f <- tempfile()
writeLines(req$rook.input$read_lines(), f)
df <- read.csv(f)
response <- list(response = df)
response
}
测试
library(httr)
library(readr)
write.csv(mtcars, "mtcars.csv")
req <- POST("http://127.0.0.1:8833/csv", body = upload_file("mtcars.csv"))
content(req)
推荐阅读
- javascript - 如何在 angular6 中创建非 SPA 页面?
- c# - 为什么在 C# 中使用 AutoResetEvent 和 ManualResetEvent 时我的代码输出不同
- java - 如何在功能/单元测试中传递查询/路径参数?
- java - Java,如何正确设置 Servlet/Tomat(未找到错误符号)?
- c - 为什么在条件语句中放置任何标识符而不给出任何条件?
- android - 在 androidx 中的 FAB 上方显示小吃店
- sql - SQL中与case语句条件关联的值
- javascript - 如何加载更多引用帖子 ID 的 ajax 数据,使用 php 获取
- rust - 如何使用固定大小的数组分段下载文件?
- python - 是否可以在 twitterStream.filiter 中添加几个复杂的条件?