API设计：实现支持可选参数及默认行为的Rust函数

你的现有代码存在两处问题：

• main函数中draw_image(Triangle);调用错误，需改为draw_image(ArgType::Triangle);
• draw_image()无参调用无法编译，因为函数定义要求必须传入ArgType类型参数

针对你需要支持传参执行对应逻辑、无参执行默认行为且后续参数会快速增加的需求，推荐以下两种实现方式：

方式一：使用Option<ArgType>接收参数 #

通过Option类型包装参数，Some(xxx)对应指定行为，None对应默认行为，调用时可以灵活传参或传None：

enum ArgType {
    Square,
    Circle,
    Triangle,
}

fn draw_image(arg: Option<ArgType>) {
    match arg {
        Some(ArgType::Square) => println!("Square!"),
        Some(ArgType::Circle) => println!("Circle!"),
        Some(ArgType::Triangle) => println!("Triangle!"),
        None => println!("Empty argument, initiating default behavior!"),
    }
}

fn main() {
    // 传参调用
    draw_image(Some(ArgType::Triangle));
    // 无参触发默认行为（显式传None）
    draw_image(None);
    
    // 可选：封装无参函数简化调用
    // fn draw_default() { draw_image(None); }
    // draw_default();
}

方式二：给枚举实现Default trait，使用参数默认值 #

这种方式更适合后续扩展场景，新增枚举成员时只需维护match分支，默认行为逻辑无需改动：

// 给枚举派生Default trait，并指定默认成员
#[derive(Default)]
enum ArgType {
    #[default]
    DefaultBehavior, // 专门用于默认行为的枚举成员
    Square,
    Circle,
    Triangle,
}

// 给函数参数设置默认值，无参时自动使用Default成员
fn draw_image(arg: ArgType = ArgType::DefaultBehavior) {
    match arg {
        ArgType::Square => println!("Square!"),
        ArgType::Circle => println!("Circle!"),
        ArgType::Triangle => println!("Triangle!"),
        ArgType::DefaultBehavior => println!("Empty argument, initiating default behavior!"),
    }
}

fn main() {
    // 传参调用
    draw_image(ArgType::Triangle);
    // 无参自动触发默认行为
    draw_image();
}

如果不想新增专门的默认成员，也可以将现有某个成员设为默认，比如把Square设为默认：

#[derive(Default)]
enum ArgType {
    #[default]
    Square,
    Circle,
    Triangle,
}

此时无参调用draw_image()会执行Square的逻辑作为默认行为。

内容的提问来源于stack exchange，提问作者Charles Jonasson