首页 > 解决方案 > 如何存储实现共同特征的结构(不是实例)?

问题描述

背景

我有一个特点MyTrait

trait MyTrait {
    fn name() -> String;

    fn foo();
}

name定义了它的名字,并且foo是它的关联函数,它为我做了一些工作。

目标

我有很多实现这个特征的结构,我想定义一个函数

fn do_work(name: &String);

找到结构TheStructwhere TheStruct::name()is name,然后调用TheStruct::foo()

问题

为了实现我的目标,我需要一个HashMapor use match,但目标值应该是一个 struct (not instance) implements MyTrait,但这似乎是不可能的。

解决方法

name()将and更改foo()为 struct 方法,即name(&self)and foo(&self),然后使用HashMapor 只是match为了使事情正确。但我认为既然name()并且foo()不需要设计实例,我为什么需要实例化它?

标签: rust

解决方案


如果您要将名称映射到操作,则使用HashMap<String, fn()>最有可能的最有意义。例如:

use std::collections::{HashMap, hash_map::Entry};

trait MyTrait {
    fn name() -> String;

    fn foo();
}

struct Example;
impl MyTrait for Example {
    fn name() -> String { 
        "Example".to_string() 
    }
    fn foo() { 
        println!("Do Example"); 
    }
}

#[derive(Default)]
struct Worker {
    handlers: HashMap<String, fn()>,
}
impl Worker {
    fn new() -> Self {
        Default::default()
    }
    fn register_handler<T: MyTrait>(&mut self) {
        let name = T::name();
        match self.handlers.entry(name) {
            Entry::Occupied(_) => panic!("handler already registered"),
            entry => {entry.or_insert(T::foo);},
        }
    }
    fn do_work(&self, name: &str) {
        let handler = self.handlers.get(name).expect("unknown handler");
        handler();
    }
}

fn main() {
    let mut worker = Worker::new();
    worker.register_handler::<Example>();

    worker.do_work("Example");
}

游乐场链接


推荐阅读