在NodeJS中使用firebase-admin创建自定义Token时报栈溢出错误如何解决？

• 可调用云函数缺少返回值：Firebase的https.onCall类型云函数要求必须返回Promise或具体值来标记异步任务的完成状态，当前代码未返回admin.auth().createCustomToken生成的Promise，云函数运行时无法正确判断任务是否结束，内部调度逻辑出现递归回调，最终触发调用栈溢出。
• Admin SDK初始化配置异常：在Firebase云函数原生运行环境中，SDK会自动读取内置的服务账号配置完成初始化，手动传入仅包含serviceAccountId的不完整配置，会导致SDK内部反复尝试补全缺失的认证参数，触发循环调用导致栈溢出。
• 冗余的TypeScript类型注解：代码中使用了data: any这类TS类型注解，如果项目未配置TS编译直接将代码作为JS运行，会触发语法解析异常，部分版本的Firebase函数runtime的异常捕获逻辑存在递归触发问题，也会导致栈溢出。

1. 修正云函数返回逻辑：给异步操作加上return，同时将结果或异常按云函数规范返回给客户端，避免云函数上下文异常挂起。
2. 简化Admin SDK初始化：直接调用无参数的admin.initializeApp()即可，不需要手动传入服务账号配置。
3. 移除不必要的TS类型注解，或配置项目的TS编译流程后再部署。
4. 新增角色校验逻辑，符合仅允许设置医生/普通用户的需求。

修复后完整代码 #

const functions = require("firebase-functions");
const admin = require("firebase-admin");
// 用云函数默认配置初始化Admin SDK
admin.initializeApp();

exports.addMessage = functions.https.onCall((data, context) => {
  // 校验请求用户的登录态，可选，根据业务需求调整
  if (!context.auth) {
    throw new functions.https.HttpsError('unauthenticated', '请先登录');
  }

  const allowedRoles = ['doctor', 'user'];
  // 从请求参数中取要设置的角色，也可根据业务逻辑调整来源
  const targetRole = data.role;
  if (!allowedRoles.includes(targetRole)) {
    throw new functions.https.HttpsError('invalid-argument', '仅支持设置医生或普通用户角色');
  }

  // 实际使用时建议从context.auth.uid获取当前用户id，或从传入的data中取目标用户id
  const userId = "some-uid";
  const additionalClaims = {
    role: targetRole,
    name: "Juan",
  };

  // 返回Promise让云函数正确调度异步任务
  return admin
    .auth()
    .createCustomToken(userId, additionalClaims)
    .then((customToken) => {
      console.log('生成令牌成功', customToken);
      // 返回结果给调用端
      return { code: 0, token: customToken };
    })
    .catch((error) => {
      console.log("生成自定义令牌失败:", error);
      // 抛出符合云函数规范的异常，前端可以统一捕获
      throw new functions.https.HttpsError('internal', '令牌生成失败', error.message);
    });
});

额外优化建议 #

如果你的需求仅为给已有的Firebase用户打角色标签，不需要集成第三方登录体系，不需要手动生成自定义令牌，直接调用admin.auth().setCustomUserClaims(userId, additionalClaims)设置自定义声明即可，前端可以通过getIdTokenResult()直接读取到声明中的role字段，无需额外处理登录态。

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