【Firestore】セキュリティールールで楽観的ロックを実装する

Posted at 2020-12-23

※本記事内の「Firestore」は、全てFirestore in native modeを指します

楽観的ロックとは

楽観的並行性制御（らっかんてきへいこうせいせいぎょ、optimistic concurrency control）とは、並行性制御(ロック)の手段の種別の一種である。楽観的ロックの概念である。他の処理と競合してはならないトランザクションにおいて、開始時には特に排他処理など行なわず、完了する際に他からの更新がされたか否かを確認し、もし他から更新されてしまっていたら自らの更新処理を破棄し、エラーとする。対照的に悲観的並行性制御がある。

例えばあるドキュメントの詳細情報を表示するページがあって、一部フィールドの値を更新して保存できるボタンがあるとします。

特に何も排他制御を行わない場合、複数のクライアントが同時に詳細情報を表示した後それぞれが更新・保存すると、後から保存した値で上書きされてしまいます。

ユーザーが元の詳細情報を参考にして更新する値を決定したい場合、保存前に他クライアントの更新があると値に不整合が生じます。
これを防ぐ為には詳細情報表示中に他クライアントによる更新が行われていたらエラーにする必要があります。
楽観的ロックという仕組みを導入することでこういった問題を解決することができます。

本記事ではFirestoreのセキュリティルールを利用して楽観的ロックを実装します。

準備

予めテスト用ドキュメントを保存しておきます。

versionというフィールドが楽観的ロックに使用されます。
更新時にversionをインクリメントし、セキュリティルールで更新前にその値をチェックします。
他ユーザーが先にインクリメントしているとエラーになります。

クライアントコード

（FirebaseやFirestoreの初期化等は割愛します）

var book;

function getBook() {

    let db = firebase.firestore();

    db.collection("books").doc("test").get().then(function(doc) {
        if (doc.exists) {
            book = doc.data();
            document.getElementById('book-price').value = book.price;
            document.getElementById('book-title').value = book.title;
        } else {
            console.log("No such document!");
        }
    }).catch(function(error) {
        window.alert(`Error submitting record: ${error}`);
    });
}

function updateBook() {

    let db = firebase.firestore();
    
    if (!book) {
        window.alert("get book first!");
        return;
    }

    const title = document.getElementById('book-title').value;

    const price = parseInt(document.getElementById('book-price').value);
    if (isNaN(price)) {
        window.alert('illegal price:' + v);
        return;
    }

    let b = {...book}; // 更新前にコピー（コピーしないとversionが不正に上がってしまう為）
    b.title = title;
    b.price = price;
    b.version++; // 保存前にバージョンをインクリメントする
    
    db.collection("books").doc("test").set(b)
        .then(function() {
            window.alert('saved');
        })
        .catch(function(error) {
            window.alert(`Error submitting record: ${error}`);
        });
}

実行

getBookを実行すると下記の様にbookドキュメントのtitle、priceフィールドがテキストフィールドに設定されます。

titleとpriceを任意の値に編集します。

updateBookを実行すると、フィールドに入力したtitle、とpriceの内容でbookドキュメントを更新します。
その際versionをインクリメントしています。

現在はまだ楽観的ロックを実装していない為、複数のウィンドウから同時に操作されてもDocumentは常に上書きされてしまいます。

セキュリティールール

セキュリティルールを設定して楽観的ロックを組み込みます。

rules_version = '2';
service cloud.firestore {
  match /databases/{database}/documents {
    match /{document=**} {
      allow read, create: if request.auth.uid != null
      allow create: if request.resource.data.version == 0
      allow update: if request.resource.data.version == resource.data.version+1
    }
  }
}

allow update: if request.resource.data.version == resource.data.version+1 の１行が楽観的ロックチェックです。
request.resource.data でこれから更新するデータ、resource.data で更新対象のデータを参照出来るので、更新データのversionが既存データのvesion+1の場合にみ更新を許可しています。

また、新規作成はversion=0の時のみ可能としています（これは必須ではないです）。