-
Notifications
You must be signed in to change notification settings - Fork 1
/
build.rs
30 lines (26 loc) · 868 Bytes
/
build.rs
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
fn main() {
gen_pb_code()
}
#[cfg(not(feature = "gen"))]
fn gen_pb_code() {}
#[cfg(feature = "gen")]
fn gen_pb_code() {
// Build auth.proto
tonic_build::configure()
.build_server(false)
.out_dir("src/pb/")
.compile(&["proto/auth.proto"], &["proto/"])
.unwrap_or_else(|e| panic!("protobuf compilation failed: {}", e));
// Build kv.proto
tonic_build::configure()
.build_server(false)
.out_dir("src/pb/")
.compile(&["proto/kv.proto"], &["proto/"])
.unwrap_or_else(|e| panic!("protobuf compilation failed: {}", e));
// Build rpc.proto
tonic_build::configure()
.build_server(false)
.out_dir("src/pb/")
.compile(&["proto/rpc.proto"], &["proto/"])
.unwrap_or_else(|e| panic!("protobuf compilation failed: {}", e));
}