如何使用容易出错的初始化器来延迟初始化/填充“Option”？

Question

假设我有一个包含 的结构Option<Resource>，其中Resource我需要使用某种类型，需要分配外部资源（在实际情况下是 GPU 内存），因此可能会失败。在方法内部，如果尚未完成分配，我想尝试分配，但如果失败则传播或处理错误。

如果没有失败案例Option::get_or_insert_with就完美了。事实上，我想到的最简洁的解决方案涉及一个unwrap()，这是不优雅的，因为它看起来像是潜在的恐慌：

struct Container {
    resource: Option<Resource>,
    ...
}

impl Container {
    ...
    fn activate(&mut self) -> Result<(), Error> {
        if self.resource.is_none() {
            self.resource = Some(Resource::new()?);
        }
        let resource: &mut Resource = self.resource.as_mut().unwrap();
        // ... now do things with `resource` ...
        Ok(())
    }
    ...
}

Option有没有比这更简单的初始化方法？需要明确的是，我不仅仅寻求避免unwrap()，而且还寻求整体可读性。如果替代方案更加复杂和间接，我宁愿坚持这个。

---

完整的示例代码（在 Rust Playground 上）：

#[derive(Debug)]
struct Resource {}
#[derive(Debug)]
struct Error;

impl Resource {
    fn new() -> Result<Self, Error> {
        Ok(Resource {})
    }

    fn write(&mut self) {}
}

#[derive(Debug)]
struct Container {
    resource: Option<Resource>,
}

impl Container {
    fn new() -> Self {
        Self { resource: None }
    }

    fn activate(&mut self) -> Result<(), Error> {
        if self.resource.is_none() {
            self.resource = Some(Resource::new()?);
        }
        self.resource.as_mut().unwrap().write();
        Ok(())
    }
}

fn main() {
    Container::new().activate();
}

Answer 1

确实如此，get_or_insert_with()但是返回Result<&mut T, E>可以用来简化代码。但是，正如您已经发现的那样，Option没有egtry_get_or_insert_with()方法。

其他解决方法也同样冗长。但是，您可以使用 andmatch来get_or_insert()避免unwrap()这样的情况：

fn activate(&mut self) -> Result<(), Error> {
    let res = match &mut self.resource {
        Some(res) => res,
        None => self.resource.get_or_insert(Resource::new()?),
    };
    res.write();

    Ok(())
}

如果经常使用，那么您也可以定义自己的try_get_or_insert_with()特征方法，并将其实现为Option<T>.

trait TryGetOrInsert<T> {
    fn try_get_or_insert_with<E, F>(&mut self, f: F) -> Result<&mut T, E>
    where
        F: FnOnce() -> Result<T, E>;
}

impl<T> TryGetOrInsert<T> for Option<T> {
    fn try_get_or_insert_with<E, F>(&mut self, f: F) -> Result<&mut T, E>
    where
        F: FnOnce() -> Result<T, E>,
    {
        match self {
            Some(value) => Ok(value),
            None => Ok(self.get_or_insert(f()?)),
        }
    }
}

那么现在您可以将您的activate()方法简化为以下内容：

fn activate(&mut self) -> Result<(), Error> {
    let res = self.resource.try_get_or_insert_with(|| Resource::new())?;
    res.write();

    Ok(())
}