Firebase Cloud Functions (第二代) 如何在超时前获取通知？

针对任务队列中的后台任务超时问题，有几种可行的方式实现超时预警、收尾记录断点，以及后续从断点续跑：

1. 监听平台终止信号 #

Cloud Functions在任务即将超时前，会向进程发送SIGTERM信号（部分场景下也会发送SIGINT），你可以监听这个信号来触发收尾逻辑。平台会给一个短暂的宽限期（通常10秒左右）让你完成轻量化的收尾操作。

示例代码（Node.js环境）：

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

exports.longRunningTask = functions.tasks.taskQueue().onRun(async (task) => {
  let taskProgress = { step: 0, data: task.data };

  // 注册终止信号处理函数
  const handleTermination = async () => {
    // 将断点进度写入Firestore
    await admin.firestore().collection("taskBreakpoints").doc(task.id).set({
      progress: taskProgress,
      timestamp: admin.firestore.FieldValue.serverTimestamp()
    });
    process.exit(0); // 确保进程正常退出
  };

  process.on("SIGTERM", handleTermination);
  process.on("SIGINT", handleTermination);

  // 执行任务核心逻辑
  for (let i = 0; i < 100; i++) {
    taskProgress.step = i;
    await doSomeHeavyWork(taskProgress.data); // 自定义任务步骤
  }

  // 任务完成后清理断点记录
  await admin.firestore().collection("taskBreakpoints").doc(task.id).delete();
  return null;
});

注意：信号处理逻辑必须轻量化，避免执行耗时操作，否则会被平台强制终止。

2. 主动监控剩余运行时间 #

通过环境变量获取函数的超时配置，在任务执行过程中定期检查剩余时间，当剩余时间低于预设阈值时，主动触发收尾逻辑。这种方式更灵活，能自定义预警提前量。

示例代码：

exports.longRunningTask = functions.tasks.taskQueue().onRun(async (task) => {
  const timeoutSec = parseInt(process.env.FUNCTION_TIMEOUT_SEC) || 90; // 默认超时90秒
  const warningThreshold = 15; // 剩余15秒时触发预警
  const startTime = Date.now();
  let taskProgress = { step: 0, data: task.data };

  // 计算剩余运行时间
  const getRemainingTime = () => {
    const elapsedSec = (Date.now() - startTime) / 1000;
    return timeoutSec - elapsedSec;
  };

  // 从断点开始执行（如果存在）
  const breakpointDoc = await admin.firestore().collection("taskBreakpoints").doc(task.id).get();
  if (breakpointDoc.exists) {
    taskProgress = breakpointDoc.data().progress;
  }

  // 执行任务步骤
  for (let i = taskProgress.step; i < 100; i++) {
    // 检查剩余时间
    if (getRemainingTime() < warningThreshold) {
      await admin.firestore().collection("taskBreakpoints").doc(task.id).set({
        progress: { ...taskProgress, step: i },
        timestamp: admin.firestore.FieldValue.serverTimestamp()
      });
      // 抛出重试错误，让任务队列后续重新调度
      throw new functions.tasks.TaskRetryError("Task approaching timeout, breakpoint saved");
    }

    taskProgress.step = i;
    await doSomeHeavyWork(taskProgress.data);
  }

  await admin.firestore().collection("taskBreakpoints").doc(task.id).delete();
  return null;
});

3. 结合任务队列重试实现断点续跑 #

在任务启动时，先检查是否存在未完成的断点记录，如果有，直接从断点进度继续执行，无需从头开始。这种方式能有效避免重复执行已完成的步骤，提升效率。

关键逻辑就是在任务初始化时读取断点，任务完成后删除断点，超时前保存断点，配合任务队列的重试机制（可在任务队列配置中设置重试次数和间隔），就能实现自动续跑。

额外注意事项 #

• 函数超时时间可在Firebase控制台或firebase.json中配置，最高支持9分钟（540秒），如果任务远超这个时长，建议拆分为多个独立的子任务，通过任务队列链式调用。
• 断点记录要保证原子性，避免并发写入冲突，可使用Firestore事务或批量操作。
• 不同运行时（如Python、Go）的信号监听方式不同，比如Python可通过signal模块监听SIGTERM。

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