forked from extern/nushell
138b5af82b
This commit is more substantial than it looks: there was basically no real support for decimals before, and that impacted values all the way through. I also made Size contain a decimal instead of an integer (`1.6kb` is a reasonable thing to type), which impacted a bunch of code. The biggest impact of this commit is that it creates many more possible ways for valid nu types to fail to serialize as toml, json, etc. which typically can't support the full range of Decimal (or Bigint, which I also think we should support). This commit makes to-toml fallible, and a similar effort is necessary for the rest of the serializations. We also need to figure out how to clearly communicate to users what has happened, but failing to serialize to toml seems clearly superior to me than weird errors in basic math operations.
73 lines
2.2 KiB
Rust
73 lines
2.2 KiB
Rust
use crate::commands::WholeStreamCommand;
|
|
use crate::errors::ShellError;
|
|
use crate::object::TaggedDictBuilder;
|
|
use crate::prelude::*;
|
|
use std::time::Duration;
|
|
use std::usize;
|
|
|
|
use futures::stream::{StreamExt, TryStreamExt};
|
|
use heim::process::{self as process, Process, ProcessResult};
|
|
use heim::units::{ratio, Ratio};
|
|
|
|
pub struct PS;
|
|
|
|
impl WholeStreamCommand for PS {
|
|
fn run(
|
|
&self,
|
|
args: CommandArgs,
|
|
registry: &CommandRegistry,
|
|
) -> Result<OutputStream, ShellError> {
|
|
ps(args, registry)
|
|
}
|
|
|
|
fn name(&self) -> &str {
|
|
"ps"
|
|
}
|
|
|
|
fn signature(&self) -> Signature {
|
|
Signature::build("ps")
|
|
}
|
|
}
|
|
|
|
async fn usage(process: Process) -> ProcessResult<(process::Process, Ratio)> {
|
|
let usage_1 = process.cpu_usage().await?;
|
|
futures_timer::Delay::new(Duration::from_millis(100)).await?;
|
|
let usage_2 = process.cpu_usage().await?;
|
|
|
|
Ok((process, usage_2 - usage_1))
|
|
}
|
|
|
|
fn ps(args: CommandArgs, registry: &CommandRegistry) -> Result<OutputStream, ShellError> {
|
|
let args = args.evaluate_once(registry)?;
|
|
let span = args.name_span();
|
|
|
|
let stream = async_stream_block! {
|
|
let processes = process::processes()
|
|
.map_ok(|process| {
|
|
// Note that there is no `.await` here,
|
|
// as we want to pass the returned future
|
|
// into the `.try_buffer_unordered`.
|
|
usage(process)
|
|
})
|
|
.try_buffer_unordered(usize::MAX);
|
|
pin_utils::pin_mut!(processes);
|
|
|
|
while let Some(res) = processes.next().await {
|
|
if let Ok((process, usage)) = res {
|
|
let mut dict = TaggedDictBuilder::new(Tag::unknown_origin(span));
|
|
dict.insert("pid", Value::int(process.pid()));
|
|
if let Ok(name) = process.name().await {
|
|
dict.insert("name", Value::string(name));
|
|
}
|
|
if let Ok(status) = process.status().await {
|
|
dict.insert("status", Value::string(format!("{:?}", status)));
|
|
}
|
|
dict.insert("cpu", Value::number(usage.get::<ratio::percent>()));
|
|
yield ReturnSuccess::value(dict.into_tagged_value());
|
|
}
|
|
}
|
|
};
|
|
|
|
Ok(stream.to_output_stream())
|
|
}
|