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

Return an iter from last command (#2143)

Browse Source
This commit is contained in:
Joseph T. Lyons 2020-07-09 09:07:51 -04:00 committed by GitHub
parent 1e62a8fb6e
commit b1b93931cb
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
crates/nu-cli/src/commands/last.rs
View File

@ -2,7 +2,7 @@ use crate::commands::WholeStreamCommand;
use crate::context::CommandRegistry; use crate::context::CommandRegistry;
use crate::prelude::*; use crate::prelude::*;
use nu_errors::ShellError; use nu_errors::ShellError;
use nu_protocol::{ReturnSuccess, Signature, SyntaxShape, UntaggedValue, Value}; use nu_protocol::{Signature, SyntaxShape, UntaggedValue, Value};
use nu_source::Tagged; use nu_source::Tagged;
pub struct Last; pub struct Last;
@ -63,27 +63,21 @@ async fn last(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStr
let (LastArgs { rows }, input) = args.process(&registry).await?; let (LastArgs { rows }, input) = args.process(&registry).await?;
let v: Vec<_> = input.into_vec().await; let v: Vec<_> = input.into_vec().await;
let rows_desired = if let Some(quantity) = rows { let end_rows_desired = if let Some(quantity) = rows {
*quantity *quantity as usize
} else { } else {
1 1
}; };
let count = rows_desired as usize; let beginning_rows_to_skip = if end_rows_desired < v.len() {
v.len() - end_rows_desired
let k = if count < v.len() {
v.len() - count
} else { } else {
return Ok(futures::stream::iter(v).to_output_stream()); 0
}; };
let mut values_vec_deque = VecDeque::new(); let iter = v.into_iter().skip(beginning_rows_to_skip);
for x in v[k..].iter() { Ok(futures::stream::iter(iter).to_output_stream())
values_vec_deque.push_back(ReturnSuccess::value(x.clone()));
}
Ok(futures::stream::iter(values_vec_deque).to_output_stream())
} }
#[cfg(test)] #[cfg(test)]