wip

javierhonduco · Jun 11, 2024 · ce0e4c5 · ce0e4c5
1 parent 38e286c
commit ce0e4c5
Show file tree

Hide file tree

Showing 6 changed files with 289 additions and 72 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -35,6 +35,7 @@ assert_cmd = { version = "2.0.14" }
 insta = { version = "1.38.0", features = ["yaml"] }
 rstest = "0.19.0"
 tempdir = "0.3.7"
+rand = "0.8.5"
 
 [build-dependencies]
 bindgen = "0.69.4"

diff --git a/build.rs b/build.rs
@@ -74,6 +74,11 @@ fn main() {
 
     // Proto-related code.
 
-    prost_build::compile_protos(&["src/proto/profile.proto"], &["src/proto"])
+    let mut config = prost_build::Config::new();
+    config.type_attribute("Sample", "#[derive(Eq, Hash)]");
+    config.type_attribute("Label", "#[derive(Eq, Hash)]");
+
+    config
+        .compile_protos(&["src/proto/profile.proto"], &["src/proto"])
         .expect("build profile.proto");
 }
diff --git a/profile.pb b/profile.pb
diff --git a/src/lib.rs b/src/lib.rs
@@ -5,7 +5,7 @@ pub mod object;
 pub mod perf_events;
 pub mod profile;
 pub mod profiler;
+pub mod proto;
 pub mod unwind_info;
 pub mod usym;
 pub mod util;
-pub mod proto;