Update remaining examples for async.

This commit is contained in:
Jeb Rosen 2019-08-27 16:40:23 -07:00 committed by Sergio Benitez
parent 560f0977d3
commit 1f0577bfc5
9 changed files with 85 additions and 62 deletions

View File

@ -6,6 +6,7 @@ edition = "2018"
publish = false
[dependencies]
futures-preview = "0.3.0-alpha.18"
rocket = { path = "../../core/lib" }
serde = "1.0"
serde_json = "1.0"

View File

@ -5,10 +5,13 @@
#[cfg(test)] mod tests;
use std::io::{self, Read};
use std::io;
use futures::io::AsyncReadExt as _;
use rocket::{Request, data::Data};
use rocket::response::{Debug, content::{Json, Html}};
use rocket::AsyncReadExt as _;
// NOTE: This example explicitly uses the `Json` type from `response::content`
// for demonstration purposes. In a real application, _always_ prefer to use
@ -38,9 +41,10 @@ fn get_hello(name: String, age: u8) -> Json<String> {
// In a real application, we wouldn't use `serde_json` directly; instead, we'd
// use `contrib::Json` to automatically serialize a type into JSON.
#[post("/<age>", format = "plain", data = "<name_data>")]
fn post_hello(age: u8, name_data: Data) -> Result<Json<String>, Debug<io::Error>> {
async fn post_hello(age: u8, name_data: Data) -> Result<Json<String>, Debug<io::Error>> {
let mut name = String::with_capacity(32);
name_data.open().take(32).read_to_string(&mut name)?;
let mut stream = name_data.open().take(32);
stream.read_to_string(&mut name).await?;
let person = Person { name: name, age: age, };
// NOTE: In a real application, we'd use `rocket_contrib::json::Json`.
Ok(Json(serde_json::to_string(&person).expect("valid JSON")))

View File

@ -7,3 +7,5 @@ publish = false
[dependencies]
rocket = { path = "../../core/lib" }
tokio = "0.2.0-alpha.2"
futures-tokio-compat = { git = "https://github.com/Nemo157/futures-tokio-compat" }

View File

@ -3,51 +3,57 @@ extern crate rocket;
#[cfg(test)]
mod tests;
use std::{io, env};
use std::fs::File;
use std::env;
use futures_tokio_compat::Compat as TokioCompat;
use tokio::fs::File;
use rocket::{Request, Handler, Route, Data, Catcher, try_outcome};
use rocket::http::{Status, RawStr};
use rocket::response::{self, Responder, status::Custom};
use rocket::handler::Outcome;
use rocket::handler::{Outcome, HandlerFuture};
use rocket::outcome::IntoOutcome;
use rocket::http::Method::*;
fn forward<'r>(_req: &'r Request, data: Data) -> Outcome<'r> {
Outcome::forward(data)
fn forward<'r>(_req: &'r Request, data: Data) -> HandlerFuture<'r> {
Box::pin(async move { Outcome::forward(data) })
}
fn hi<'r>(req: &'r Request, _: Data) -> Outcome<'r> {
Outcome::from(req, "Hello!")
fn hi<'r>(req: &'r Request, _: Data) -> HandlerFuture<'r> {
Box::pin(async move { Outcome::from(req, "Hello!").await })
}
fn name<'a>(req: &'a Request, _: Data) -> Outcome<'a> {
fn name<'a>(req: &'a Request, _: Data) -> HandlerFuture<'a> {
Box::pin(async move {
let param = req.get_param::<&'a RawStr>(0)
.and_then(|res| res.ok())
.unwrap_or("unnamed".into());
Outcome::from(req, param.as_str())
Outcome::from(req, param.as_str()).await
})
}
fn echo_url<'r>(req: &'r Request, _: Data) -> Outcome<'r> {
fn echo_url<'r>(req: &'r Request, _: Data) -> HandlerFuture<'r> {
Box::pin(async move {
let param_outcome = req.get_param::<&RawStr>(1)
.and_then(|res| res.ok())
.into_outcome(Status::BadRequest);
let param = try_outcome!(param_outcome);
Outcome::try_from(req, RawStr::from_str(param).url_decode())
Outcome::try_from(req, RawStr::from_str(param).url_decode()).await
})
}
fn upload<'r>(req: &'r Request, data: Data) -> Outcome<'r> {
fn upload<'r>(req: &'r Request, data: Data) -> HandlerFuture<'r> {
Box::pin(async move {
if !req.content_type().map_or(false, |ct| ct.is_plain()) {
println!(" => Content-Type of upload must be text/plain. Ignoring.");
return Outcome::failure(Status::BadRequest);
}
let file = File::create(env::temp_dir().join("upload.txt"));
if let Ok(mut file) = file {
if let Ok(n) = io::copy(&mut data.open(), &mut file) {
return Outcome::from(req, format!("OK: {} bytes uploaded.", n));
let file = File::create(env::temp_dir().join("upload.txt")).await;
if let Ok(file) = file {
if let Ok(n) = data.stream_to(TokioCompat::new(file)).await {
return Outcome::from(req, format!("OK: {} bytes uploaded.", n)).await;
}
println!(" => Failed copying.");
@ -56,13 +62,14 @@ fn upload<'r>(req: &'r Request, data: Data) -> Outcome<'r> {
println!(" => Couldn't open file: {:?}", file.unwrap_err());
Outcome::failure(Status::InternalServerError)
}
})
}
fn get_upload<'r>(req: &'r Request, _: Data) -> Outcome<'r> {
Outcome::from(req, File::open(env::temp_dir().join("upload.txt")).ok())
fn get_upload<'r>(req: &'r Request, _: Data) -> HandlerFuture<'r> {
Outcome::from(req, std::fs::File::open(env::temp_dir().join("upload.txt")).ok())
}
fn not_found_handler<'r>(req: &'r Request) -> response::Result<'r> {
fn not_found_handler<'r>(req: &'r Request) -> response::ResultFuture<'r> {
let res = Custom(Status::NotFound, format!("Couldn't find: {}", req.uri()));
res.respond_to(req)
}
@ -79,13 +86,16 @@ impl CustomHandler {
}
impl Handler for CustomHandler {
fn handle<'r>(&self, req: &'r Request, data: Data) -> Outcome<'r> {
fn handle<'r>(&self, req: &'r Request, data: Data) -> HandlerFuture<'r> {
let self_data = self.data;
Box::pin(async move {
let id_outcome = req.get_param::<&RawStr>(0)
.and_then(|res| res.ok())
.or_forward(data);
let id = try_outcome!(id_outcome);
Outcome::from(req, format!("{} - {}", self.data, id))
Outcome::from(req, format!("{} - {}", self_data, id)).await
})
}
}

