Generate proto code into out directory

This avoids polluting the source tree. Signed-off-by: Pascal Bach <[email protected]>
hasheddan · May 30, 2022 · b14f805 · b14f805
1 parent d1efa0e
commit b14f805
Show file tree

Hide file tree

Showing 4 changed files with 5 additions and 668 deletions.
diff --git a/k8s-deviceplugin/build.rs b/k8s-deviceplugin/build.rs
@@ -1,9 +1,6 @@
 fn main() {
     tonic_build::configure()
-        .build_server(false)
-        .out_dir("src")
-        .compile(
-            &["proto/v1beta1.proto", "proto/v1alpha.proto"],
-            &["proto"],
-        ).unwrap();
+        .build_server(true)
+        .compile(&["proto/v1beta1.proto", "proto/v1alpha.proto"], &["proto"])
+        .unwrap();
 }
diff --git a/k8s-deviceplugin/src/deviceplugin.rs b/k8s-deviceplugin/src/deviceplugin.rs
diff --git a/k8s-deviceplugin/src/lib.rs b/k8s-deviceplugin/src/lib.rs
@@ -34,7 +34,7 @@
 //!
 
 pub mod v1beta1 {
-    include!("./v1beta1.rs");
+    tonic::include_proto!("v1beta1");
 
     /// Means that the device is healthy.
     pub const HEALTHY: &str = "Healthy";
@@ -67,7 +67,7 @@ pub mod v1beta1 {
 }
 
 pub mod v1alpha {
-    include!("./deviceplugin.rs");
+    tonic::include_proto!("deviceplugin");
 
     /// Means that the device is healthy.
     pub const HEALTHY: &str = "Healthy";