mirror of
https://github.com/denoland/deno.git
synced 2024-12-27 01:29:14 -05:00
28b5a7e2ec
This commit allows to execute more JS code from extensions when creating a snapshot from an existing snapshot. "deno_core::RuntimeOptions::extensions_with_js" field was added that is used to pass a list of extensions whose both "ops" and associated JS source should be executed upon start. Co-authored-by: crowlkats <crowlkats@toaxl.com>
121 lines
2.8 KiB
Rust
121 lines
2.8 KiB
Rust
// Copyright 2018-2022 the Deno authors. All rights reserved. MIT license.
|
|
use bencher::Bencher;
|
|
use deno_core::v8;
|
|
use deno_core::Extension;
|
|
use deno_core::JsRuntime;
|
|
use deno_core::RuntimeOptions;
|
|
|
|
use crate::profiling::is_profiling;
|
|
|
|
pub fn create_js_runtime(setup: impl FnOnce() -> Vec<Extension>) -> JsRuntime {
|
|
JsRuntime::new(RuntimeOptions {
|
|
extensions_with_js: setup(),
|
|
..Default::default()
|
|
})
|
|
}
|
|
|
|
fn loop_code(iters: u64, src: &str) -> String {
|
|
format!(r#"for(let i=0; i < {}; i++) {{ {} }}"#, iters, src,)
|
|
}
|
|
|
|
#[derive(Copy, Clone)]
|
|
pub struct BenchOptions {
|
|
pub benching_inner: u64,
|
|
pub profiling_inner: u64,
|
|
pub profiling_outer: u64,
|
|
}
|
|
|
|
impl Default for BenchOptions {
|
|
fn default() -> Self {
|
|
Self {
|
|
benching_inner: 1_000,
|
|
profiling_inner: 1_000,
|
|
profiling_outer: 10_000,
|
|
}
|
|
}
|
|
}
|
|
|
|
pub fn bench_js_sync(
|
|
b: &mut Bencher,
|
|
src: &str,
|
|
setup: impl FnOnce() -> Vec<Extension>,
|
|
) {
|
|
bench_js_sync_with(b, src, setup, Default::default())
|
|
}
|
|
|
|
pub fn bench_js_sync_with(
|
|
b: &mut Bencher,
|
|
src: &str,
|
|
setup: impl FnOnce() -> Vec<Extension>,
|
|
opts: BenchOptions,
|
|
) {
|
|
let mut runtime = create_js_runtime(setup);
|
|
let scope = &mut runtime.handle_scope();
|
|
|
|
// Increase JS iterations if profiling for nicer flamegraphs
|
|
let inner_iters = if is_profiling() {
|
|
opts.profiling_inner * opts.profiling_outer
|
|
} else {
|
|
opts.benching_inner
|
|
};
|
|
// Looped code
|
|
let looped_src = loop_code(inner_iters, src);
|
|
|
|
let code = v8::String::new(scope, looped_src.as_ref()).unwrap();
|
|
let script = v8::Script::compile(scope, code, None).unwrap();
|
|
// Run once if profiling, otherwise regular bench loop
|
|
if is_profiling() {
|
|
script.run(scope).unwrap();
|
|
} else {
|
|
b.iter(|| {
|
|
script.run(scope).unwrap();
|
|
});
|
|
}
|
|
}
|
|
|
|
pub fn bench_js_async(
|
|
b: &mut Bencher,
|
|
src: &str,
|
|
setup: impl FnOnce() -> Vec<Extension>,
|
|
) {
|
|
bench_js_async_with(b, src, setup, Default::default())
|
|
}
|
|
|
|
pub fn bench_js_async_with(
|
|
b: &mut Bencher,
|
|
src: &str,
|
|
setup: impl FnOnce() -> Vec<Extension>,
|
|
opts: BenchOptions,
|
|
) {
|
|
let mut runtime = create_js_runtime(setup);
|
|
let tokio_runtime = tokio::runtime::Builder::new_current_thread()
|
|
.enable_all()
|
|
.build()
|
|
.unwrap();
|
|
|
|
// Looped code
|
|
let inner_iters = if is_profiling() {
|
|
opts.profiling_inner
|
|
} else {
|
|
opts.benching_inner
|
|
};
|
|
let looped = loop_code(inner_iters, src);
|
|
let src = looped.as_ref();
|
|
runtime
|
|
.execute_script("init", "Deno.core.initializeAsyncOps();")
|
|
.unwrap();
|
|
if is_profiling() {
|
|
for _ in 0..opts.profiling_outer {
|
|
tokio_runtime.block_on(inner_async(src, &mut runtime));
|
|
}
|
|
} else {
|
|
b.iter(|| {
|
|
tokio_runtime.block_on(inner_async(src, &mut runtime));
|
|
});
|
|
}
|
|
}
|
|
|
|
async fn inner_async(src: &str, runtime: &mut JsRuntime) {
|
|
runtime.execute_script("inner_loop", src).unwrap();
|
|
runtime.run_event_loop(false).await.unwrap();
|
|
}
|