获取自动生成的文档 ID 以创建文档或更新文档 Firestore / Firebase 中的数组的更好方法

Question

我正在为每个用户创建一个文档来存储用户书签的 ID。 我正在使用 addDoc，因此创建的文档的 ID 由 firebase 生成。我想检查该文档是否存在以创建新文档或更新旧文档。 我需要一个文档参考来更新文档，我花了很长时间试图找出一种方法来获取自动生成的 ID。 我的代码有效，但感觉真的很糟糕 - 有没有更好的方法来完成这个？

//Add a value to the bookmarks array for individual user
export const addBookmarkForUser = async (userAuth, showId) => {
  const bookmarkDocRef = collection(db, 'users', userAuth.uid, 'bookmarks')
  const bookmarkSnapshot = await getDocs(bookmarkDocRef)
  let userBookmarkDocId
  if(bookmarkSnapshot){   
    bookmarkSnapshot.forEach((doc) => {
      if(doc.id){
        userBookmarkDocId = doc.id
        console.log(userBookmarkDocId)
      }
    })    
  }  
  try {    
    if(!bookmarkSnapshot) {
      await addDoc((collection(db, 'users', userAuth.uid, 'bookmarks'), {
        favorites: [{showId: showId}],
      }))
    }else {
      const userBookmarkRef = doc(db, 'users', userAuth.uid, 'bookmarks', userBookmarkDocId)
      await updateDoc(userBookmarkRef, {
        favorites: arrayUnion({showId: showId})
      })}
    
  } catch (error) {
    console.log('Error creating bookmark', error.message);
  }
};

我知道我只会为每个用户提供一个文档 - 是否有更好的方法来查找 doc.id？

这是 Firestore 结构 -

Answer 1

如果您只想检查集合中有多少文档，那么您可以使用getCountFromServer()来加载计数而不需要实际的文档数据。

import { collection, getCountFromServer } from 'firebase/firestore';

const bookmarkColRef = collection(db, 'users', userAuth.uid, 'bookmarks');
const bookmarksCount = (await getCountFromServer(bookmarksColRef)).data().count

console.log(`${bookmarksCount} documents found in ${bookmarksColRef.path}`)\

if (!bookmarksCount) {
  // no document exists, create new 
}

但是，由于您需要文档 ID 及其数据。 您可以运行一个简单的查询：

import { collection, getDocs } from 'firebase/firestore';

const bookmarkColRef = collection(db, 'users', userAuth.uid, 'bookmarks');
const bookmarksSnap = await getDocs(bookmarksColRef);

if (bookmarksSnap.empty()) {
  // no document
} else {
  const docData = bookmarksSnap.docs[0].data();
  console.log(docData)
}