nushell/crates/nu-command/src/system/benchmark.rs

49 lines
1.3 KiB
Rust
Raw Normal View History

2021-09-29 20:25:05 +02:00
use std::time::Instant;
use nu_engine::eval_block;
use nu_protocol::ast::Call;
2021-10-25 18:58:58 +02:00
use nu_protocol::engine::{Command, EngineState, Stack};
use nu_protocol::{PipelineData, Signature, SyntaxShape};
2021-09-29 20:25:05 +02:00
2021-10-25 06:01:02 +02:00
#[derive(Clone)]
2021-09-29 20:25:05 +02:00
pub struct Benchmark;
impl Command for Benchmark {
fn name(&self) -> &str {
"benchmark"
}
fn usage(&self) -> &str {
"Time the running time of a block"
}
fn signature(&self) -> nu_protocol::Signature {
Signature::build("benchmark").required(
"block",
SyntaxShape::Block(Some(vec![])),
"the block to run",
)
}
fn run(
&self,
2021-10-25 08:31:39 +02:00
engine_state: &EngineState,
stack: &mut Stack,
2021-09-29 20:25:05 +02:00
call: &Call,
2021-10-25 06:01:02 +02:00
_input: PipelineData,
) -> Result<nu_protocol::PipelineData, nu_protocol::ShellError> {
2021-09-29 20:25:05 +02:00
let block = call.positional[0]
.as_block()
.expect("internal error: expected block");
2021-10-25 08:31:39 +02:00
let block = engine_state.get_block(block);
2021-09-29 20:25:05 +02:00
2021-10-25 22:04:23 +02:00
let mut stack = stack.collect_captures(&block.captures);
2021-09-29 20:25:05 +02:00
let start_time = Instant::now();
2021-10-25 18:58:58 +02:00
eval_block(engine_state, &mut stack, block, PipelineData::new())?.into_value();
2021-09-29 20:25:05 +02:00
let end_time = Instant::now();
println!("{} ms", (end_time - start_time).as_millis());
2021-10-25 06:01:02 +02:00
Ok(PipelineData::new())
2021-09-29 20:25:05 +02:00
}
}