Rustの設計と実装Tipsを学ぶ

Vaultwarden に学ぶ Rust 非同期処理と共有状態管理 Part 1

解析日: 2026/7/18
対象コミット: 169aa5e
リポジトリ: dani-garcia/vaultwarden
Rust非同期処理TokioArc共有状態WebフレームワークVaultwardenパフォーマンス

Vaultwarden に学ぶ Rust 非同期処理と共有状態管理 Part 1

分析対象コミットSHA: 169aa5efcc8d94684ff3bc813a00e6bcc0cc537a 分析日: 2026-07-18T23:02:18.785Z

1. 概要

Rustで書かれた軽量なBitwarden互換サーバー「Vaultwarden」は、高いパフォーマンスと堅牢なセキュリティを両立しながら、BitwardenクライアントAPIをフルサポートしています。自己ホスト環境での利用を念頭に、リソース効率とデプロイの容易さが追求されており、その設計にはRustの強力な機能がふんだんに活用されています。

本記事では、Vaultwardenがどのようにして高効率な非同期処理と、複数のコンポーネント間で安全な共有状態を実現しているのかに焦点を当てます。非同期Webサーバーやバックグラウンドタスクを持つシステムを構築する上で不可欠な、Rustにおける実践的なパターンを学びましょう。

2. アーキテクチャ

Vaultwardenは主にRocket.rsフレームワークを基盤とした非同期Webサーバーとして動作します。そのアーキテクチャは、Webレイヤー、APIレイヤー、サービスレイヤー、永続化レイヤーといった複数の層に分かれています。このうち、非同期処理と共有状態管理は、Webレイヤーとサービスレイヤーの連携において特に重要な役割を果たします。

具体的には、データベース接続プールやWebSocket接続中のユーザーリストといった、アプリケーション全体で共有されるリソースが、Rocketの管理機能やArc<T>を通じて、非同期ハンドラやバックグラウンドタスクから安全にアクセスできるように設計されています。

Rocketのマネージドステートによる共有リソース管理

graph TD A[Rocket アプリケーション] --> B{".manage() で状態を管理"}; B --> C["DbPool (Arc#lt;R2D2Pool#gt;)"]; B --> D["WS_USERS (Arc#lt;DashMap#gt;)"]; B --> E["WS_ANONYMOUS_SUBSCRIPTIONS (Arc#lt;DashMap#gt;)"]; C -- "HTTPハンドラからアクセス" --> F[APIエンドポイント]; D -- "HTTPハンドラからアクセス" --> F; E -- "HTTPハンドラからアクセス" --> F; C -- "バックグラウンドジョブも利用" --> G[Jobスケジューラ];

3. この記事で学べること

本記事を通じて、Vaultwardenのコードベースから以下の実践的なパターンを学ぶことができます。

  1. Arc<T> を活用したスレッドセーフな共有所有権の管理: 複数の非同期タスクやスレッド間でリソースを安全に共有する方法。
  2. Webフレームワークのマネージドステートによる依存性注入: Rocket.rsが提供する機能を使って、共有リソースをHTTPハンドラに効率的に渡す方法。
  3. tokio::spawn を使ったバックグラウンドタスクの分離: メインのイベントループをブロックせずに、長時間実行される処理や独立した処理を実行する方法。
  4. tokio::select! による堅牢なシャットダウン処理の実装: 複数の非同期イベント（特にシグナル）を待機し、アプリケーションを安全に停止させる方法。

4. 実践的な実装・コード解説

Tip 1: Arc<T> を活用したスレッドセーフな共有所有権の管理

Vaultwardenでは、データベース接続プールやWebSocket接続を管理するDashMapなど、複数の非同期タスクから同時にアクセスされる必要のあるリソースが多数存在します。これらのリソースは、Rustの所有権ルールに従いつつ、安全に共有するために Arc<T> (Atomic Reference Counted) を使って管理されています。

Arc<T> は、参照カウントに基づいて所有権を共有するスマートポインタであり、複数のスレッドからの安全なアクセスを可能にします。Arc::clone() は新しい参照カウント付きポインタを作成し、内部の参照カウントを増やします。これにより、元のリソースが解放されるのは、すべての Arc インスタンスがスコープから外れるか、明示的にドロップされた後になります。

