axios拦截器底层实现原理

Axios 的拦截器通过内部的Promise 链实现了对请求和响应的拦截与修改。了解其底层原理需要深入到 Axios 源码中,特别是其请求发起和响应处理的逻辑。

Axios 拦截器实现流程

  1. 拦截器队列

    Axios 在内部维护了两个拦截器队列:requestresponse。当开发者通过 interceptors.request.useinterceptors.response.use 注册拦截器时,拦截器会被存入队列。

  2. Promise 链

    每次调用 Axios 实例发起请求时,会依次将拦截器队列中的函数(请求/响应拦截器)和实际的请求函数组成一个 Promise 链。

  3. 执行顺序

    • 请求拦截器按注册的顺序执行。
    • 响应拦截器按注册的顺序逆序执行。

核心源码分析

以下是 Axios 源码的关键部分,用于理解拦截器的实现。

1. 拦截器管理

Axios 内部的 InterceptorManager 类用于管理拦截器:

javascript 复制代码
function InterceptorManager() {
  this.handlers = [];
}

// 注册拦截器
InterceptorManager.prototype.use = function use(fulfilled, rejected) {
  this.handlers.push({
    fulfilled: fulfilled, // 成功处理函数
    rejected: rejected,   // 失败处理函数
  });
  return this.handlers.length - 1; // 返回拦截器的索引
};

// 移除拦截器
InterceptorManager.prototype.eject = function eject(id) {
  if (this.handlers[id]) {
    this.handlers[id] = null; // 将对应的拦截器置为 null
  }
};

InterceptorManager 的核心功能是管理拦截器的添加和删除,所有的拦截器函数存储在 handlers 数组中。


2. 请求的 Promise 链构建

当调用 axios.request(config) 时,会构建一个基于拦截器的 Promise 链:

javascript 复制代码
Axios.prototype.request = function request(config) {
  // 初始化配置
  config = config || {};

  // 将实际请求函数添加到链末尾
  let promise = Promise.resolve(config);

  // 构建拦截器链
  const chain = [];

  // 添加请求拦截器(顺序)
  this.interceptors.request.handlers.forEach((interceptor) => {
    if (interceptor !== null) {
      chain.push(interceptor.fulfilled, interceptor.rejected);
    }
  });

  // 添加实际的请求函数
  chain.push(dispatchRequest, undefined);

  // 添加响应拦截器(逆序)
  this.interceptors.response.handlers.forEach((interceptor) => {
    if (interceptor !== null) {
      chain.push(interceptor.fulfilled, interceptor.rejected);
    }
  });

  // 执行链条中的每个函数
  while (chain.length) {
    promise = promise.then(chain.shift(), chain.shift());
  }

  return promise;
};

在上面的实现中:

  • dispatchRequest 是 Axios 实际发起网络请求的函数。
  • 请求拦截器按注册的顺序加入链。
  • 响应拦截器按逆序加入链。
  • 最终,Promise 链依次执行每个拦截器和请求逻辑。

3. 实际请求的实现

dispatchRequest 负责发起 HTTP 请求并返回响应:

javascript 复制代码
function dispatchRequest(config) {
  // 检查是否已取消请求
  if (config.cancelToken) {
    config.cancelToken.throwIfRequested();
  }

  // 发起请求
  return xhrAdapter(config).then(
    (response) => {
      return response;
    },
    (error) => {
      return Promise.reject(error);
    }
  );
}

xhrAdapter 是默认的适配器,用于在浏览器环境中通过 XMLHttpRequestfetch 发起请求。


工作流程总结

  1. 注册拦截器

    开发者通过 interceptors.request.useinterceptors.response.use 向 Axios 的拦截器队列中添加拦截函数。

  2. 发起请求

    调用 axios(config) 时,Axios 会根据请求拦截器、实际请求函数和响应拦截器构建一个 Promise 链。

  3. 执行链条

    按以下顺序依次执行:

    • 请求拦截器:按注册顺序执行。
    • 实际的请求函数:发送 HTTP 请求。
    • 响应拦截器:按注册顺序逆序执行。
  4. 返回结果

    Promise 链最终返回请求的结果或抛出错误,供调用者处理。


图示说明

plaintext 复制代码
请求拦截器1 --> 请求拦截器2 --> dispatchRequest --> 响应拦截器2 --> 响应拦截器1
  • 请求拦截器按顺序执行请求拦截器1 → 请求拦截器2
  • 响应拦截器按逆序执行响应拦截器2 → 响应拦截器1

Axios 拦截器的特点

  1. 链式结构 :通过 Promise 链灵活地插入和执行拦截器。
  2. 易扩展 :通过 InterceptorManager 动态注册、移除拦截器。
  3. 一致性:无论请求还是响应,拦截器均可统一添加逻辑。

通过上述机制,Axios 实现了强大且灵活的拦截器功能,开发者可以轻松扩展请求和响应的处理逻辑。

相关推荐
WeiXiao_Hyy24 分钟前
成为 Top 1% 的工程师
java·开发语言·javascript·经验分享·后端
吃杠碰小鸡41 分钟前
高中数学-数列-导数证明
前端·数学·算法
kingwebo'sZone1 小时前
C#使用Aspose.Words把 word转成图片
前端·c#·word
xjt_09011 小时前
基于 Vue 3 构建企业级 Web Components 组件库
前端·javascript·vue.js
我是伪码农1 小时前
Vue 2.3
前端·javascript·vue.js
夜郎king2 小时前
HTML5 SVG 实现日出日落动画与实时天气可视化
前端·html5·svg 日出日落
辰风沐阳2 小时前
JavaScript 的宏任务和微任务
javascript
夏幻灵3 小时前
HTML5里最常用的十大标签
前端·html·html5
冰暮流星3 小时前
javascript之二重循环练习
开发语言·javascript·数据库
Mr Xu_3 小时前
Vue 3 中 watch 的使用详解:监听响应式数据变化的利器
前端·javascript·vue.js