rust - 如何选择有一些限制的随机枚举值

语境

• 对象在一个方向上移动：上，右，下，左。
• 下一个方向将是随机选择的。
• 下一个可能的方向不能向后。

例如：

如果向右移动，则可以向上，向右或向下移动，但不能向左移动。

枚举

有一个简单的方向指示枚举：

#[derive(Debug, PartialEq)]
enum Direction {
  Up,
  Right,
  Down,
  Left,
}

功能签名

我会说这是完成任务的功能签名：

fn next_direction(current_dir: Direction) -> Direction

当前实施

这是我当前的实现：

use rand::prelude::*;

fn next_direction(current_dir: Direction) -> Direction {
    let mut rng = thread_rng();
    // Possible next direction
    let next_dir_iter = [
        Direction::Up,
        Direction::Down,
        Direction::Left,
        Direction::Right,
    ]
    .iter()
    .filter(|&dir| match (current_dir, dir) {
        (Direction::Up, Direction::Down) => false,
        (Direction::Down, Direction::Up) => false,
        (Direction::Left, Direction::Right) => false,
        (Direction::Right, Direction::Left) => false,
        (_, _) => true,
    });
    // Choose one
    let dir = next_dir_iter.choose(&mut rng).unwrap();
    // Return Direction instead of &Direction
    match dir {
        Direction::Up => Direction::Up,
        Direction::Down => Direction::Down,
        Direction::Left => Direction::Left,
        Direction::Right => Direction::Right,
    }
}

题

可以用更清晰，更简单，更有效的方式编写此函数吗？

我会说可读性是一个加号，所以一个划线员或代码高尔夫实施不可能是最佳的。

我已经发现了一个相关的问题：如何从枚举中选择随机值？。

谢谢 =)