432222f098
Makes use of https://github.com/tokio-rs/prost/pull/341, which makes our bytes field cheaper to clone. It's a bit annoying to configure due to https://github.com/hyperium/tonic/issues/908, but the workaround does get the job done. Change-Id: I25714600b041bb5432d3adf5859b151e72b12778 Reviewed-on: https://cl.tvl.fyi/c/depot/+/8975 Reviewed-by: raitobezarius <tvl@lahfa.xyz> Tested-by: BuildkiteCI Reviewed-by: tazjin <tazjin@tvl.su> Autosubmit: flokli <flokli@flokli.de>
40 lines
1.4 KiB
Rust
40 lines
1.4 KiB
Rust
use std::io::Result;
|
|
|
|
fn main() -> Result<()> {
|
|
#[allow(unused_mut)]
|
|
let mut builder = tonic_build::configure();
|
|
|
|
#[cfg(feature = "reflection")]
|
|
{
|
|
let out_dir = std::path::PathBuf::from(std::env::var("OUT_DIR").unwrap());
|
|
let descriptor_path = out_dir.join("tvix.store.v1.bin");
|
|
|
|
builder = builder.file_descriptor_set_path(descriptor_path);
|
|
};
|
|
|
|
// https://github.com/hyperium/tonic/issues/908
|
|
let mut config = prost_build::Config::new();
|
|
config.bytes(["."]);
|
|
|
|
builder
|
|
.build_server(true)
|
|
.build_client(true)
|
|
.compile_with_config(
|
|
config,
|
|
&[
|
|
"tvix/store/protos/castore.proto",
|
|
"tvix/store/protos/pathinfo.proto",
|
|
"tvix/store/protos/rpc_blobstore.proto",
|
|
"tvix/store/protos/rpc_directory.proto",
|
|
"tvix/store/protos/rpc_pathinfo.proto",
|
|
],
|
|
// If we are in running `cargo build` manually, using `../..` works fine,
|
|
// but in case we run inside a nix build, we need to instead point PROTO_ROOT
|
|
// to a sparseTree containing that structure.
|
|
&[match std::env::var_os("PROTO_ROOT") {
|
|
Some(proto_root) => proto_root.to_str().unwrap().to_owned(),
|
|
None => "../..".to_string(),
|
|
}],
|
|
)?;
|
|
Ok(())
|
|
}
|