Merge pull request #23 from rabee-inc/feature/batch-getter-cache

同じものを何度も取得しないように修正
rabee-inc · Jun 2, 2023 · 8a9a49b · 8a9a49b
2 parents d828689 + 8b1fdca
commit 8a9a49b
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/cloudfirestore/batch_getter_impl.go b/cloudfirestore/batch_getter_impl.go
@@ -36,6 +36,10 @@ func (bg *batchGetter) Add(docRef *firestore.DocumentRef, dst any) {
 	if docRef == nil || docRef.ID == "" || !ValidateDocumentID(docRef.ID) {
 		return
 	}
+	// 既に登録済みの場合は何もしない
+	if _, ok := bg.docMap[docRef.Path]; ok {
+		return
+	}
 	bg.docMap[docRef.Path] = &batchGetterItem{
 		docRef: docRef,
 		data:   dst,