假设我有一个包含的结构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();
}
确实，get_or_insert_with()但是返回Result<&mut T, E>是您可以用来简化代码的东西。但是，您已经发现Option没有egtry_get_or_insert_with()方法。
其他解决方法也将同样冗长。但是，可以使用match和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()trait方法，并将其实现为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(())
}
本文收集自互联网，转载请注明来源。
如有侵权，请联系 [email protected] 删除。
我来说两句