Rust/rocket将变量传递给端点

Question

这不是我喜欢的，但是我今天不得不写一些Rust，所以我试图创建一个只有一个端点的Rocket实例，但是在这个端点上，我需要访问一个在main期间创建的变量。该变量需要很长时间才能实例化，所以这就是为什么我在这里这样做的原因。

我的问题是我找不到安全通过它的方法。无论我做什么，编译器都会抱怨线程安全，尽管这个库看起来是线程安全的：https://github.com/brave/adblock-rust/pull/130 (提交的代码在我的本地实例上)

这是我得到的错误信息：

   |
18 | / lazy_static! {
19 | |     static ref rules_engine: Mutex<Vec<Engine>> = Mutex::new(vec![]);
20 | | }
   | |_^ `std::rc::Rc<std::cell::RefCell<lifeguard::CappedCollection<std::vec::Vec<u64>>>>` cannot be sent between threads safely
   | 

...and这是我的代码：

#![feature(proc_macro_hygiene, decl_macro)]

#[macro_use]
extern crate rocket;

use std::fs::File;
use std::io::{self, BufRead};
use std::path::Path;

use lazy_static::lazy_static;
use std::sync::Mutex;

use adblock::engine::Engine;
use adblock::lists::FilterFormat;

use rocket::request::{Form, FormError, FormDataError};

lazy_static! {
    static ref rules_engine: Mutex<Vec<Engine>> = Mutex::new(vec![]);
}

fn main() {
    if !Path::new("./rules.txt").exists() {
        println!("rules file does not exist")
    } else {
        println!("loading rules");

        let mut rules = vec![];

        if let Ok(lines) = read_lines("./rules.txt") {
            for line in lines {
                if let Ok(ip) = line {
                    rules.insert(0, ip)
                }
            }

            let eng = Engine::from_rules(&rules, FilterFormat::Standard);
            rules_engine.lock().unwrap().push(eng);
            rocket().launch();
        }
    }
}

#[derive(Debug, FromForm)]
struct FormInput<> {
    #[form(field = "textarea")]
    text_area: String
}

#[post("/", data = "<sink>")]
fn sink(sink: Result<Form<FormInput>, FormError>) -> String {
    match sink {
        Ok(form) => {
            format!("{:?}", &*form)
        }
        Err(FormDataError::Io(_)) => format!("Form input was invalid UTF-8."),
        Err(FormDataError::Malformed(f)) | Err(FormDataError::Parse(_, f)) => {
            format!("Invalid form input: {}", f)
        }
    }
}

fn rocket() -> rocket::Rocket {
    rocket::ignite().mount("/", routes![sink])
}

fn read_lines<P>(filename: P) -> io::Result<io::Lines<io::BufReader<File>>>
    where P: AsRef<Path>, {
    let file = File::open(filename)?;
    Ok(io::BufReader::new(file).lines())
}

有没有办法让sink端点方法中的eng可用？

Answer 1

Rc不是线程安全的，即使在互斥之后也是如此。它看起来像是在eng.blocker.pool.pool中使用了Rc，它是一个lifeguard::Pool。所以，Engine不是线程安全的(至少在默认情况下是这样)。

幸运的是，adblock crate似乎有一个名为“对象池”的feature，它启用了特定的功能。删除该功能将(希望)使其线程安全。

Answer 2

main使得在路由之间(以及在main或从Rocket产生的任何其他线程之间)共享资源变得非常容易。他们称他们的机制为state。请查看其文档here。

举一个简短的例子来说明它是如何工作的:创建要在应用程序中共享的类型，并在应用程序使用的rocket实例中manage该类型的一个实例。在指南中，他们给出了这个例子：

use std::sync::atomic::AtomicUsize;

struct HitCount {
    count: AtomicUsize
}

rocket::build().manage(HitCount { count: AtomicUsize::new(0) });

在路由中，您可以像这样访问资源(同样是从指南中)：

use rocket::State;

#[get("/count")]
fn count(hit_count: &State<HitCount>) -> String {
    let current_count = hit_count.count.load(Ordering::Relaxed);
    format!("Number of visits: {}", current_count)
}

当我学习rocket时，我需要共享一个包含String的结构，这本身就不是线程安全的。这意味着在使用manage之前，您需要将其包装到一个Mutex中。

另外，据我所知，任何特定类型的资源只能与manage共享。但在这种情况下，您可以创建不同的命名包装器类型，并解决该限制。