rocket请求守卫

最新推荐文章于 2024-09-12 14:35:52 发布

Cra te

最新推荐文章于 2024-09-12 14:35:52 发布

阅读量188

点赞数

文章标签： rust 后端服务器

本文链接：https://blog.csdn.net/DCABQ/article/details/130062368

版权

Rocket是一个Web框架，提供请求守卫功能，包括Guard（检查请求条件）、Catch（处理异常）和Fairing（启动时操作）。Guard用于验证请求，如检查APIKey；Catch可捕获并定制404错误响应；Fairing则用于加载配置或环境变量。

摘要由CSDN通过智能技术生成

Rocket是一个Web框架，它提供了一些请求守卫（request guard）的功能，可以用于过滤和验证传入的HTTP请求。请求守卫是在处理请求之前执行的函数，它们可以检查请求是否满足某些特定的条件，如果满足条件，则请求将被传递到处理程序（handler）中进行处理，否则将返回错误响应。

Rocket提供了三种类型的请求守卫：

Guard: 检查传入的请求是否满足某些特定的条件，如果满足条件，则请求将被传递到处理程序中进行处理。否则，请求将被拒绝，返回错误响应。例如，可以使用Header守卫检查请求头是否包含指定的标头。

use rocket::http::Status;
use rocket::request::{FromRequest, Outcome};
use rocket::{Request, State};

struct ApiKey(String);

impl<'a, 'r> FromRequest<'a, 'r> for ApiKey {
    type Error = ();

    fn from_request(request: &'a Request<'r>) -> Outcome<Self, Self::Error> {
        let keys: Vec<_> = request.headers().get("X-Api-Key").collect();
        if keys.len() != 1 {
            return Outcome::Failure((Status::BadRequest, ()));
        }

        Outcome::Success(ApiKey(keys[0].to_string()))
    }
}

#[get("/sensitive")]
fn sensitive(key: ApiKey) -> String {
    format!("Sensitive data, api key is: {}", key.0)
}

Catch: 用于处理某些类型的异常并返回错误响应。例如，可以使用Catch守卫处理404 Not Found异常并返回自定义错误响应。

use rocket::{catch, catchers};

#[catch(404)]
fn not_found() -> &'static str {
    "Page not found."
}

#[get("/unknown")]
fn unknown() -> &'static str {
    "This page does not exist."
}

fn main() {
    rocket::ignite()
        .register(catchers![not_found])
        .mount("/", routes![unknown])
        .launch();
}

Fairing: 用于在应用程序启动时执行一些操作，例如设置全局状态，加载配置文件等。例如，可以使用Fairing守卫加载环境变量并将其设置为全局状态。

use std::env;
use rocket::{fairing::AdHoc, State};

struct Config {
    port: u16,
}

impl Config {
    fn new() -> Config {
        let port = env::var("PORT")
            .unwrap_or_else(|_| "8000".to_string())
            .parse()
            .unwrap_or(8000);

        Config { port }
    }
}

#[get("/")]
fn index(config: &State<Config>) -> String {
    format!("Server is running on port {}", config.port)
}

fn main() {
    let config = Config::new();

    rocket::ignite()
        .manage(config)
        .mount("/", routes![index])
        .attach(AdHoc::on_launch("Loading environment variables", |_| {
            println!("Loading environment variables...");
        }))
        .launch();
}