如何在actix-web的服务代码中注入不可变的对象?

时间:2019-01-30 16:49:37

标签: rust rust-actix

我正在使用actix framework来构建服务器,该服务器应支持向给定user_id的用户显示年龄/平衡的机会:

fn show_balance(req: &HttpRequest) -> HttpResponse {
    let client = create_client();
    let user_id = req.match_info().get("user_id").unwrap();
    let balance = client.load_grade(user_id); // Returns a balance as a String
    HttpResponse::Ok()
        .content_type("text/plain")
        .body(format!("Hello! Your balance is {}", balance))
}

fn show_age(req: &HttpRequest) -> HttpResponse {
    let client = create_client();
    let user_id = req.match_info().get("user_id").unwrap();
    let age = client.load_grade(user_id); // Returns an age as a String
    HttpResponse::Ok()
        .content_type("text/plain")
        .body(format!("Hello! Your balance is {}", age))
}

fn main() {
    env::set_var("RUST_LOG", "actix_web=debug");
    env::set_var("RUST_BACKTRACE", "1");
    env_logger::init();
    let sys = actix::System::new("basic-example");

    let addr = server::new(
        || App::new()
            // enable logger
            .middleware(middleware::Logger::default())
            .resource("/balance/{user_id}", |r| r.method(Method::GET).f(show_balance))
            .resource("/age/{user_id}", |r| r.method(Method::GET).f(show_age))
        .bind("127.0.0.1:8080").expect("Can not bind to 127.0.0.1:8080")
        .start();

    println!("Starting http server: 127.0.0.1:8080");
    let _ = sys.run();
}

fn create_client() -> UserDataClient {
    let enviroment = grpcio::EnvBuilder::new().build();
    let channel = grpcio::ChannelBuilder::new(enviroment)
        .connect(API_URL);
    UserDataClient::new(channel)
}

此代码有效,但我担心的是,我必须为效率低下且可读性强的每个传入请求创建一个客户端(并打开一个通道),我认为最好改成一个单例(因为我可以重复使用)。我浏览了example folder,发现todo example与我正在做的事情有点相似。因此,我找到了以下两个选项来注入我的client对象(在main()中创建它的单个实例之后:

  1. 将其放入app state
  2. 将其作为中间件(12)注入吗?

实施哪种最佳/正确的方法是什么?

我考虑过仅将一个客户端对象作为参数传递给每个处理程序,但是我没有设法使其起作用(反正看起来也不好)。

0 个答案:

没有答案