我试图通过使用Hyper 0.10实现一个简单的内存URL缩短器来学习Rust。我遇到了一个问题,我认为这是因为我试图在我的处理程序中关闭一个可变的HashMap
:
fn post(mut req: Request, mut res: Response, short_uris: &mut HashMap<&str, &str>) {
let mut body = String::new();
match req.read_to_string(&mut body) {
Ok(_) => {
let key = short_uris.len();
short_uris.insert(&key.to_string(), &body.to_string());
*res.status_mut() = StatusCode::Created;
res.start().unwrap().write(&key.to_string().into_bytes());
},
Err(_) => *res.status_mut() = StatusCode::BadRequest
}
}
fn get(req: Request, mut res: Response, short_uris: &HashMap<&str, &str>) {
match req.uri.clone() {
AbsolutePath(path) => {
match short_uris.get::<str>(&path) {
Some(short_uri) => {
*res.status_mut() = StatusCode::MovedPermanently;
res.headers_mut().set(Location(short_uri.to_string()));
},
None => *res.status_mut() = StatusCode::NotFound
}
},
_ => *res.status_mut() = StatusCode::BadRequest
}
}
fn main() {
let mut short_uris: HashMap<&str, &str> = HashMap::new();
short_uris.insert("/example", "http://www.example.com");
Server::http("0.0.0.0:3001").unwrap().handle(move |req: Request, mut res: Response| {
match req.method {
hyper::Post => post(req, res, &mut short_uris),
hyper::Get => get(req, res, &short_uris),
_ => *res.status_mut() = StatusCode::MethodNotAllowed
}
}).unwrap();
}
src/main.rs:42:40: 42:46 error: the trait bound `for<'r, 'r, 'r> [closure@src/main.rs:42:47: 48:3 short_uris:std::collections::HashMap<&str, &str>]: std::ops::Fn<(hyper::server::Request<'r, 'r>, hyper::server::Response<'r>)>` is not satisfied [E0277]
src/main.rs:42 Server::http("0.0.0.0:3001").unwrap().handle(move |req: Request, mut res: Response| {
我是否需要使用Arc
在线程之间共享HashMap
?如果是这样,那会是什么样子?此外,我对这个问题完全错了。错误信息对我来说非常神秘。
答案 0 :(得分:6)
请在下次包含所有必要的use
声明,谢谢!
如果您每晚使用Rust,则错误消息不那么神秘:
期望一个实现
Fn
特征的闭包,但这个闭包仅实现FnMut
这意味着Hyper需要在线程之间共享闭包,因此闭包只需要通过不可变或共享方法来使用它的环境 - 所以&mut short_uris
的使用就是攻击者。要在Rust中提供共享线程安全可变性,您应该使用Mutex
或RwLock
。
请注意,不需要Arc
- Hyper管理关闭本身的所有权(可能是通过将封闭包裹在Arc
引擎盖下,或者使用像scoped-threads这样的东西。
您的代码还存在第二个问题 - 您使用HashMap<&str, &str>
。 &str
是借来的参考。每当你在Rust中借用一些东西时,你应该问问自己 - 从哪里来?在这里,您尝试借用非常短暂的字符串 - key.to_string()
和body.to_string()
。它无法正常工作。只需让您的hashmap完全拥有 - HashMap<String, String>
。这是编译代码的版本:
extern crate hyper;
use hyper::server::{Request, Response, Server};
use std::collections::HashMap;
use hyper::status::StatusCode;
use hyper::uri::RequestUri::AbsolutePath;
use hyper::header::Location;
use std::io::prelude::*;
fn post(mut req: Request, mut res: Response, short_uris: &mut HashMap<String, String>) {
let mut body = String::new();
match req.read_to_string(&mut body) {
Ok(_) => {
let key = short_uris.len();
short_uris.insert(key.to_string(), body);
*res.status_mut() = StatusCode::Created;
res.start()
.unwrap()
.write(&key.to_string().into_bytes())
.unwrap();
}
Err(_) => *res.status_mut() = StatusCode::BadRequest,
}
}
fn get(req: Request, mut res: Response, short_uris: &HashMap<String, String>) {
match req.uri {
AbsolutePath(ref path) => match short_uris.get(path) {
Some(short_uri) => {
*res.status_mut() = StatusCode::MovedPermanently;
res.headers_mut().set(Location(short_uri.to_string()));
}
None => *res.status_mut() = StatusCode::NotFound,
},
_ => *res.status_mut() = StatusCode::BadRequest,
}
}
fn main() {
let mut short_uris: HashMap<String, String> = HashMap::new();
short_uris.insert("/example".into(), "http://www.example.com".into());
let short_uris = std::sync::RwLock::new(short_uris);
Server::http("0.0.0.0:3001")
.unwrap()
.handle(move |req: Request, mut res: Response| match req.method {
hyper::Post => post(req, res, &mut short_uris.write().unwrap()),
hyper::Get => get(req, res, &short_uris.read().unwrap()),
_ => *res.status_mut() = StatusCode::MethodNotAllowed,
})
.unwrap();
}
我还摆脱了.clone()
函数中不必要的get
。
请注意,此代码在编译时尚不完美 - RwLock
锁定应该更短(get
和post
应该将&RwLock<HashMap<String,String>>
作为参数并且自己执行锁定)。 .unwrap()
也可以更好的方式处理。你也可以考虑使用一些无锁的并发hashmap,应该有一些板条箱,但是我没有进入主题,所以我不会推荐任何一个。