View File

@ -30,8 +30,8 @@ fn test_echo() {
test(&uri, ContentType::Plain, Status::Ok, "echo this text".into());
}
#[test]
fn test_upload() {
#[rocket::async_test]
async fn test_upload() {
let client = Client::new(rocket()).unwrap();
let expected_body = "Lorem ipsum dolor sit amet, consectetur adipiscing elit, \
sed do eiusmod tempor incididunt ut labore et dolore \

View File

@ -7,7 +7,7 @@ mod paste_id;
use std::io;
use std::fs::File;
use std::path::Path;
use std::path::PathBuf;
use rocket::Data;
use rocket::response::{content, Debug};
@ -18,12 +18,12 @@ const HOST: &str = "http://localhost:8000";
const ID_LENGTH: usize = 3;
#[post("/", data = "<paste>")]
fn upload(paste: Data) -> Result<String, Debug<io::Error>> {
async fn upload(paste: Data) -> Result<String, Debug<io::Error>> {
let id = PasteID::new(ID_LENGTH);
let filename = format!("upload/{id}", id = id);
let url = format!("{host}/{id}\n", host = HOST, id = id);
paste.stream_to_file(Path::new(&filename))?;
paste.stream_to_file(PathBuf::from(filename)).await?;
Ok(url)
}

View File

@ -8,10 +8,8 @@ use std::{io, env};
use rocket::{Data, response::Debug};
#[post("/upload", format = "plain", data = "<data>")]
fn upload(data: Data) -> Result<String, Debug<io::Error>> {
data.stream_to_file(env::temp_dir().join("upload.txt"))
.map(|n| n.to_string())
.map_err(Debug)
async fn upload(data: Data) -> Result<String, Debug<io::Error>> {
Ok(data.stream_to_file(env::temp_dir().join("upload.txt")).await?.to_string())
}
#[get("/")]

View File

@ -7,3 +7,6 @@ publish = false
[dependencies]
rocket = { path = "../../core/lib" }
futures-preview = "0.3.0-alpha.18"
tokio = "0.2.0-alpha.2"
futures-tokio-compat = { git = "https://github.com/Nemo157/futures-tokio-compat" }

View File

@ -6,22 +6,27 @@
use rocket::response::{content, Stream};
use std::io::{repeat, Repeat, Read, Take};
use std::fs::File;
use std::io::repeat;
type LimitedRepeat = Take<Repeat>;
use tokio::fs::File;
use futures_tokio_compat::Compat as TokioCompat;
use rocket::AsyncReadExt as _;
//type LimitedRepeat = Take<Repeat>;
type LimitedRepeat = Box<dyn futures::io::AsyncRead + Send + Unpin>;
// Generate this file using: head -c BYTES /dev/random > big_file.dat
const FILENAME: &str = "big_file.dat";
#[get("/")]
fn root() -> content::Plain<Stream<LimitedRepeat>> {
content::Plain(Stream::from(repeat('a' as u8).take(25000)))
content::Plain(Stream::from(Box::new(repeat('a' as u8).take(25000)) as Box<_>))
}
#[get("/big_file")]
fn file() -> Option<Stream<File>> {
File::open(FILENAME).map(|file| Stream::from(file)).ok()
async fn file() -> Option<Stream<TokioCompat<File>>> {
File::open(FILENAME).await.map(|file| Stream::from(TokioCompat::new(file))).ok()
}
fn rocket() -> rocket::Rocket {