如何原子化将Realtime Database触发器升级至第二代Cloud Functions？

现有实现背景 #

我基于Firebase Realtime Database实现了两个触发器，用于监听用户创建与删除事件以统计用户数量，代码如下：

import * as functions from "firebase-functions"
import * as admin from "firebase-admin"

admin.initializeApp()
const db = admin.database()

const userRef = functions.runWith({failurePolicy: true}).database.ref("users/{userId}")
export const onUserCreate = userRef.onCreate(() =>
  db.ref("nUsers").set(admin.database.ServerValue.increment(1)))
export const onUserDelete = userRef.onDelete(() =>
  db.ref("nUsers").set(admin.database.ServerValue.increment(-1)))

由于函数启用了failurePolicy且保证恰好触发一次，即便事件顺序错乱，最终计数也能保持一致。

迁移困境 #

现在要升级到第二代Cloud Functions，但面临两难：

• 先部署新函数：可能导致同一用户事件被新旧函数同时处理，造成重复计数
• 先删除旧函数：会出现短暂的空白期，期间用户增删事件无法被计数

我想到的两种解决方案：

1. 让计数操作具备幂等性，但受限于Realtime Database的事务支持，想不到合适的实现方式
2. 接受计数错误，迁移完成后运行脚本修正，但用户频繁增删时修正难度大

是否有更优的解决方案？

更优解决方案 #

1. 利用函数版本管理+流量切换（推荐） #

这是风险最低的方案，无需修改业务逻辑：

• 第一步：部署第二代版本的函数，保持函数名与v1不同（比如onUserCreateV2、onUserDeleteV2），此时v1函数仍在处理所有触发器事件
• 第二步：通过Cloud Console或gcloud命令，将users/{userId}路径的触发器流量100%切换到v2版本的函数
• 第三步：验证v2函数运行正常后，删除旧的v1函数

整个过程中不会出现计数空白期，也不会同时运行两个版本导致重复统计，因为流量切换是原子性的（Cloud Functions会确保流量切换完成后才停止旧函数的事件处理）。

2. 临时修改计数逻辑为幂等性实现 #

如果无法使用流量切换，可以临时修改v2函数的计数逻辑，避免重复处理同一事件：
核心思路是记录已处理的用户事件，确保每个用户的增删操作只会被计数一次。示例代码：

import * as functions from "firebase-functions/v2"
import * as admin from "firebase-admin"

admin.initializeApp()
const db = admin.database()

export const onUserCreateV2 = functions.database.ref("users/{userId}").onCreate(async (event) => {
  const userId = event.params.userId;
  const processedRef = db.ref(`processedUserEvents/${userId}/create`);

  // 用事务确保仅处理一次
  return processedRef.transaction(async (processed) => {
    if (processed === null) {
      // 标记为已处理
      await db.ref("nUsers").set(admin.database.ServerValue.increment(1));
      return true;
    }
    // 已处理过，不修改计数
    return processed;
  });
});

export const onUserDeleteV2 = functions.database.ref("users/{userId}").onDelete(async (event) => {
  const userId = event.params.userId;
  const processedRef = db.ref(`processedUserEvents/${userId}/delete`);

  return processedRef.transaction(async (processed) => {
    if (processed === null) {
      await db.ref("nUsers").set(admin.database.ServerValue.increment(-1));
      return true;
    }
    return processed;
  });
});

迁移完成并确认无问题后，可以删除processedUserEvents节点，改回原有的简单计数逻辑。

3. 原子化部署配置（通过firebase.json） #

如果你使用Firebase CLI部署，可以通过firebase.json配置同时部署新函数并删除旧函数：

1. 在firebase.json中定义v2函数的配置，移除v1函数的配置
2. 运行firebase deploy --only functions命令，CLI会自动部署新函数并删除未在配置中的旧函数
   这个操作的窗口时间极短，能最大程度减少计数遗漏或重复的概率，但需要提前测试配置的正确性。

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