能否扩展Rust Rocket的路由属性宏以自动完成路由配置？

技术上完全可以实现，不需要修改Rocket源码，有两种成熟的实现思路：

方案1：构建脚本+代码生成（最稳定无黑魔法） #

这种方案利用Rust编译期的build.rs脚本，在编译时自动扫描所有源码文件识别路由函数，自动生成包含所有路由的宏，是无副作用、兼容性最好的方案。

实现逻辑： #

1. 在项目根目录创建build.rs构建脚本，引入syn、walkdir依赖用于解析Rust代码AST和扫描源码文件
2. 递归扫描src目录下所有.rs文件，识别所有标注了Rocket原生路由属性（#[get]/#[post]等）的函数名
3. 自动生成all_routes!宏的定义，写入编译输出目录的自动生成文件中
4. 业务代码中直接引入自动生成的宏，挂载路由时调用即可

核心代码示例 #

build.rs 简化实现：

use std::env;
use std::fs;
use std::path::Path;
use syn::{File, Item, Attribute};
use walkdir::WalkDir;

fn is_rocket_route_attr(attr: &Attribute) -> bool {
    let ident = attr.path().get_ident().map(|i| i.to_string()).unwrap_or_default();
    matches!(ident.as_str(), "get" | "post" | "put" | "delete" | "patch" | "head" | "options")
}

fn main() {
    let mut route_names = Vec::new();
    // 扫描src下所有Rust文件
    for entry in WalkDir::new("src")
        .into_iter()
        .filter_map(|e| e.ok())
        .filter(|e| e.path().extension().map_or(false, |ext| ext == "rs"))
    {
        let content = fs::read_to_string(entry.path()).unwrap();
        let ast = syn::parse_str::<File>(&content).unwrap();
        for item in ast.items {
            if let Item::Fn(func) = item {
                if func.attrs.iter().any(is_rocket_route_attr) {
                    route_names.push(func.sig.ident.to_string());
                }
            }
        }
    }
    // 生成路由宏代码
    let generated_code = format!(
        "macro_rules! all_routes {{ () => {{ routes![{}] }}; }}",
        route_names.join(", ")
    );
    // 写入编译输出目录
    let out_dir = env::var_os("OUT_DIR").unwrap();
    fs::write(Path::new(&out_dir).join("all_routes.rs"), generated_code).unwrap();
    // 源码变更时重新运行构建脚本
    println!("cargo:rerun-if-changed=src");
}

main.rs 调用示例：

#[macro_use] extern crate rocket;
// 导入自动生成的宏
include!(concat!(env!("OUT_DIR"), "/all_routes.rs"));

#[get("/hello/<name>/<age>")]
fn hello(name: &str, age: u8) -> String {
    format!("Hello, {} year old named {}!", age, name)
}

// 其他任意路由函数，不需要手动注册
#[get("/about")]
fn about() -> &'static str {
    "about page"
}

#[launch]
fn rocket() -> _ {
    // 直接调用自动生成的宏即可挂载所有路由
    rocket::build().mount("/", all_routes!())
}

方案2：自定义过程宏+全局路由注册表 #

这种方案通过封装Rocket原生路由宏，在宏展开阶段自动将路由注册到全局静态容器中，适合需要更灵活配置（比如分前缀挂载）的场景。

实现逻辑： #

1. 定义自定义属性宏（比如#[auto_route]），内部先调用Rocket对应的原生路由宏处理函数，再将当前路由函数的标识符插入到全局静态注册表中（可以用once_cell或者lazy_static实现线程安全的全局容器）
2. 定义mount_auto_routes!宏，从全局注册表中取出所有路由，传给Rocket的mount方法

注意事项： #

需要确保所有带#[auto_route]的模块都被编译器引入，否则宏不会展开，对应的路由不会被注册到全局容器中。

两种方案都完全兼容现有Rocket的API，不需要对Rocket本身做任何修改，也不违背Rust的语法规则，生产环境已经有不少项目在用类似的封装减少重复劳动。

内容的提问来源于stack exchange，提问作者at54321