snowykami/meilisearch
1
0
Fork 0
mirror of https://github.com/meilisearch/meilisearch.git synced 2024-11-26 03:55:07 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Always keep the new threads and new callsites

Browse Source
This commit is contained in:
Clément Renault 2024-02-08 16:44:28 +01:00
parent 173aad6090
commit e64ff1fa0c
No known key found for this signature in database
GPG Key ID: F250A4C4E3AE5F5F
1 changed files with 12 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
tracing-trace/src/bin/filter-trace.rs
View File

@ -1,6 +1,6 @@
use std::collections::vec_deque::Drain; use std::collections::vec_deque::Drain;
use std::collections::VecDeque; use std::collections::VecDeque;
use std::io::{self, BufReader, Write}; use std::io::{self, BufReader, BufWriter, Stdout, Write};
use std::mem; use std::mem;
use anyhow::Context; use anyhow::Context;
@ -32,34 +32,36 @@ fn main() -> anyhow::Result<()> {
let mut output = io::BufWriter::new(io::stdout()); let mut output = io::BufWriter::new(io::stdout());
for result in tracing_trace::TraceReader::new(input) { for result in tracing_trace::TraceReader::new(input) {
let entry = result?; let entry = result?;
if entry.memory().map_or(true, |m| m.resident < memory_threshold.as_u64()) { if matches!(entry, Entry::NewCallsite(_) | Entry::NewThread(_)) {
write_to_output(&mut output, &entry)?;
} else if entry.memory().map_or(true, |m| m.resident < memory_threshold.as_u64()) {
if mem::replace(&mut currently_in_threshold, false) { if mem::replace(&mut currently_in_threshold, false) {
for entry in context.drain() { for entry in context.drain() {
serde_json::to_writer(&mut output, &entry) write_to_output(&mut output, &entry)?;
.context("while serializing and writing to stdout")?;
} }
} }
context.push(entry); context.push(entry);
} else { } else {
currently_in_threshold = true; currently_in_threshold = true;
for entry in context.drain() { for entry in context.drain() {
serde_json::to_writer(&mut output, &entry) write_to_output(&mut output, &entry)?;
.context("while serializing and writing to stdout")?;
} }
serde_json::to_writer(&mut output, &entry) write_to_output(&mut output, &entry)?;
.context("while serializing and writing to stdout")?;
} }
} }
for entry in context.drain() { for entry in context.drain() {
serde_json::to_writer(&mut output, &entry) write_to_output(&mut output, &entry)?;
.context("while serializing and writing to stdout")?;
} }
output.flush().context("flushing stdout")?; output.flush().context("flushing stdout")?;
Ok(()) Ok(())
} }
fn write_to_output(writer: &mut BufWriter<Stdout>, entry: &Entry) -> anyhow::Result<()> {
serde_json::to_writer(writer, &entry).context("while serializing and writing to stdout")
}
/// Keeps only the last `size` element in memory. /// Keeps only the last `size` element in memory.
/// It's basically a sliding window. /// It's basically a sliding window.
pub struct EntryContext { pub struct EntryContext {