Rust中用enum实现多参数Hook机制完整代码

use std::sync::{Arc, Mutex};
// 定义一个多参数枚举
enum HookInput {
    Add(i32, i32),
    Multiply(i32, i32),
    Concat(String, String),
}
// 定义一个多返回值枚举
enum HookOutput {
    Sum(i32),
    Product(i32),
    Combined(String),
    None,
}
// 定义 Hook 类型
type HookFn = Arc<dyn Fn(HookInput) -> HookOutput + Send + Sync>;
// Hook 管理器
struct HookManager {
    hooks: Mutex<Vec<HookFn>>,
}
impl HookManager {
    fn new() -> Self {
        Self {
            hooks: Mutex::new(Vec::new()),
        }
    }
    // 添加 Hook 函数
    fn add_hook<F>(&self, hook: F)
    where
        F: Fn(HookInput) -> HookOutput + Send + Sync + 'static,
    {
        self.hooks.lock().unwrap().push(Arc::new(hook));
    }
    // 执行所有 Hook
    fn execute_hooks(&self, input: HookInput) -> Vec<HookOutput> {
        let hooks = self.hooks.lock().unwrap();
        hooks.iter().map(|hook| hook(input.clone())).collect()
    }
}
// 实现 Clone 对 HookInput 的支持
impl Clone for HookInput {
    fn clone(&self) -> Self {
        match self {
            HookInput::Add(a, b) => HookInput::Add(*a, *b),
            HookInput::Multiply(a, b) => HookInput::Multiply(*a, *b),
            HookInput::Concat(a, b) => HookInput::Concat(a.clone(), b.clone()),
        }
    }
}

fn main() {
    let manager = HookManager::new();
    // 添加第一个 Hook：处理 Add
    manager.add_hook(|input| match input {
        HookInput::Add(a, b) => {
            println!("Adding: {} + {}", a, b);
            HookOutput::Sum(a + b)
        }
        _ => HookOutput::None,
    });
    // 添加第二个 Hook：处理 Multiply
    manager.add_hook(|input| match input {
        HookInput::Multiply(a, b) => {
            println!("Multiplying: {} * {}", a, b);
            HookOutput::Product(a * b)
        }
        _ => HookOutput::None,
    });
    // 添加第三个 Hook：处理 Concat
    manager.add_hook(|input| match input {
        HookInput::Concat(a, b) => {
            println!("Concatenating: {} + {}", a, b);
            HookOutput::Combined(format!("{}{}", a, b))
        }
        _ => HookOutput::None,
    });
    // 执行 Hook：Add
    let add_results = manager.execute_hooks(HookInput::Add(3, 4));
    println!("Results for Add: {:?}", add_results);
    // 执行 Hook：Multiply
    let multiply_results = manager.execute_hooks(HookInput::Multiply(5, 6));
    println!("Results for Multiply: {:?}", multiply_results);
    // 执行 Hook：Concat
    let concat_results = manager.execute_hooks(HookInput::Concat(
        "Hello".to_string(),
        "World".to_string(),
    ));
    println!("Results for Concat: {:?}", concat_results);
}

use rayon::prelude::*;
hooks.par_iter().map(|hook| hook(input.clone())).collect()