use std::env;
fn main() {
if env::var("PROTO_CODE_GEN").unwrap_or("0".to_string()) != "1" {
return;
}
let generated_out_dir = "src/generated";
std::fs::create_dir_all(generated_out_dir)
.unwrap_or_else(|e| panic!("failed to create generated output directory: {:?}", e));
tonic_prost_build::configure()
.out_dir(generated_out_dir)
.compile_protos(
&[
"proto/source.proto",
"proto/sourcetransform.proto",
"proto/map.proto",
"proto/reduce.proto",
"proto/sink.proto",
"proto/sideinput.proto",
"proto/store.proto",
"proto/sessionreduce.proto",
"proto/accumulator.proto",
],
&["proto"],
)
.unwrap_or_else(|e| panic!("failed to compile the proto, {:?}", e));
}