如何在Rust中通过宏将文件内容编译时加载为静态数组？

要实现你需要的define_included_bytes!宏，得利用Rust的过程宏和编译期文件读取能力，普通声明宏没法直接获取文件长度并生成对应长度的数组定义。以下是具体实现步骤：

1. 添加依赖 #

先在Cargo.toml里配置过程宏相关依赖：

[lib]
proc-macro = true

[dependencies]
proc-macro2 = "1.0"
quote = "1.0"
syn = { version = "2.0", features = ["full"] }

2. 编写过程宏 #

创建过程宏库，在src/lib.rs中写入以下代码：

use proc_macro::TokenStream;
use proc_macro2::Span;
use quote::quote;
use syn::{parse_macro_input, Ident, LitStr};
use std::fs;
use std::path::Path;

#[proc_macro]
pub fn define_included_bytes(input: TokenStream) -> TokenStream {
    // 解析宏输入：静态变量名和文件路径
    let input = parse_macro_input!(input as (Ident, LitStr));
    let (var_name, path_lit) = input;
    let path = path_lit.value();
    
    // 编译期读取文件内容
    let bytes = fs::read(Path::new(&path))
        .expect(&format!("Failed to read file: {}", path));
    
    // 生成数组长度和元素列表
    let len = bytes.len();
    let elements = bytes.iter().map(|b| quote!(#b));
    
    // 展开为静态数组定义
    let expanded = quote! {
        static #var_name: [u8; #len] = [ #(#elements),* ];
    };
    
    expanded.into()
}

3. 使用宏 #

在主项目中引入宏库后直接调用：

use your_macro_crate::define_included_bytes;

define_included_bytes!(PROG_ROM, "foo.dat");

// 直接使用PROG_ROM，类型为[u8; N]（N是文件实际字节长度）
fn main() {
    println!("File length: {}", PROG_ROM.len());
}

关键说明 #

• 该宏在编译期读取文件，数组内容直接嵌入最终二进制，完全满足静态数组的需求。
• 无需硬编码长度，宏会自动获取文件实际字节数，生成对应长度的数组类型。
• 过程宏支持编译期IO操作与动态代码生成，这是普通声明宏无法实现的能力。

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