1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19

// Copyright 2020-2021 IOTA Stiftung
// SPDX-License-Identifier: Apache-2.0

use rand::{distributions::Alphanumeric, thread_rng, Rng};

/// Generates a random string with a given charset.
pub fn rand_string_charset(charset: &str, len: usize) -> String {
    let charset = charset.as_bytes();
    let mut rng = rand::thread_rng();

    (0..len)
        .map(|_| charset[rng.gen_range(0..charset.len())] as char)
        .collect()
}

/// Generates a random string.
pub fn rand_string(len: usize) -> String {
    String::from_utf8(thread_rng().sample_iter(&Alphanumeric).take(len).collect()).unwrap()
}