编译时如何将Rust构建脚本中的数据传递至主程序？

针对你当前的问题，推荐以下两种更优方案，解决IDE类型提示缺失且简化流程：

方案一：生成带明确类型的常量定义 #

修改构建脚本，直接生成包含类型标注的Rust常量，而非仅数组内容。这样IDE能识别常量类型，提供完整的代码提示。

build.rs 修改后代码： #

use std::env;
use std::fs::File;
use std::path::Path;
use std::io::Write;

let path = env::var("OUT_DIR").expect("OUT_DIR not set");
let path = Path::new(&path).join("tags.rs");
let mut output_file = File::create(&path).expect("Failed to create tags.rs");

// 写入带类型标注的常量定义
writeln!(output_file, "pub const TAGS: &[(&str, &str)] = &[")?;
for (service, tag) in &services_and_tags {
    writeln!(output_file, "    ({:?}, {:?}),", service, tag)?;
}
writeln!(output_file, "];")?;

main.rs 使用方式： #

// 直接引入常量，IDE可识别其类型
include!(concat!(env!("OUT_DIR"), "/tags.rs"));

// 使用时直接调用TAGS，自动获得类型提示
fn main() {
    for (service, tag) in TAGS {
        println!("{}: {}", service, tag);
    }
}

方案二：二进制序列化传递（适合不可读需求） #

使用二进制序列化库（如bincode）将数据序列化为二进制文件，主程序通过include_bytes!引入后反序列化。此方案文件无可读性，且主程序可明确指定类型，IDE能提供完整提示。

步骤1：添加依赖到Cargo.toml #

[dependencies]
bincode = "1.3"
serde = { version = "1.0", features = ["derive"] }

[build-dependencies]
bincode = "1.3"
serde = { version = "1.0", features = ["derive"] }

步骤2：build.rs 序列化数据 #

use bincode;
use serde::Serialize;
use std::env;
use std::fs::File;
use std::path::Path;

// 为数据结构派生Serialize trait
#[derive(Serialize)]
struct ServiceTag(String, String);

// 转换为可序列化的结构
let serializable_data: Vec<ServiceTag> = services_and_tags
    .into_iter()
    .map(|(s, t)| ServiceTag(s, t))
    .collect();

let path = env::var("OUT_DIR").expect("OUT_DIR not set");
let path = Path::new(&path).join("tags.bin");
let data = bincode::serialize(&serializable_data).expect("Failed to serialize data");
File::create(&path).expect("Failed to create tags.bin").write_all(&data).expect("Failed to write data");

步骤3：main.rs 反序列化使用 #

use bincode;
use serde::Deserialize;

#[derive(Deserialize)]
struct ServiceTag(String, String);

// 引入二进制数据
const TAGS_BIN: &[u8] = include_bytes!(concat!(env!("OUT_DIR"), "/tags.bin"));

fn main() {
    // 明确指定反序列化类型，IDE可识别Vec<ServiceTag>的结构
    let tags: Vec<ServiceTag> = bincode::deserialize(TAGS_BIN).expect("Failed to deserialize data");
    for ServiceTag(service, tag) in tags {
        println!("{}: {}", service, tag);
    }
}

方案对比 #

• 方案一适合需要保留Rust代码可读性、追求简单实现的场景，IDE提示友好。
• 方案二完全满足不可读需求，数据体积更小，适合大量数据传递的场景。

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