Splitting profile_proto example

Cargo.toml

@@ -59,9 +59,13 @@ name = "flamegraph"
 required-features = ["flamegraph"]
 
 [[example]]
-name = "profile_proto"
+name = "profile_proto_with_prost"
 required-features = ["protobuf", "prost-codec"]
 
+[[example]]
+name = "profile_proto_with_protobuf_codec"
+required-features = ["protobuf", "protobuf-codec"]
+
 [[example]]
 name = "multithread_flamegraph"
 required-features = ["flamegraph"]

examples/profile_proto_with_prost.rs

@@ -0,0 +1,112 @@
+// Copyright 2019 TiKV Project Authors. Licensed under Apache-2.0.
+
+use pprof::protos::Message;
+use std::fs::File;
+use std::io::Write;
+
+#[inline(never)]
+fn is_prime_number1(v: usize, prime_numbers: &[usize]) -> bool {
+ if v < 10000 {
+ let r = prime_numbers.binary_search(&v);
+ return r.is_ok();
+ }
+
+ for n in prime_numbers {
+ if v % n == 0 {
+ return false;
+ }
+ }
+
+ true
+}
+
+#[inline(always)]
+fn is_prime_number2(v: usize, prime_numbers: &[usize]) -> bool {
+ if v < 10000 {
+ let r = prime_numbers.binary_search(&v);
+ return r.is_ok();
+ }
+
+ for n in prime_numbers {
+ if v % n == 0 {
+ return false;
+ }
+ }
+
+ true
+}
+
+#[inline(never)]
+fn is_prime_number3(v: usize, prime_numbers: &[usize]) -> bool {
+ if v < 10000 {
+ let r = prime_numbers.binary_search(&v);
+ return r.is_ok();
+ }
+
+ for n in prime_numbers {
+ if v % n == 0 {
+ return false;
+ }
+ }
+
+ true
+}
+
+#[inline(never)]
+fn prepare_prime_numbers() -> Vec<usize> {
+ // bootstrap: Generate a prime table of 0..10000
+ let mut prime_number_table: [bool; 10000] = [true; 10000];
+ prime_number_table[0] = false;
+ prime_number_table[1] = false;
+ for i in 2..10000 {
+ if prime_number_table[i] {
+ let mut v = i * 2;
+ while v < 10000 {
+ prime_number_table[v] = false;
+ v += i;
+ }
+ }
+ }
+ let mut prime_numbers = vec![];
+ for (i, exist) in prime_number_table.iter().enumerate().skip(2) {
+ if *exist {
+ prime_numbers.push(i);
+ }
+ }
+ prime_numbers
+}
+
+fn main() {
+ let prime_numbers = prepare_prime_numbers();
+
+ let guard = pprof::ProfilerGuard::new(100).unwrap();
+
+ let mut v = 0;
+
+ for i in 2..5000000 {
+ if i % 4 == 0 {
+ if is_prime_number1(i, &prime_numbers) {
+ v += 1;
+ }
+ } else if i % 4 == 1 {
+ if is_prime_number2(i, &prime_numbers) {
+ v += 1;
+ }
+ } else if is_prime_number3(i, &prime_numbers) {
+ v += 1;
+ }
+ }
+
+ println!("Prime numbers: {}", v);
+
+ if let Ok(report) = guard.report().build() {
+ let mut file = File::create("profile.pb").unwrap();
+ let profile = report.pprof().unwrap();
+
+ let mut content = Vec::new();
+ profile.encode(&mut content).unwrap();
+ file.write_all(&content).unwrap();
+
+ println!("report: {:?}", report);
+ };
+}

examples/profile_proto.rs → examples/profile_proto_with_protobuf_codec.rs

@@ -104,9 +104,6 @@ fn main() {
  let profile = report.pprof().unwrap();
 
  let mut content = Vec::new();
- #[cfg(not(feature = "protobuf-codec"))]
- profile.encode(&mut content).unwrap();
- #[cfg(feature = "protobuf-codec")]
  profile.write_to_vec(&mut content).unwrap();
  file.write_all(&content).unwrap();