2021-09-29 20:25:05 +02:00
|
|
|
use std::time::Instant;
|
|
|
|
|
|
|
|
use nu_engine::eval_block;
|
|
|
|
use nu_protocol::ast::Call;
|
|
|
|
use nu_protocol::engine::{Command, EvaluationContext};
|
2021-10-25 06:01:02 +02:00
|
|
|
use nu_protocol::{PipelineData, Signature, SyntaxShape, Value};
|
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,
|
|
|
|
context: &EvaluationContext,
|
|
|
|
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 06:01:02 +02:00
|
|
|
let block = context.engine_state.get_block(block);
|
2021-09-29 20:25:05 +02:00
|
|
|
|
|
|
|
let state = context.enter_scope();
|
|
|
|
let start_time = Instant::now();
|
2021-10-25 06:01:02 +02:00
|
|
|
eval_block(&state, block, PipelineData::new())?;
|
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
|
|
|
}
|
|
|
|
}
|