主页 / coding / 问题 / 79527198
Accepted
Cutty Shark
Asked: 2025-03-22 17:29:43 +0800 CST

Rust mongodb 重复键错误处理

  • 772

在我的 Rust 应用程序中，我需要对重复键（代码 11000）错误进行特殊处理。我想出了这个解决方案，但它看起来很复杂。有没有更好的方法？

let res = collection.insert_one(entry).await;
match res {
        Ok(_) => Ok(()),
        Err(e) => {
            // println!("{:?}", e);
            match *e.kind {
                ErrorKind::Write(failure) => {
                    match failure {
                        WriteFailure::WriteError(we) => {
                            if we.code == 11000 { Ok(()) }
                            else { Err("other write failure".into()) }
                        },
                        _ => Err("other write error".into()),
                    }
                },
                _ => Err("other error".into()),
            }
       },
    }
mongodb

1 个回答

  1. Best Answer

    目前尚不清楚您是否需要Err在此处返回所有不同的可能值，但假设在其他情况下仅返回原始错误就可以了，这会变得简单得多。

    首先，将“此错误是否为重复键错误”的测试分解为一个函数。这样您可以在需要时重用此逻辑。

    fn is_duplicate_key_error(error: &mongodb::error::Error) -> bool {
    matches!(
        *error.kind,
        ErrorKind::Write(WriteFailure::WriteError(WriteError { code: 11000, .. }))
    )
}

    现在我们可以简单地使用匹配保护将其应用到您的代码中，仅将这种特定类型的错误转换为成功。

    let res = collection.insert_one(entry).await;
match res {
    Ok(_) => Ok(()),
    Err(e) if is_duplicate_key_error(&e) => Ok(()),
    Err(e) => Err(e),
}

    如果这是你程序中非常常见的模式，你可以将其放在Result这样的扩展方法后面：

    pub trait ResultExt {
    fn ignore_duplicate_key_error(self) -> Self;
}

impl ResultExt for Result<(), mongodb::error::Error> {
    fn ignore_duplicate_key_error(self) -> Self {
        match self {
            Err(e) if is_duplicate_key_error(&e) => Ok(()),
            v => v,
        }
    }
}

    现在你的代码变成：

    collection
    .insert_one(entry)
    .await
    .map(|_| ())
    .ignore_duplicate_key_error()

    这是否更简单/更清晰是主观的，但这是一种有效的方法。

    • 0

相关问题