nu-command 0.106.0

Nushell's built-in commands
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50

use nu_engine::command_prelude::*;
use nu_protocol::Signals;
use rand::{
    Rng,
    distr::{Alphanumeric, StandardUniform},
    rng,
};

pub(super) enum RandomDistribution {
    Binary,
    Alphanumeric,
}

pub(super) fn random_byte_stream(
    distribution: RandomDistribution,
    length: usize,
    span: Span,
    signals: Signals,
) -> PipelineData {
    let stream_type = match distribution {
        RandomDistribution::Binary => ByteStreamType::Binary,
        RandomDistribution::Alphanumeric => ByteStreamType::String,
    };

    const OUTPUT_CHUNK_SIZE: usize = 8192;
    let mut remaining_bytes = length;
    PipelineData::ByteStream(
        ByteStream::from_fn(span, signals.clone(), stream_type, move |out| {
            if remaining_bytes == 0 || signals.interrupted() {
                return Ok(false);
            }

            let bytes_to_write = std::cmp::min(remaining_bytes, OUTPUT_CHUNK_SIZE);

            let rng = rng();
            let byte_iter: Box<dyn Iterator<Item = u8>> = match distribution {
                RandomDistribution::Binary => Box::new(rng.sample_iter(StandardUniform)),
                RandomDistribution::Alphanumeric => Box::new(rng.sample_iter(Alphanumeric)),
            };
            out.extend(byte_iter.take(bytes_to_write));

            remaining_bytes -= bytes_to_write;

            Ok(true)
        })
        .with_known_size(Some(length as u64)),
        None,
    )
    .with_span(span)
}