コード例 (src/main.rs から抜粋):

let pool = create_db_pool().await;
// ...
instance
    .manage(pool.clone()) // DbPoolをRocketの管理下に置く
    .manage(Arc::clone(&WS_USERS)) // WS_USERS (DashMap) を管理下に置く
    .manage(Arc::clone(&WS_ANONYMOUS_SUBSCRIPTIONS)); // WS_ANONYMOUS_SUBSCRIPTIONS (DashMap) を管理下に置く
// ...
schedule_jobs(pool.clone()); // データベースプールをジョブスケジューラにも渡す

解説: DbPoolやWS_USERSは、アプリケーションの起動時に一度だけ初期化され、その後の各リクエストハンドラやバックグラウンドジョブから参照されるため、Arc::clone()によって参照を共有しています。これにより、不必要なデータコピーを避けつつ、安全な並行アクセスが保証されます。

Tip 2: Webフレームワークのマネージドステートによる依存性注入

Rocket.rsは、.manage() メソッドを通じてアプリケーション全体で共有される状態を管理する強力な機能を提供します。これにより、HTTPハンドラは、グローバルなミュータブルステートに依存することなく、必要な共有リソースに型安全な形でアクセスできます。

コード例 (src/main.rs から抜粋):

// `pool`、`WS_USERS`、`WS_ANONYMOUS_SUBSCRIPTIONS` は Arc でラップされている
instance
    .manage(pool)
    .manage(Arc::clone(&WS_USERS))
    .manage(Arc::clone(&WS_ANONYMOUS_SUBSCRIPTIONS))
    // ... 他のルートやキャッチャーをマウント
    .launch().await?;

解説: pool（内部的にはArcでラップされたデータベースプール）やArc<DashMap<...>>で管理されるWebSocketユーザー情報などは、rocket::Rocket::manage()に渡されます。これにより、Rocketはこれらのリソースをアプリケーションの状態として保持し、後続のHTTPハンドラが State<T> 型の引数を通じて自動的にこれらを受け取れるようになります。このパターンは、テスト容易性を高め、ハンドラコードの記述を簡潔にします。

Tip 3: tokio::spawn を使ったバックグラウンドタスクの分離

Vaultwardenは、メインのWebサーバーの応答性を保ちながら、定期的なデータパージやシグナルハンドリングといった独立した非同期タスクを実行するために tokio::spawn を活用しています。これにより、時間がかかる処理やブロッキングしうる処理がメインのイベントループを妨げることがありません。

コード例 (src/main.rs から抜粋):

// メイン関数の例 (#[rocket::main] 属性が tokio ランタイムを提供)
#[rocket::main]
async fn main() -> Result<(), Error> {
    // ... データベースプール作成など
    
    // シグナルハンドラをバックグラウンドで起動
    #[cfg(unix)]
    spawn_shutdown_signal_handler();

    // ... Rocketインスタンスの構築と起動
    instance.launch().await?;
    Ok(())
}

// シグナルハンドラ関数の例
#[cfg(unix)]
fn spawn_shutdown_signal_handler() {
    tokio::spawn(async move {
        use tokio::signal::unix::signal;
        // SIGINT, SIGTERMなどを待機
        let mut sigint = signal(SignalKind::interrupt()).expect("Error setting SIGINT handler");
        tokio::select! {
            _ = sigint.recv() => "SIGINT",
            // ... 他のシグナル
        };
        info!("Received {signal_name}, initiating graceful shutdown");
        CONFIG.shutdown(); // グレースフルシャットダウンを開始
    });
}

解説: spawn_shutdown_signal_handler() は tokio::spawn を使って、OSからのシグナル（例: Ctrl+Cによる終了要求）を非同期に監視するタスクをバックグラウンドで起動します。これにより、シグナル待機中にWebサーバーが停止することなく、メイン処理とは独立してシャットダウンの準備を進めることができます。

Tip 4: tokio::select! による堅牢なシャットダウン処理の実装

複数の非同期イベント（例えば、異なる種類のシグナル）を同時に監視し、いずれかが発生したときに適切に反応するために、Vaultwardenは tokio::select! マクロを利用しています。これは、複数のFutureを競合させ、最初に完了したFutureの結果を処理し、他のFutureをキャンセルする強力なツールです。

