From 3d00a45a06c41141011549af87c6fed974f00447 Mon Sep 17 00:00:00 2001
From: gusto <bacvinka@gmail.com>
Date: Thu, 27 Jun 2024 11:35:43 +0300
Subject: [PATCH] Use write_to_vec for profile data

---
 nomos-da/kzgrs-backend/Cargo.toml     | 2 +-
 nomos-da/kzgrs-backend/src/encoder.rs | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/nomos-da/kzgrs-backend/Cargo.toml b/nomos-da/kzgrs-backend/Cargo.toml
index 762fa52a..22e1d62a 100644
--- a/nomos-da/kzgrs-backend/Cargo.toml
+++ b/nomos-da/kzgrs-backend/Cargo.toml
@@ -26,7 +26,7 @@ rayon = { version = "1.10.0", optional = true }
 rand = "0.8"
 ark-bls12-381 = "0.4.0"
 divan = "0.1"
-pprof = { version = "0.13.0" }
+pprof = { version = "0.13.0", features = ['protobuf-codec', 'protobuf', 'cpp' ]}
 
 [features]
 default = []
diff --git a/nomos-da/kzgrs-backend/src/encoder.rs b/nomos-da/kzgrs-backend/src/encoder.rs
index 6c6fe41c..1856a5b5 100644
--- a/nomos-da/kzgrs-backend/src/encoder.rs
+++ b/nomos-da/kzgrs-backend/src/encoder.rs
@@ -263,6 +263,7 @@ pub mod test {
         decode, verify_element_proof, FieldElement, PolynomialEvaluationDomain,
         BYTES_PER_FIELD_ELEMENT,
     };
+    use pprof::protos::Message;
     use rand::RngCore;
     use std::ops::Div;
 
@@ -472,7 +473,7 @@ pub mod test {
     fn profile_full_encode_flow() {
         // use ~1MB worth of data
         let data = rand_data(1024 * 1024 / DaEncoderParams::MAX_BLS12_381_ENCODING_CHUNK_SIZE);
-        let domain = GeneralEvaluationDomain::new(4096).unwrap();
+        let domain = PolynomialEvaluationDomain::new(4096).unwrap();
         let encoder = DaEncoder::new(DaEncoderParams::new(4096, true));
 
         let guard = pprof::ProfilerGuardBuilder::default()
@@ -487,9 +488,9 @@ pub mod test {
                 let profile = report.pprof().unwrap();
 
                 let mut content = Vec::new();
-                profile.encode(&mut content).unwrap();
+                profile.write_to_vec(&mut content).unwrap();
                 std::io::Write::write_all(&mut file, &content).unwrap();
-                println!("report: {}", &report);
+                println!("report: {:?}", &report);
             }
             Err(_) => {}
         };