mirror of
https://github.com/denoland/deno.git
synced 2024-12-11 01:58:05 -05:00
340 lines
9.4 KiB
Rust
340 lines
9.4 KiB
Rust
// Copyright 2018-2023 the Deno authors. All rights reserved. MIT license.
|
|
|
|
use deno_core::serde::Deserialize;
|
|
use deno_core::serde_json;
|
|
use deno_core::serde_json::json;
|
|
use deno_core::serde_json::Value;
|
|
use deno_core::url::Url;
|
|
use std::collections::HashMap;
|
|
use std::path::Path;
|
|
use std::time::Duration;
|
|
use test_util::lsp::LspClientBuilder;
|
|
use tower_lsp::lsp_types as lsp;
|
|
|
|
static FIXTURE_CODE_LENS_TS: &str = include_str!("testdata/code_lens.ts");
|
|
static FIXTURE_DB_TS: &str = include_str!("testdata/db.ts");
|
|
static FIXTURE_DB_MESSAGES: &[u8] = include_bytes!("testdata/db_messages.json");
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
enum FixtureType {
|
|
#[serde(rename = "action")]
|
|
Action,
|
|
#[serde(rename = "change")]
|
|
Change,
|
|
#[serde(rename = "completion")]
|
|
Completion,
|
|
#[serde(rename = "highlight")]
|
|
Highlight,
|
|
#[serde(rename = "hover")]
|
|
Hover,
|
|
}
|
|
|
|
#[derive(Debug, Deserialize)]
|
|
struct FixtureMessage {
|
|
#[serde(rename = "type")]
|
|
fixture_type: FixtureType,
|
|
params: Value,
|
|
}
|
|
|
|
/// A benchmark that opens a 8000+ line TypeScript document, adds a function to
|
|
/// the end of the document and does a level of hovering and gets quick fix
|
|
/// code actions.
|
|
fn bench_big_file_edits(deno_exe: &Path) -> Duration {
|
|
let mut client = LspClientBuilder::new()
|
|
.use_diagnostic_sync(false)
|
|
.deno_exe(deno_exe)
|
|
.build();
|
|
client.initialize_default();
|
|
|
|
client.write_notification(
|
|
"textDocument/didOpen",
|
|
json!({
|
|
"textDocument": {
|
|
"uri": "file:///testdata/db.ts",
|
|
"languageId": "typescript",
|
|
"version": 1,
|
|
"text": FIXTURE_DB_TS
|
|
}
|
|
}),
|
|
);
|
|
|
|
let (id, method, _): (u64, String, Option<Value>) = client.read_request();
|
|
assert_eq!(method, "workspace/configuration");
|
|
|
|
client.write_response(
|
|
id,
|
|
json!({
|
|
"enable": true
|
|
}),
|
|
);
|
|
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
|
|
let messages: Vec<FixtureMessage> =
|
|
serde_json::from_slice(FIXTURE_DB_MESSAGES).unwrap();
|
|
|
|
for msg in messages {
|
|
match msg.fixture_type {
|
|
FixtureType::Action => {
|
|
client.write_request("textDocument/codeAction", msg.params);
|
|
}
|
|
FixtureType::Change => {
|
|
client.write_notification("textDocument/didChange", msg.params);
|
|
}
|
|
FixtureType::Completion => {
|
|
client.write_request("textDocument/completion", msg.params);
|
|
}
|
|
FixtureType::Highlight => {
|
|
client.write_request("textDocument/documentHighlight", msg.params);
|
|
}
|
|
FixtureType::Hover => {
|
|
client.write_request("textDocument/hover", msg.params);
|
|
}
|
|
}
|
|
}
|
|
|
|
client.write_request("shutdown", json!(null));
|
|
client.write_notification("exit", json!(null));
|
|
|
|
client.duration()
|
|
}
|
|
|
|
fn bench_code_lens(deno_exe: &Path) -> Duration {
|
|
let mut client = LspClientBuilder::new()
|
|
.use_diagnostic_sync(false)
|
|
.deno_exe(deno_exe)
|
|
.build();
|
|
client.initialize_default();
|
|
|
|
client.write_notification(
|
|
"textDocument/didOpen",
|
|
json!({
|
|
"textDocument": {
|
|
"uri": "file:///testdata/code_lens.ts",
|
|
"languageId": "typescript",
|
|
"version": 1,
|
|
"text": FIXTURE_CODE_LENS_TS
|
|
}
|
|
}),
|
|
);
|
|
|
|
let (id, method, _): (u64, String, Option<Value>) = client.read_request();
|
|
assert_eq!(method, "workspace/configuration");
|
|
|
|
client.write_response(
|
|
id,
|
|
json!({
|
|
"enable": true
|
|
}),
|
|
);
|
|
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
|
|
let res = client.write_request_with_res_as::<Vec<lsp::CodeLens>>(
|
|
"textDocument/codeLens",
|
|
json!({
|
|
"textDocument": {
|
|
"uri": "file:///testdata/code_lens.ts"
|
|
}
|
|
}),
|
|
);
|
|
assert!(!res.is_empty());
|
|
|
|
for code_lens in res {
|
|
client.write_request("codeLens/resolve", code_lens);
|
|
}
|
|
|
|
client.duration()
|
|
}
|
|
|
|
fn bench_find_replace(deno_exe: &Path) -> Duration {
|
|
let mut client = LspClientBuilder::new()
|
|
.use_diagnostic_sync(false)
|
|
.deno_exe(deno_exe)
|
|
.build();
|
|
client.initialize_default();
|
|
|
|
for i in 0..10 {
|
|
client.write_notification(
|
|
"textDocument/didOpen",
|
|
json!({
|
|
"textDocument": {
|
|
"uri": format!("file:///a/file_{i}.ts"),
|
|
"languageId": "typescript",
|
|
"version": 1,
|
|
"text": "console.log(\"000\");\n"
|
|
}
|
|
}),
|
|
);
|
|
}
|
|
|
|
for _ in 0..10 {
|
|
let (id, method, _) = client.read_request::<Value>();
|
|
assert_eq!(method, "workspace/configuration");
|
|
client.write_response(id, json!({ "enable": true }));
|
|
}
|
|
|
|
for _ in 0..3 {
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
}
|
|
|
|
for i in 0..10 {
|
|
let file_name = format!("file:///a/file_{i}.ts");
|
|
client.write_notification(
|
|
"textDocument/didChange",
|
|
lsp::DidChangeTextDocumentParams {
|
|
text_document: lsp::VersionedTextDocumentIdentifier {
|
|
uri: Url::parse(&file_name).unwrap(),
|
|
version: 2,
|
|
},
|
|
content_changes: vec![lsp::TextDocumentContentChangeEvent {
|
|
range: Some(lsp::Range {
|
|
start: lsp::Position {
|
|
line: 0,
|
|
character: 13,
|
|
},
|
|
end: lsp::Position {
|
|
line: 0,
|
|
character: 16,
|
|
},
|
|
}),
|
|
range_length: None,
|
|
text: "111".to_string(),
|
|
}],
|
|
},
|
|
);
|
|
}
|
|
|
|
for i in 0..10 {
|
|
let file_name = format!("file:///a/file_{i}.ts");
|
|
client.write_request(
|
|
"textDocument/formatting",
|
|
lsp::DocumentFormattingParams {
|
|
text_document: lsp::TextDocumentIdentifier {
|
|
uri: Url::parse(&file_name).unwrap(),
|
|
},
|
|
options: lsp::FormattingOptions {
|
|
tab_size: 2,
|
|
insert_spaces: true,
|
|
..Default::default()
|
|
},
|
|
work_done_progress_params: Default::default(),
|
|
},
|
|
);
|
|
}
|
|
|
|
for _ in 0..3 {
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
}
|
|
|
|
client.write_request("shutdown", json!(null));
|
|
client.write_notification("exit", json!(null));
|
|
|
|
client.duration()
|
|
}
|
|
|
|
/// A test that starts up the LSP, opens a single line document, and exits.
|
|
fn bench_startup_shutdown(deno_exe: &Path) -> Duration {
|
|
let mut client = LspClientBuilder::new()
|
|
.use_diagnostic_sync(false)
|
|
.deno_exe(deno_exe)
|
|
.build();
|
|
client.initialize_default();
|
|
|
|
client.write_notification(
|
|
"textDocument/didOpen",
|
|
json!({
|
|
"textDocument": {
|
|
"uri": "file:///a/file.ts",
|
|
"languageId": "typescript",
|
|
"version": 1,
|
|
"text": "console.log(Deno.args);\n"
|
|
}
|
|
}),
|
|
);
|
|
|
|
let (id, method, _) = client.read_request::<Value>();
|
|
assert_eq!(method, "workspace/configuration");
|
|
|
|
client.write_response(
|
|
id,
|
|
json!({
|
|
"enable": true
|
|
}),
|
|
);
|
|
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
let (method, _): (String, Option<Value>) = client.read_notification();
|
|
assert_eq!(method, "textDocument/publishDiagnostics");
|
|
|
|
client.write_request("shutdown", json!(null));
|
|
|
|
client.write_notification("exit", json!(null));
|
|
|
|
client.duration()
|
|
}
|
|
|
|
/// Generate benchmarks for the LSP server.
|
|
pub fn benchmarks(deno_exe: &Path) -> HashMap<String, i64> {
|
|
println!("-> Start benchmarking lsp");
|
|
let mut exec_times = HashMap::new();
|
|
|
|
println!(" - Simple Startup/Shutdown ");
|
|
let mut times = Vec::new();
|
|
for _ in 0..10 {
|
|
times.push(bench_startup_shutdown(deno_exe));
|
|
}
|
|
let mean =
|
|
(times.iter().sum::<Duration>() / times.len() as u32).as_millis() as i64;
|
|
println!(" ({} runs, mean: {}ms)", times.len(), mean);
|
|
exec_times.insert("startup_shutdown".to_string(), mean);
|
|
|
|
println!(" - Big Document/Several Edits ");
|
|
let mut times = Vec::new();
|
|
for _ in 0..5 {
|
|
times.push(bench_big_file_edits(deno_exe));
|
|
}
|
|
let mean =
|
|
(times.iter().sum::<Duration>() / times.len() as u32).as_millis() as i64;
|
|
println!(" ({} runs, mean: {}ms)", times.len(), mean);
|
|
exec_times.insert("big_file_edits".to_string(), mean);
|
|
|
|
println!(" - Find/Replace");
|
|
let mut times = Vec::new();
|
|
for _ in 0..10 {
|
|
times.push(bench_find_replace(deno_exe));
|
|
}
|
|
let mean =
|
|
(times.iter().sum::<Duration>() / times.len() as u32).as_millis() as i64;
|
|
println!(" ({} runs, mean: {}ms)", times.len(), mean);
|
|
exec_times.insert("find_replace".to_string(), mean);
|
|
|
|
println!(" - Code Lens");
|
|
let mut times = Vec::new();
|
|
for _ in 0..10 {
|
|
times.push(bench_code_lens(deno_exe));
|
|
}
|
|
let mean =
|
|
(times.iter().sum::<Duration>() / times.len() as u32).as_millis() as i64;
|
|
println!(" ({} runs, mean: {}ms)", times.len(), mean);
|
|
exec_times.insert("code_lens".to_string(), mean);
|
|
|
|
println!("<- End benchmarking lsp");
|
|
|
|
exec_times
|
|
}
|