Question

假设我有一个包含一些常量的文件。此文件可能来自不同的服务，并且对于不同的场景可能会有所不同。现在我想从我的程序中挤出最后一点性能，希望编译器做尽可能多的优化，如果需要，我会为每个场景单独编译项目。

有没有办法在编译时获取文件，通过一些转换逻辑将其转换为 Rust 代码，例如生成一些常量，并编译结果（全部在 rust 工具中，没有其他代码生成）？

Answer 1

如果您在 crate 的根目录中编写一个 build.rs 文件，它将在每次 crate 编译时进行编译和运行。

它通常用于构建 C 绑定等，但没有什么可以阻止您将其用于自己的目的。

通常这个 build.rs 在输出目录的某处创建一个 Rust 源文件，读取 OUT_DIR 环境变量：

fn main() {
    println!("cargo:rerun-if-changed=build.rs");
    println!("cargo:rerun-if-changed=data.txt");

    let out_dir = std::env::var_os("OUT_DIR").unwrap();
    let path = std::path::Path::new(&out_dir).join("test.rs");
    std::fs::write(&path, "pub fn test() { todo!() }").unwrap();
}

然后将源代码包含在您的项目中，通常在它自己的模块中：

mod test {
    include!(concat!(env!("OUT_DIR"), "/test.rs"));
}

Answer 2

您可以只生成包含常量的 Rust 源文件。

这些部分可以是任何语言：

生成要用作常量的数据
使用一些字符串操作将它们作为常量保存到 Rust 源文件中

生锈：

pub mod const_file 在 path/to/mod.rs
use path::to::const_file::MY_CONST;
cargo build 你的 Rust 程序

从 python 获取的常量数组示例：

array = [0, 10, 34]
const_file_str = f"pub const MY_CONST: [u16; 3] = {array};\n"

with open("path/to/const_file.rs", "w") as rust_file:
    rust_file.write(const_file_str)

从文件内容生成常量ar编译时间

2 个答案: