actix/examples
1
0
Fork 0
mirror of https://github.com/actix/examples synced 2025-02-22 17:23:18 +01:00
Code Issues Releases Wiki Activity

better multipart example

Browse Source
This commit is contained in:
Nikolay Kim 2018-06-07 21:34:14 -07:00
parent cc2cfaab6e
commit f8e3570bd1
1 changed files with 79 additions and 32 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

111
multipart/src/main.rs
View File

@ -4,54 +4,101 @@ extern crate actix_web;
extern crate env_logger; extern crate env_logger;
extern crate futures; extern crate futures;
use actix::*; use std::fs;
use std::io::Write;
use actix_web::{ use actix_web::{
http, middleware, multipart, server, App, AsyncResponder, Error, HttpMessage, error, http, middleware, multipart, server, App, Error, FutureResponse, HttpMessage,
HttpRequest, HttpResponse, HttpRequest, HttpResponse,
}; };
use futures::future::{result, Either}; use futures::future;
use futures::{Future, Stream}; use futures::{Future, Stream};
fn index(req: HttpRequest) -> Box<Future<Item = HttpResponse, Error = Error>> { pub fn save_file(
println!("{:?}", req); field: multipart::Field<HttpRequest>,
) -> Box<Future<Item = i64, Error = Error>> {
let file_path_string = "upload.png";
let mut file = match fs::File::create(file_path_string) {
Ok(file) => file,
Err(e) => return Box::new(future::err(error::ErrorInternalServerError(e))),
};
Box::new(
field
.fold(0i64, move |acc, bytes| {
let rt = file
.write_all(bytes.as_ref())
.map(|_| acc + bytes.len() as i64)
.map_err(|e| {
println!("file.write_all failed: {:?}", e);
error::MultipartError::Payload(error::PayloadError::Io(e))
});
future::result(rt)
})
.map_err(|e| {
println!("save_file failed, {:?}", e);
error::ErrorInternalServerError(e)
}),
)
}
req.multipart() // <- get multipart stream for current request pub fn handle_multipart_item(
.from_err() // <- convert multipart errors item: multipart::MultipartItem<HttpRequest>,
.and_then(|item| { // <- iterate over multipart items ) -> Box<Stream<Item = i64, Error = Error>> {
match item { match item {
// Handle multipart Field multipart::MultipartItem::Field(field) => {
multipart::MultipartItem::Field(field) => { Box::new(save_file(field).into_stream())
println!("==== FIELD ==== {:?}", field); }
multipart::MultipartItem::Nested(mp) => Box::new(
mp.map_err(error::ErrorInternalServerError)
.map(handle_multipart_item)
.flatten(),
),
}
}
// Field in turn is stream of *Bytes* object pub fn upload(req: HttpRequest) -> FutureResponse<HttpResponse> {
Either::A( Box::new(
field.map_err(Error::from) req.clone()
.map(|chunk| { .multipart()
println!("-- CHUNK: \n{}", .map_err(error::ErrorInternalServerError)
std::str::from_utf8(&chunk).unwrap());}) .map(handle_multipart_item)
.finish()) .flatten()
}, .collect()
multipart::MultipartItem::Nested(mp) => { .map(|sizes| HttpResponse::Ok().json(sizes))
// Or item could be nested Multipart stream .map_err(|e| {
Either::B(result(Ok(()))) println!("failed: {}", e);
} e
} }),
}) )
.finish() // <- Stream::finish() combinator from actix }
.map(|_| HttpResponse::Ok().into())
.responder() fn index(_req: HttpRequest) -> Result<HttpResponse, error::Error> {
let html = r#"<html>
<head><title>Upload Test</title></head>
<body>
<form target="/" method="post" enctype="multipart/form-data">
<input type="file" name="file"/>
<input type="submit" value="Submit"></button>
</form>
</body>
</html>"#;
Ok(HttpResponse::Ok().body(html))
} }
fn main() { fn main() {
::std::env::set_var("RUST_LOG", "actix_web=info"); ::std::env::set_var("RUST_LOG", "actix_web=info");
let _ = env_logger::init(); env_logger::init();
let sys = actix::System::new("multipart-example"); let sys = actix::System::new("multipart-example");
server::new(|| { server::new(|| {
App::new() App::new()
.middleware(middleware::Logger::default()) // <- logger .middleware(middleware::Logger::default())
.resource("/multipart", |r| r.method(http::Method::POST).a(index)) .resource("/", |r| {
r.method(http::Method::GET).with(index);
r.method(http::Method::POST).with(upload);
})
}).bind("127.0.0.1:8080") }).bind("127.0.0.1:8080")
.unwrap() .unwrap()
.start(); .start();