extern/nushell
1
0
Fork 1
mirror of https://github.com/nushell/nushell.git synced 2025-01-02 20:39:13 +01:00
Code Issues Packages Projects Releases Wiki Activity

make du streaming (#14665)

Browse Source 
# Description
Following up for issue comment:
https://github.com/nushell/nushell/pull/14407#issuecomment-2532343036

> it looks like it just hangs when it's actually counting things

I noticed that `du` command collects output internally, so it doesn't
streaming.

This pr is trying to make it streaming

# User-Facing Changes
NaN

# Tests + Formatting
NaN
This commit is contained in:
Wind 2024-12-25 21:40:02 +08:00 committed by GitHub
parent 1b01598840
commit 38ffcaad7b
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

29
crates/nu-command/src/filesystem/du.rs
View File

@ -165,6 +165,7 @@ fn du_for_one_pattern(
span: Span,
signals: &Signals,
) -> Result<impl Iterator<Item = Value> + Send, ShellError> {
let signals_clone = signals.clone();
let exclude = args.exclude.map_or(Ok(None), move |x| {
Pattern::new(x.item.as_ref())
.map(Some)
@ -174,7 +175,7 @@ fn du_for_one_pattern(
})
})?;
let mut paths = match args.path {
let paths = match args.path {
Some(p) => nu_engine::glob_from(&p, current_dir, span, None),
// The * pattern should never fail.
None => nu_engine::glob_from(
@ -202,22 +203,22 @@ fn du_for_one_pattern(
long,
};
let mut output: Vec<Value> = vec![];
for p in paths.by_ref() {
match p {
Ok(a) => {
if a.is_dir() {
output.push(DirInfo::new(a, &params, max_depth, span, signals)?.into());
} else if let Ok(v) = FileInfo::new(a, deref, span, params.long) {
output.push(v.into());
Ok(paths.filter_map(move |p| match p {
Ok(a) => {
if a.is_dir() {
match DirInfo::new(a, &params, max_depth, span, &signals_clone) {
Ok(v) => Some(Value::from(v)),
Err(_) => None,
}
} else {
match FileInfo::new(a, deref, span, params.long) {
Ok(v) => Some(Value::from(v)),
Err(_) => None,
}
}
Err(e) => {
output.push(Value::error(e, span));
}
}
}
Ok(output.into_iter())
Err(e) => Some(Value::error(e, span)),
}))
}
#[cfg(test)]