如何随机选择格式字符串

时间:2016-09-11 14:37:23

标签: random rust string-formatting

有时,程序可以通过多种方式将包含动态值的消息短语发送给用户。例如:

  • “{}分钟。”
  • “你需要在不到{}分钟内完成。”

并非所有消息都包含仅作为前缀或后缀的值。在动态语言中,这似乎是字符串格式化的逻辑任务。

对于不希望重复性的媒体(例如Slack频道),有许多不同的措辞,使用以下内容产生每个最终String输出:

pub fn h(x: usize) -> String {
    rand::sample(rand::thread_rng(), vec![
        format!("{} minutes remain.", x),
        format!("Hurry up; only {} minutes left to finish.", x),
        format!("Haste advisable; time ends in {}.", x),
        /* (insert many more elements here) */
    ], 1).first().unwrap_or(format!("{}", x))
}

将是:

  • 单独输入format!(/*...*/, x)每次都是单调乏味。
  • 浪费记忆+时钟周期,因为在选择一个之前完全生成每一种可能性,丢弃其他可能性。

有没有办法避免这些缺点?

如果不是格式字符串的编译时评估,则返回随机选择的&'static str(从静态切片)传递到format!的函数将是首选解决方案。

3 个答案:

答案 0 :(得分:5)

Rust支持在函数内定义函数。我们可以构建一个函数指针片段,让rand::sample选择其中一个,然后调用所选函数。

extern crate rand;

use rand::Rng;

pub fn h(x: usize) -> String {
    fn f0(x: usize) -> String {
        format!("{} minutes remain.", x)
    }

    fn f1(x: usize) -> String {
        format!("Hurry up; only {} minutes left to finish.", x)
    }

    fn f2(x: usize) -> String {
        format!("Haste advisable; time ends in {}.", x)
    }

    let formats: &[fn(usize) -> String] = &[f0, f1, f2];
    (*rand::thread_rng().choose(formats).unwrap())(x)
}

这解决了原始解决方案的“浪费”方面,但不是“乏味”方面。我们可以通过使用宏来减少重复次数。请注意,函数中定义的宏也是该函数的本地宏!这个宏利用了Rust的卫生宏来定义多个名为f的函数,因此在使用宏时我们不需要为每个函数提供名称。

extern crate rand;

use rand::Rng;

pub fn h(x: usize) -> String {
    macro_rules! messages {
        ($($fmtstr:tt,)*) => {
            &[$({
                fn f(x: usize) -> String {
                    format!($fmtstr, x)
                }
                f
            }),*]
        }
    }

    let formats: &[fn(usize) -> String] = messages!(
        "{} minutes remain.",
        "Hurry up; only {} minutes left to finish.",
        "Haste advisable; time ends in {}.",
    );
    (*rand::thread_rng().choose(formats).unwrap())(x)
}

答案 1 :(得分:4)

我的建议是使用匹配来避免不必要的计算并使代码保持尽可能紧凑:

use rand::{thread_rng, Rng};

let mut rng = thread_rng();
let code: u8 = rng.gen_range(0, 5);
let time = 5;
let response = match code {
    0 => format!("Running out of time! {} seconds left", time),
    1 => format!("Quick! {} seconds left", time),
    2 => format!("Hurry, there are {} seconds left", time),
    3 => format!("Faster! {} seconds left", time),
    4 => format!("Only {} seconds left", time),
    _ => unreachable!()
};

Playground link

不可否认,从字面上匹配数字有点难看,但它可能是你能得到它的最短时间。

答案 2 :(得分:2)

通过使用闭包(或函数指针)来创建多个字符串是很简单的:

extern crate rand;

use rand::Rng;

pub fn h(x: usize) -> String {
    let messages: &[&Fn() -> String] = &[
        &|| format!("{} minutes remain.", x),
        &|| format!("Hurry up; only {} minutes left to finish.", x),
        &|| format!("Haste advisable; time ends in {}.", x),
    ];
    let default_message = || format!("{}", x);

    rand::thread_rng().choose(messages).unwrap_or(&&(&default_message as &Fn()->String))()
}

fn main() {
    println!("{}", h(1));
}

注意:

  1. choose代替sample获取一个值。
  2. 无需Vec;数组应该没问题。
  3. 这不太可能改善"美丽"方面。宏可以消除苦差事:

    extern crate rand;
    
    macro_rules! messages {
        {$default: expr, $($msg: expr,)*} => {
            use rand::Rng;
    
            let messages: &[&Fn() -> String] = &[
                $(&|| $msg),*
            ];
            let default_message = || $default;
    
            rand::thread_rng().choose(messages).unwrap_or(&&(&default_message as &Fn() -> String))()
        }
    }
    
    pub fn h(x: usize) -> String {
        messages! {
            format!("{}", x),
            format!("{} minutes remain.", x),
            format!("Hurry up; only {} minutes left to finish.", x),
            format!("Haste advisable; time ends in {}.", x),
        }
    }
    
    fn main() {
        println!("{}", h(1));
    }
    

    注意:

    1. 宏需要至少一个参数;这将用作默认消息。