如何强制FiRecovery客户端应用程序为集合维护正确的文档计数？

Question

Firestore无法查询集合的大小，因此如果客户端应用程序需要知道这个值，它需要对另一个集合中的文档进行一些维护来保存这个计数。但是，它要求客户端正确执行事务，以便在添加和删除文档时保持该计数最新。恶意的或损坏的客户端可以独立地修改集合或计数，并造成计数不准确的情况。

可以通过后端强制客户端执行此操作，或者使用云函数触发器自动维护计数（延迟一段时间后生效）。然而，我不想引入后端，我宁愿使用安全规则。我该怎么做？

Answer 1

建议的解决方案仍然会失败，因为您只需向批处理写入添加其他文档即可。

但是，您可以将以下内容添加到邮件收集规则中:

<代码>

rules_version = '2';
service cloud.firestore {
  match /databases/{database}/documents {

    match /messages/{id} {
      allow read;
      allow create: if
        getAfter(/databases/$(database)/documents/messages-stats/data).data.count ==
             get(/databases/$(database)/documents/messages-stats/data).data.count + 1 
             && get(/databases/$(database)/documents/messages-stats/data).data.messageId == request.resource.id;
      allow delete: if
        getAfter(/databases/$(database)/documents/messages-stats/data).data.count ==
             get(/databases/$(database)/documents/messages-stats/data).data.count - 1
             && get(/databases/$(database)/documents/messages-stats/data).data.messageId == request.resource.id;
    }

    match /messages-stats/data {
      allow read;
      allow update: if (
        request.resource.data.count == resource.data.count + 1 &&
        existsAfter(/databases/$(database)/documents/messages/$(request.resource.data.messageId)) &&
           ! exists(/databases/$(database)/documents/messages/$(request.resource.data.messageId))
      ) || (
        request.resource.data.count == resource.data.count - 1 &&
        ! existsAfter(/databases/$(database)/documents/messages/$(request.resource.data.messageId)) &&
               exists(/databases/$(database)/documents/messages/$(request.resource.data.messageId))
      );
    }

  }
}

Answer 2

假设您有一个“消息”集合，其中包含客户端可以添加和删除的消息。还可以想象另一个集合中的文档，路径为“messages stats/data”，其中有一个名为“count”的字段，用于维护消息中文档的准确计数。如果客户端应用程序执行如下事务以添加文档：

async function addDocumentTransaction() {
    try {
        const ref = firestore.collection("messages").doc()
        const statsRef = firestore.collection("messages-stats").doc("data")
        await firestore.runTransaction(transaction => {
            transaction.set(ref, {
                foo: "bar"
            })
            transaction.update(statsRef, {
                count: firebase.firestore.FieldValue.increment(1),
                messageId: ref.id
            })
            return Promise.resolve()
        })
        console.log(`Added message ${ref.id}`)
    }
    catch (error) {
        console.error(error)
    }
}

或者像这样的一批:

async function addDocumentBatch() {
    try {
        const batch = firestore.batch()
        const ref = firestore.collection("messages").doc()
        const statsRef = firestore.collection("messages-stats").doc("data")
        batch.set(ref, {
            foo: "bar"
        })
        batch.update(statsRef, {
            count: firebase.firestore.FieldValue.increment(1),
            messageId: ref.id
        })
        await batch.commit()
        console.log(`Added message ${ref.id}`)
    }
    catch (error) {
        console.error(error)
    }
}

像这样使用事务删除文档：

async function deleteDocumentTransaction(id) {
    try {
        const ref = firestore.collection("messages").doc(id)
        const statsRef = firestore.collection("messages-stats").doc("data")
        await firestore.runTransaction(transaction => {
            transaction.delete(ref)
            transaction.update(statsRef, {
                count: firebase.firestore.FieldValue.increment(-1),
                messageId: ref.id
            })
            return Promise.resolve()
        })
        console.log(`Deleted message ${ref.id}`)
    }
    catch (error) {
        console.error(error)
    }
}

或类似于批次：

async function deleteDocumentBatch(id) {
    try {
        const batch = firestore.batch()
        const ref = firestore.collection("messages").doc(id)
        const statsRef = firestore.collection("messages-stats").doc("data")
        batch.delete(ref)
        batch.update(statsRef, {
            count: firebase.firestore.FieldValue.increment(-1),
            messageId: ref.id
        })
        await batch.commit()
        console.log(`Deleted message ${ref.id}`)
    }
    catch (error) {
        console.error(error)
    }
}

然后，您可以使用安全规则要求要添加或删除的文档只能与具有 count 字段的文档同时更改。微：

rules_version = '2';
service cloud.firestore {
  match /databases/{database}/documents {

    match /messages/{id} {
      allow read;
      allow create: if
        getAfter(/databases/$(database)/documents/messages-stats/data).data.count ==
             get(/databases/$(database)/documents/messages-stats/data).data.count + 1;
      allow delete: if
        getAfter(/databases/$(database)/documents/messages-stats/data).data.count ==
             get(/databases/$(database)/documents/messages-stats/data).data.count - 1;
    }

    match /messages-stats/data {
      allow read;
      allow update: if (
        request.resource.data.count == resource.data.count + 1 &&
        existsAfter(/databases/$(database)/documents/messages/$(request.resource.data.messageId)) &&
           ! exists(/databases/$(database)/documents/messages/$(request.resource.data.messageId))
      ) || (
        request.resource.data.count == resource.data.count - 1 &&
        ! existsAfter(/databases/$(database)/documents/messages/$(request.resource.data.messageId)) &&
               exists(/databases/$(database)/documents/messages/$(request.resource.data.messageId))
      );
    }

  }
}

请注意，客户必须：

< li >增加或减少< code >/messages-stats/data 中的计数。 < li >必须在名为< code>messageId的字段中提供要在“数据”文档中添加或删除的文档的id。 < li >递增计数要求< code>messageId中标识的新文档在批处理/事务提交之前不存在，在事务之后存在。 < li >减少计数要求< code>messageId中标识的旧文档必须在批处理/事务提交之前存在，并且在事务之后不能存在。

请注意，existsAfter（）在事务完成后检查命名文档的状态，而exists（）在事务结束前检查该状态。这两个函数之间的差异对于这些规则的工作方式很重要。

另请注意，这在重载下不会很好地扩展。如果文档的添加和删除速度超过每秒10，则数据文档的每个文档写入速率将被超过，并且事务将失败。

一旦你有了这个，现在你实际上可以编写安全规则来限制集合的大小，如下所示：

match /messages/{id} {
  allow create: if
    get(/databases/$(database)/documents/messages-stats/data).data.count < 5;
}