rust - Hyper 中的共享可变状态

标签 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)))
}

最佳答案

事实证明我非常接近,并且查看其他几个示例帮助我意识到了这个问题。由于这里有两层闭包,我需要将 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)))
}

2020 年 2 月更新这是一个使用 hyper 0.13 的版本:

use hyper::{Body, Response, Server, Request};
use std::sync::{Arc, Mutex};
use hyper::service::{make_service_fn, service_fn};
use std::convert::Infallible;

#[tokio::main]
async fn main() -> Result<(), Box<dyn std::error::Error>> {
    let addr = "0.0.0.0:3000".parse()?;
    let counter = Arc::new(Mutex::new(0));

    let make_service = make_service_fn(move |_conn| {
        let counter = counter.clone();
        async move {
            Ok::<_, Infallible>(service_fn(move |_req: Request<Body>| {
                let counter = counter.clone();
                async move {
                    Ok::<_, Infallible>(use_counter(counter))
                }
            }))
        }
    });

    Server::bind(&addr).serve(make_service).await?;
    Ok(())
}

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)))
}

关于rust - Hyper 中的共享可变状态,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/52974081/

相关文章:

python - Python中的TypeError在反射(reflect)的数字仿真器(例如__radd__)中使用PyAny作为Rust pyo3 pyclass结构

json - 从 Hyper 获取请求,解析 JSON,并将其转换为 Rust 中的结构

rust - 无法将闭包传递给 `Hyper::service_fn`

http - 带有Rust的AWS Lambda的reqwest段错误

将字符串解析为整数

rust - 如何将不可变切片传递给需要 &mut impl Read 的 fn?

Rust Tokio - 监控停滞的任务

rust - 在异步任务中读取 stdin 时为 "blocking annotated I/O must be called from the context of the Tokio runtime"

rust - 具有可变引用的临时元组的模式匹配

rust - 使用hyper时如何直接控制http body(大尺寸)?