Rust Gnuplot - хранить PNG в памяти как байты
Я пытаюсь настроить простой сервер actix-web с одной конечной точкой, называемой plot. По сути, он просто потребляет некоторые данные, отображает их с помощью gnuplot и возвращает байты полученного PNG. проблема в том, что, как вы увидите в коде, я не нашел способа сделать все это в памяти, что означает, что я должен сохранить файл на диск, снова открыть его в читателе и затем отправить ответ обратно. В зависимости от уровня параллелизма я начну получать { code: 24, kind: Other, message: "Too many open files" } сообщения.
Кто-нибудь знает, как мне это сделать, чтобы весь процесс выполнялся в памяти? Я использую:
actix-web = "3"
gnuplot = "0.0.37"
image = "0.23.12"
Буду признателен за любую помощь, вот код:
use actix_web::{post, web, App, HttpResponse, HttpServer, Responder};
use gnuplot::{AxesCommon, Color, Figure, LineWidth};
use image::io::Reader;
use rand::distributions::Alphanumeric;
use rand::{thread_rng, Rng};
use std::any::type_name;
use std::collections::HashMap;
use std::fs;
#[post("/")]
async fn plot(req_body: web::Json<HashMap<String, Vec<f64>>>) -> impl Responder {
let data = req_body.get("data").unwrap();
let mut fg = Figure::new();
let fid: String = thread_rng().sample_iter(&Alphanumeric).take(10).collect();
let fname: String = format!("./{fid}.png", fid = fid);
fg.set_terminal("pngcairo", &fname);
let ax = fg.axes2d();
ax.set_border(false, &[], &[]);
ax.set_pos(0.0, 0.0);
ax.set_x_ticks(None, &[], &[]);
ax.set_y_ticks(None, &[], &[]);
let x: Vec<usize> = (1..data.len()).collect();
ax.lines(&x, data, &[LineWidth(4.0), Color("black")]);
fg.set_post_commands("unset output").show();
let image = Reader::open(&fname).unwrap().decode().unwrap();
let mut bytes: Vec<u8> = Vec::new();
image.write_to(&mut bytes, image::ImageOutputFormat::Png);
fs::remove_file(fname);
HttpResponse::Ok().body(bytes)
}
#[actix_web::main]
async fn main() -> std::io::Result<()> {
HttpServer::new(|| App::new().service(plot))
.bind("127.0.0.1:8080")?
.run()
.await
}
Ответы 2
Crate gnuplot может не иметь такой функции. Но, к счастью, gnuplot может выводить байты изображения на стандартный вывод с помощью set term png и set output. Запустите gnuplot напрямую с помощью std::process::Command и сохраните вывод в памяти.
Чтобы не создавать файл, вы можете сделать то, что описал Акихито КИРИСАКИ. Вы делаете это, вызывая set_terminal() , но вместо имени файла вы передаете пустую строку. Затем вы создаете Command и echo() в stdin.
use std::process::{Command, Stdio};
#[post("/")]
async fn plot(req_body: web::Json<HashMap<String, Vec<f64>>>) -> impl Responder {
...
fg.set_terminal("pngcairo", "");
...
fg.set_post_commands("unset output");
let mut child = Command::new("gnuplot")
.stdin(Stdio::piped())
.stdout(Stdio::piped())
.spawn()
.expect("expected gnuplot");
let mut stdin = child.stdin.take().expect("expected stdin");
fg.echo(&mut stdin);
// Drop `stdin` such that it is flused and closed,
// otherwise some programs might block until stdin
// is closed.
drop(stdin);
let output = child.wait_with_output().unwrap();
let png_image_data = output.stdout;
HttpResponse::Ok().body(png_image_data)
}
Вам также нужно убрать вызов show().
Забыл включить output, я обновил ответ :)
let output = child.wait_with_output(); Попался! Большое спасибо :)
Удивительный! Это именно то, чего я пытался достичь, я думал, что мне придется подключиться к стандартному вводу/выводу через эхо, но понятия не имел, как это сделать. Я пытаюсь воспроизвести ваше решение, но не могу понять, откуда берется
output.