From 6ab39bfde817f20703a94fbef5d3540862a324ef Mon Sep 17 00:00:00 2001
From: angelip2303 <angel.iglesias.prestamo@gmail.com>
Date: Tue, 27 Jun 2023 09:48:59 +0000
Subject: [PATCH] improving the work done :D

---
 examples/from_uniprot/main.rs | 27 ++++++++++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/examples/from_uniprot/main.rs b/examples/from_uniprot/main.rs
index 817b1d8..7a1219c 100644
--- a/examples/from_uniprot/main.rs
+++ b/examples/from_uniprot/main.rs
@@ -1,3 +1,5 @@
+use std::time::Instant;
+
 use pregel_rs::graph_frame::GraphFrame;
 use pschema_rs::backends::ntriples::NTriples;
 use pschema_rs::backends::parquet::Parquet;
@@ -5,9 +7,23 @@ use pschema_rs::backends::Backend;
 use pschema_rs::pschema::PSchema;
 use pschema_rs::shape::shex::{ShapeAnd, ShapeReference, TripleConstraint};
 
+#[cfg(not(target_env = "msvc"))]
+use jemallocator::Jemalloc;
+
+#[cfg(target_env = "msvc")]
+use mimalloc::MiMalloc;
+
+#[cfg(not(target_env = "msvc"))]
+#[global_allocator]
+static GLOBAL: Jemalloc = Jemalloc;
+
+#[cfg(target_env = "msvc")]
+#[global_allocator]
+static GLOBAL: MiMalloc = MiMalloc;
+
 fn main() -> Result<(), String> {
     // Define validation rules
-    let start = ShapeAnd::new(
+    let shape = ShapeAnd::new(
         "protein",
         vec![
             TripleConstraint::new(
@@ -35,9 +51,14 @@ fn main() -> Result<(), String> {
     let edges = NTriples::import("./examples/from_uniprot/uniprotkb_reviewed_viruses_10239.nt")?;
 
     // Perform schema validation
+    let start = Instant::now();
     match GraphFrame::from_edges(edges) {
-        Ok(graph) => match PSchema::new(start).validate(graph) {
-            Ok(subset) => Parquet::export("uniprotkb_reviewed_viruses_10239.parquet", subset),
+        Ok(graph) => match PSchema::new(shape).validate(graph) {
+            Ok(subset) => {
+                let duration = start.elapsed();
+                println!("Time elapsed in validate() is: {:?}", duration);
+                Parquet::export("uniprotkb_reviewed_viruses_10239.parquet", subset)
+            }
             Err(error) => Err(error.to_string()),
         },
         Err(error) => Err(format!("Cannot create a GraphFrame: {}", error)),