feat: speed

javascript-container/src/main.rs

@@ -1,33 +1,49 @@
+use std::time::Instant;
+
 use anyhow::Result;
 use wasmtime::{Config, Engine, Linker, Module, Store};
 
 fn main() -> Result<()> {
+    let instant = Instant::now();
     let config = Config::default();
     let engine = Engine::new(&config).unwrap();
+    println!("T1: {}ms", instant.elapsed().as_millis());
 
-    let wasm_bytes = include_bytes!("/Users/hatterjiang/Code/hattergit/secure-javascript-container/javascript-engine/target/wasm32-unknown-unknown/debug/javascript_engine.wasm");
-    let module = Module::new(&engine, wasm_bytes)?;
+    let wasm_bytes_release = include_bytes!("/Users/hatterjiang/Code/hattergit/secure-javascript-container/javascript-engine/target/wasm32-unknown-unknown/release/javascript_engine.wasm");
+    let wasm_bytes_debug = include_bytes!("/Users/hatterjiang/Code/hattergit/secure-javascript-container/javascript-engine/target/wasm32-unknown-unknown/debug/javascript_engine.wasm");
+    let is_release = std::env::var("RELEASE").unwrap_or_else(|_| "no".into()) == "yes";
+    println!("T1.2: {}ms", instant.elapsed().as_millis());
+    let module = Module::new(&engine, if is_release { &wasm_bytes_release[..] } else { &wasm_bytes_debug[..] })?;
+    println!("T2: {}ms", instant.elapsed().as_millis());
 
     let linker = Linker::new(&engine);
     let mut store = Store::new(&engine, ());
+    println!("T3: {}ms", instant.elapsed().as_millis());
 
     let instance = linker.instantiate(&mut store, &module)?;
+    println!("T4: {}ms", instant.elapsed().as_millis());
 
     let memory = instance.get_memory(&mut store, "memory").expect("SHOULD NOT HAPPEN");
     let malloc = instance.get_typed_func::<i32, i32>(&mut store, "malloc")?;
     let free = instance.get_typed_func::<i32, ()>(&mut store, "free")?;
     let eval = instance.get_typed_func::<i32, i32>(&mut store, "eval")?;
+    println!("T5: {}ms", instant.elapsed().as_millis());
 
-    let js = "1+2";
+    let js = std::env::var("JS").unwrap_or_else(|_| "1+1".into());
+    println!("JS: {}", js);
     let js_bytes = js.as_bytes();
     let js_bytes_len = js_bytes.len();
+    println!("T5.1: {}ms", instant.elapsed().as_millis());
     let ptr = malloc.call(&mut store, js.as_bytes().len() as i32)?;
+    println!("T5.2: {}ms", instant.elapsed().as_millis());
     let memory_data_mut = memory.data_mut(&mut store);
     for i in 0..js_bytes_len {
         memory_data_mut[4 + ptr as usize + i] = js_bytes[i];
     }
+    println!("T5.3: {}ms", instant.elapsed().as_millis());
     let eval_result = eval.call(&mut store, ptr)?;
     println!("Eval result: {:?}", eval_result);
+    println!("T6: {}ms", instant.elapsed().as_millis());
 
     let memory = instance.get_memory(&mut store, "memory").expect("SHOULD NOT HAPPEN");
     let memory_data_mut = memory.data_mut(&mut store);
@@ -44,6 +60,8 @@ fn main() -> Result<()> {
     }
     let ss = unsafe { String::from_utf8_unchecked(s) };
     println!("Result: {}", ss);
+    free.call(&mut store, eval_result)?;
+    println!("T7: {}ms", instant.elapsed().as_millis());
 
     Ok(())
 }