如何在 Firestore 中执行批量更新

firebase java google-cloud-firestore firebase-realtime-database

2022-09-03 00:50:08

我正在使用Cloud Firestore并拥有一系列文档。对于集合中的每个文档，我想更新其中一个字段。

使用事务执行更新效率低下，因为我不需要在更新时读取任何数据。

批量更新似乎是正确的方向，但是文档不包括一次更新多个文档的示例。请参阅此处：批处理写入

答案 1

如果您使用过 Firebase 数据库，则无法以原子方式写入完全独立的位置，因此您必须使用批量写入，这意味着要么所有操作都成功，要么不应用任何操作。

关于Firestore，现在所有操作都以原子方式处理。但是，您可以将多个写入操作作为包含 set（）、update（）或 delete（）操作的任意组合的单个批处理来执行。一批写入以原子方式完成，并且可以写入多个文档。

这是一个关于写入、更新和删除操作的批处理操作的简单示例。

WriteBatch batch = db.batch();

DocumentReference johnRef = db.collection("users").document("John");
batch.set(johnRef, new User());

DocumentReference maryRef = db.collection("users").document("Mary");
batch.update(maryRef, "Anna", 20); //Update name and age

DocumentReference alexRef = db.collection("users").document("Alex");
batch.delete(alexRef);

batch.commit().addOnCompleteListener(new OnCompleteListener<Void>() {
    @Override
    public void onComplete(@NonNull Task<Void> task) {
        // ...
    }
});

在批处理对象上调用方法意味着提交整个批处理。commit()

答案 2

我正在寻找一个解决方案，没有找到，所以如果有人感兴趣，我做了这个。

public boolean bulkUpdate() {
  try {
    // see https://firebase.google.com/docs/firestore/quotas#writes_and_transactions
    int writeBatchLimit = 500;
    int totalUpdates = 0;

    while (totalUpdates % writeBatchLimit == 0) {
      WriteBatch writeBatch = this.firestoreDB.batch();

      List<QueryDocumentSnapshot> documentsInBatch =
          this.firestoreDB.collection("animals")
              .whereEqualTo("species", "cat")
              .limit(writeBatchLimit)
              .get()
              .get()
              .getDocuments();

      if (documentsInBatch.isEmpty()) {
        break;
      }

      documentsInBatch.forEach(
          document -> writeBatch.update(document.getReference(), "hasTail", true));

      writeBatch.commit().get();

      totalUpdates += documentsInBatch.size();
    }

    System.out.println("Number of updates: " + totalUpdates);

  } catch (Exception e) {
    return false;
  }
  return true;
}