如何在Node.js第三方SDK中实现过期Auth Token自动重试

场景背景 #

某Node.js应用使用第三方SDK调用需授权的API，授权方式为请求携带access token；当token过期时，API返回401响应。该SDK自带获取、设置token的方法，但未实现响应错误处理逻辑，导致每次都需要手动检查请求是否返回401、手动刷新token后重试，完全违反DRY原则。

以下是Node.js开发者针对这类场景的常见解决方案：

1. 用高阶函数包装SDK API方法 #

这是最灵活且无需修改SDK源码的方案，通过通用包装函数统一处理401重试逻辑：

// 全局锁，避免并发请求重复刷新token
let isRefreshing = false;
let retryQueue = [];

async function withAutoRetry(sdkMethod) {
  // 如果正在刷新token，将请求加入队列等待
  if (isRefreshing) {
    return new Promise(resolve => {
      retryQueue.push(() => resolve(sdkMethod()));
    });
  }

  try {
    return await sdkMethod();
  } catch (error) {
    // 根据SDK返回的错误结构判断是否为401
    if (error.response?.status === 401) {
      isRefreshing = true;
      try {
        // 调用SDK自带的刷新token方法
        const newToken = await sdk.refreshAccessToken();
        // 将新token设置到SDK
        sdk.setAccessToken(newToken);
        // 执行队列中等待的所有请求
        retryQueue.forEach(cb => cb());
        retryQueue = [];
        // 重试当前请求
        return await sdkMethod();
      } finally {
        isRefreshing = false;
      }
    }
    // 非401错误直接抛出
    throw error;
  }
}

// 使用方式：将SDK方法传入包装函数
const apiResult = await withAutoRetry(() => sdk.getResource({ id: 123 }));

2. 利用SDK的响应拦截器（若支持） #

部分SDK会提供类似axios的请求/响应拦截器钩子，可直接在拦截器中处理401逻辑：

// 假设SDK提供interceptors.response.use接口
sdk.interceptors.response.use(
  response => response,
  async error => {
    const originalRequest = error.config;
    // 避免无限重试，标记已重试的请求
    if (error.response.status === 401 && !originalRequest._retry) {
      originalRequest._retry = true;
      // 刷新并设置新token
      const newToken = await sdk.refreshAccessToken();
      sdk.setAccessToken(newToken);
      // 更新请求头中的token（若需要手动设置）
      originalRequest.headers['Authorization'] = `Bearer ${newToken}`;
      // 重试原请求
      return sdk.request(originalRequest);
    }
    return Promise.reject(error);
  }
);

3. 子类扩展的改进思路 #

如果坚持用子类扩展，不要逐个重写API方法，而是重写SDK内部的核心请求方法（比如sendRequest、makeRequest这类统一处理请求的方法）：

class EnhancedSDK extends ThirdPartySDK {
  // 重写SDK的核心请求方法（需确认该方法可被重写）
  async sendRequest(requestConfig) {
    try {
      return await super.sendRequest(requestConfig);
    } catch (error) {
      if (error.response?.status === 401) {
        // 刷新token
        const newToken = await this.refreshAccessToken();
        this.setAccessToken(newToken);
        // 更新请求配置中的token
        requestConfig.headers['Authorization'] = `Bearer ${newToken}`;
        // 重试请求
        return await super.sendRequest(requestConfig);
      }
      throw error;
    }
  }
}

// 使用增强后的SDK实例
const sdk = new EnhancedSDK();
const result = await sdk.getResource({ id: 123 });

注意：该方案依赖SDK内部核心方法的可访问性，如果SDK将核心请求方法私有化，则无法使用此方式。

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