コード例 (Tip 3の spawn_shutdown_signal_handler 内から抜粋):

    tokio::spawn(async move {
        // ... シグナルハンドラの初期化
        tokio::select! {
            _ = sigint.recv() => "SIGINT", // SIGINTを受け取ったら
            _ = sigterm.recv() => "SIGTERM", // SIGTERMを受け取ったら
            // ... 他のシグナルや非同期イベント
        };
        info!("Received {signal_name}, initiating graceful shutdown");
        CONFIG.shutdown();
    });

解説: 上記のコードでは、tokio::select! を使用して、SIGINT または SIGTERM のいずれかのシグナルが受信されるのを待機しています。どちらかのシグナルが到着すると、その分岐が実行され、シャットダウン処理が開始されます。これにより、アプリケーションは異なるシャットダウン要求に対して一貫したグレースフルな終了動作を実行できます。

非同期タスクフロー

graph TD A["メインスレッド (Rocket.rs)"] --> B["tokio::spawn() でタスクを起動"]; B --> C["シグナルハンドラ (非同期タスク)"]; B --> D["Jobスケジューラ (別スレッド)"]; C -- "SIGINT/SIGTERM受信" --> E[グレースフルシャットダウン処理]; D -- "定期タスク実行" --> F[データパージ/通知送信]; A -- "HTTPリクエストを処理" --> G[APIエンドポイント];

5. 実務に持ち帰れるTips

Vaultwardenの設計から、Rustでの非同期処理と共有状態管理について以下の点を実務に活かせます。

  1. スレッド間で共有されるデータには Arc<T> を使う: データベース接続プールやキャッシュなど、複数のタスクから読み書きされる可能性のあるリソースは Arc<T> でラップすることで、Rustの所有権ルールを守りつつ安全に共有できます。ミュータブルな共有が必要な場合は、Arc<Mutex<T>> や Arc<RwLock<T>>、あるいは DashMap のような並行コレクションを検討しましょう。
  2. Webフレームワークのアプリケーションステートを活用する: RocketやActix-webなどのフレームワークは、アプリケーション全体で共有される依存関係（例: DbPool, Config）を管理する機能を提供します。これを活用することで、グローバル変数への依存を避け、クリーンでテストしやすいコードを維持できます。
  3. 長時間実行されるタスクは tokio::spawn で分離する: HTTPリクエストの処理など、メインのイベントループの応答性を保つ必要がある場合、バックグラウンドでのデータ処理、ファイルI/O、外部サービス呼び出しなど、ブロッキングする可能性のあるタスクは tokio::spawn で別の非同期タスクとして実行しましょう。
  4. tokio::select! で複雑な非同期イベントを調整する: 複数のイベント（例: ユーザーからの入力、ネットワークイベント、システムシグナル）のいずれかを待機し、最初に完了したイベントに反応するロジックには tokio::select! が非常に有効です。これにより、グレースフルシャットダウンやタイムアウト処理などを簡潔に記述できます。

6. トレードオフと注意点

Vaultwardenでは、unsafe_code = "forbid" の設定により、意図的に unsafe コードの使用が禁じられています。これは、パスワードマネージャーというセキュリティが極めて重要なアプリケーションにおいて、メモリ安全性に関する潜在的な脆弱性のリスクを最小限に抑えるための重要なトレードオフです。パフォーマンスは、Arc や DashMap のような安全な並行コレクション、そしてコンパイラ最適化（LTO、MiMallocなどのアロケータ）によって達成されています。

7. まとめ

Vaultwardenは、Rustの非同期機能とスマートポインタを巧みに活用することで、パフォーマンスと安全性を両立したWebサーバーを実現しています。

特に、Arc<T> による共有所有権の管理、Rocketのマネージドステートによる依存性注入、tokio::spawn によるバックグラウンドタスクの分離、そして tokio::select! による堅牢なシャットダウン処理は、Rustで堅牢かつ高性能なシステムを構築する上で非常に実践的なパターンです。これらの知識は、皆さんの今後のRustプロジェクトにおいても大いに役立つでしょう。

Part 2では、Vaultwardenがどのようにしてエラー処理と設定管理を構造化しているかを探っていきます。お楽しみに！