Firestore onCreate触发器：单函数实现多文档更新的方法

当然不需要拆成四个独立函数啦！你完全可以在单个Firestore onCreate触发器里一次性搞定yearlyStats、quartlyStats、monthlyStats和dailyStats这四个文档的更新操作，而且这样做更简洁高效。

核心思路 #

当sessions集合新增文档时，我们可以：

1. 从新会话文档中提取关键信息（比如会话发生的时间，用来定位各个统计文档的路径）
2. 分别创建四个统计文档的更新任务
3. 用Promise.all并行执行所有更新，确保所有操作完成后再结束函数

完整代码示例 #

const functions = require("firebase-functions");
const admin = require("firebase-admin");
admin.initializeApp();
const db = admin.firestore();

exports.updateStatsOnSessionCreate = functions.firestore
  .document("sessions/{sessionId}")
  .onCreate(async (snap, context) => {
    // 1. 获取新增会话的原始数据
    const sessionData = snap.data();
    
    // 假设会话文档包含timestamp字段（记录会话实际发生时间）
    const sessionTimestamp = sessionData.timestamp;
    if (!sessionTimestamp) {
      functions.logger.error("新增会话缺少timestamp字段，无法更新统计");
      return null;
    }

    // 2. 解析时间，生成各统计文档的唯一标识
    const sessionDate = sessionTimestamp.toDate();
    const year = sessionDate.getFullYear();
    const quarter = Math.floor(sessionDate.getMonth() / 3) + 1; // 转换为1-4季度
    const month = sessionDate.getMonth() + 1; // 转换为1-12月
    const day = sessionDate.getDate();

    // 3. 定义四个统计文档的引用（命名规则可根据你的业务需求调整）
    const yearlyStatsRef = db.collection("stats").doc(`yearly-${year}`);
    const quarterlyStatsRef = db.collection("stats").doc(`quarterly-${year}-Q${quarter}`);
    const monthlyStatsRef = db.collection("stats").doc(`monthly-${year}-${month}`);
    const dailyStatsRef = db.collection("stats").doc(`daily-${year}-${month}-${day}`);

    // 4. 构建每个统计文档的更新操作（这里以递增会话计数为例，可替换为你的业务逻辑）
    const updateYearly = yearlyStatsRef.set(
      { 
        sessionCount: admin.firestore.FieldValue.increment(1),
        lastUpdated: admin.firestore.FieldValue.serverTimestamp()
      },
      { merge: true } // 文档不存在则创建，存在则仅更新指定字段
    );

    const updateQuarterly = quarterlyStatsRef.set(
      { 
        sessionCount: admin.firestore.FieldValue.increment(1),
        lastUpdated: admin.firestore.FieldValue.serverTimestamp()
      },
      { merge: true }
    );

    const updateMonthly = monthlyStatsRef.set(
      { 
        sessionCount: admin.firestore.FieldValue.increment(1),
        lastUpdated: admin.firestore.FieldValue.serverTimestamp()
      },
      { merge: true }
    );

    const updateDaily = dailyStatsRef.set(
      { 
        sessionCount: admin.firestore.FieldValue.increment(1),
        lastUpdated: admin.firestore.FieldValue.serverTimestamp()
      },
      { merge: true }
    );

    // 5. 并行执行所有更新操作，等待全部完成
    try {
      await Promise.all([updateYearly, updateQuarterly, updateMonthly, updateDaily]);
      functions.logger.info("所有统计文档更新成功");
    } catch (error) {
      functions.logger.error("统计文档更新失败:", error);
      throw error; // 抛出错误触发Cloud Functions自动重试（默认策略）
    }

    return null;
  });

关键细节说明 #

• 并行执行：Promise.all会让四个更新操作同时发起，比串行执行节省时间，而且Cloud Functions会等待所有操作完成后才标记函数执行成功。
• 原子更新：使用FieldValue.increment(1)确保计数是原子操作，避免多个并发会话创建时出现计数错误。
• 合并写入：{ merge: true }参数保证如果统计文档不存在会自动创建，存在时不会覆盖已有字段（比如你可能还有其他统计指标，不会被这次更新清空）。
• 错误处理：用try/catch捕获异常并抛出，Cloud Functions会根据默认重试策略自动重试失败的操作，提升可靠性。

可选优化 #

如果你的更新逻辑更复杂（比如需要读取统计文档当前值再计算），可以考虑使用Firestore的批量写入（Batch Write），不过对于简单的递增操作，上面的方式已经足够高效。

内容的提问来源于stack exchange，提问作者Michael Vish