Undici 请求延迟测量的技术实现方案

在 Node.js 生态系统中，Undici 作为一款高效的 HTTP/1.1 客户端库，被广泛应用于各种网络请求场景。本文将深入探讨如何在 Undici 中准确测量请求执行时间的技术实现方案。

测量请求延迟的挑战

在实际开发中，我们经常需要测量完整的 HTTP 请求生命周期时间，包括连接建立、请求体写入、等待响应、读取响应头和读取响应体等阶段。在 Undici 中实现这一目标面临几个技术挑战：

1. 需要区分全局测量和局部测量
2. 需要处理请求重试的情况
3. 需要准确捕获各个阶段的耗时
4. 需要关联请求开始和结束的事件

三种测量方案对比

1. 诊断通道方案

诊断通道(Diagnostic Channel)可以捕获所有 Undici 请求，但这种方案过于全局化，无法针对特定请求进行测量，且会捕获应用中所有第三方库的请求，不适合精细化的性能监控需求。

2. 拦截器方案

拦截器(Interceptor)方案通过在请求前后记录时间戳来测量耗时：

function createMetricsInterceptor() {
  return async (dispatch) => {
    return (opts, handler) => {
      const start = performance.now();
      const res = await dispatch(opts, handler);
      const totalTime = performance.now() - start;
      return res;
    };
  };
}

这种方案的局限性在于：

• 包含重试时间
• 不包含响应体读取时间
• 无法细分请求各阶段耗时

3. 请求处理器方案

请求处理器(Request Handler)方案是最为推荐的实现方式，它利用 Undici 的生命周期钩子函数来精确测量各阶段耗时：

class RequestMetricsHandler {
  constructor() {
    this.startTime = 0;
    this.endTime = 0;
    this.errored = false;
  }
  
  onConnect() {
    this.startTime = Date.now();
  }
  
  onHeaders(status, headers) {
    const timestamp = Date.now();
    console.log('响应头耗时', timestamp - this.startTime, 'ms');
  }
  
  onComplete() {
    this.endTime = Date.now();
    console.log('请求总耗时', this.endTime - this.startTime, 'ms');
  }
  
  onError(err) {
    this.endTime = Date.now();
    console.log('请求出错总耗时', this.endTime - this.startTime, 'ms');
  }
}

技术实现要点

1. 请求处理器生命周期：每个请求处理器实例与单个请求绑定，其内部状态变量自然关联到该请求的生命周期。

2. 精确阶段测量：

   • onConnect：记录连接建立开始时间
   • onHeaders：测量到收到响应头的时间
   • onComplete/onError：记录请求最终完成时间

3. 状态管理：通过类实例变量保存测量相关的状态，确保在请求的各个阶段都能访问相同的上下文。

最佳实践建议

1. 对于需要细粒度测量的场景，推荐使用请求处理器方案
2. 考虑将测量结果输出到监控系统而非仅打印日志
3. 可以扩展处理器类，加入更多性能指标如DNS查询时间、TLS握手时间等
4. 在生产环境中，考虑采样测量而非全量记录以减少性能开销

通过合理利用 Undici 提供的请求处理器机制，开发者可以实现精确到各个阶段的请求性能监控，为系统优化提供可靠的数据支持。