build.rs 563 B

12345678910111213141516171819
  1. extern crate rustc_version;
  2. use std::io::Result;
  3. fn main() -> Result<()> {
  4. const PROTOC_ENVAR: &str = "PROTOC";
  5. if std::env::var(PROTOC_ENVAR).is_err() {
  6. #[cfg(not(windows))]
  7. std::env::set_var(PROTOC_ENVAR, protobuf_src::protoc());
  8. }
  9. let proto_base_path = std::path::PathBuf::from("proto");
  10. let proto = proto_base_path.join("wen_restart.proto");
  11. println!("cargo:rerun-if-changed={}", proto.display());
  12. // Generate rust files from protos.
  13. prost_build::compile_protos(&[proto], &[proto_base_path])?;
  14. Ok(())
  15. }