Hyper中的共享可变状态

时间:2018-10-24 16:39:09

标签: rust hyper rust-tokio

我正在尝试在Hyper Web服务器中创建一个计数器,该计数器对已收到的请求数进行计数。我正在使用Arc<Mutex<u64>>来保持计数。但是,我无法找出move.clone()的正确组合来满足闭包的类型。以下是一些可编译的代码,但会在每个请求时重置计数器:

extern crate hyper;

use hyper::rt::Future;
use hyper::service::service_fn_ok;
use hyper::{Body, Response, Server};
use std::sync::{Arc, Mutex};

fn main() {
    let addr = "0.0.0.0:3000".parse().unwrap();
    // FIXME want to create the counter here, not below
    let server = Server::bind(&addr)
        .serve(|| {
            service_fn_ok(|_req| {
                let counter = Arc::new(Mutex::new(0));
                use_counter(counter)
            })
        })
        .map_err(|e| eprintln!("Error: {}", e));
    hyper::rt::run(server)
}

fn use_counter(counter: Arc<Mutex<u64>>) -> Response<Body> {
    let mut data = counter.lock().unwrap();
    *data += 1;
    Response::new(Body::from(format!("Counter: {}\n", data)))
}

1 个答案:

答案 0 :(得分:1)

事实证明,我已经很接近了,看看其他几个例子可以帮助我意识到问题所在。由于这里有两层闭包,所以我需要将counter移到外部闭包中,对其进行克隆,然后将该克隆移入内部闭包中,然后再次在其中进行克隆。发挥作用:

extern crate hyper; // 0.12.10

use hyper::rt::Future;
use hyper::service::service_fn_ok;
use hyper::{Body, Response, Server};
use std::sync::{Arc, Mutex};

fn main() {
    let addr = "0.0.0.0:3000".parse().unwrap();
    let counter = Arc::new(Mutex::new(0));
    let server = Server::bind(&addr)
        .serve(move || {
            let counter = counter.clone();
            service_fn_ok(move |_req| use_counter(counter.clone()))
        })
        .map_err(|e| eprintln!("Error: {}", e));
    hyper::rt::run(server)
}

fn use_counter(counter: Arc<Mutex<u64>>) -> Response<Body> {
    let mut data = counter.lock().unwrap();
    *data += 1;
    Response::new(Body::from(format!("Counter: {}\n", data)))
}