面试题：Rust OnceCell初始化失败时的自定义错误处理与资源管理

1. 自定义错误类型

首先，定义一个自定义错误类型，用于表示初始化过程中可能出现的各种错误。

use std::fmt;
use std::io;
use std::sync::Mutex;

// 定义一个自定义错误类型，包含文件句柄错误和数据库连接错误
#[derive(Debug)]
enum MyAppError {
    FileError(io::Error),
    DatabaseError(String),
}

impl fmt::Display for MyAppError {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            MyAppError::FileError(e) => write!(f, "File error: {}", e),
            MyAppError::DatabaseError(e) => write!(f, "Database error: {}", e),
        }
    }
}

impl std::error::Error for MyAppError {}

2. 包含外部资源的对象

定义一个包含多个外部资源的结构体。

struct MyResources {
    file_handle: Option<std::fs::File>,
    db_connection: Option<Mutex<()>>, // 这里用Mutex<()>简单示意数据库连接
}

impl Drop for MyResources {
    fn drop(&mut self) {
        // 释放文件句柄
        if let Some(file) = self.file_handle.take() {
            let _ = file.sync_all();
        }
        // 释放数据库连接（这里只是示意，实际可能需要关闭连接等操作）
        if let Some(_) = self.db_connection.take() {
            // 实际关闭数据库连接的逻辑
        }
    }
}

3. 使用OnceCell初始化对象

使用OnceCell来初始化MyResources对象，并处理错误。

use std::sync::OnceCell;

static RESOURCES: OnceCell<Result<MyResources, MyAppError>> = OnceCell::new();

fn initialize_resources() -> Result<MyResources, MyAppError> {
    let file = std::fs::File::open("example.txt").map_err(MyAppError::FileError)?;
    let db = Some(Mutex::new(())); // 这里只是简单示意数据库连接的获取
    Ok(MyResources {
        file_handle: Some(file),
        db_connection: db,
    })
}

fn get_resources() -> Result<&'static MyResources, MyAppError> {
    RESOURCES.get_or_try_init(initialize_resources).map_err(|e| e.into())
}

4. 错误传播

在initialize_resources函数中，通过?操作符传播错误。如果文件打开失败，函数会立即返回错误，并且MyResources结构体的drop方法会被调用，释放已分配的资源。

5. 资源释放顺序

在MyResources的drop方法中，首先释放文件句柄，然后释放数据库连接。这样确保了资源按照合理的顺序释放，避免资源泄漏。

6. 并发控制

OnceCell本身是线程安全的，适用于并发环境。在获取资源时，多个线程可以并发调用get_resources函数，OnceCell会保证资源只被初始化一次。对于数据库连接，使用Mutex进行线程安全的访问控制。

通过上述设计，实现了安全、高效的资源管理，结合了OnceCell的特性以及自定义错误处理机制，确保在复杂的Rust项目中资源的正确初始化、错误处理和释放。

星途面试题库