fix: global dispatcher

Author: akitaSummer

src/HttpClient.ts

@@ -22,8 +22,10 @@ import {
   request as undiciRequest,
   Dispatcher,
   Agent,
-  getGlobalDispatcher,
   Pool,
+  getGlobalDispatcher,
+  MockAgent,
+  setGlobalDispatcher,
 } from 'undici';
 import undiciSymbols from 'undici/lib/core/symbols.js';
 import { FormData as FormDataNode } from 'formdata-node';
@@ -207,7 +209,28 @@ export class HttpClient extends EventEmitter {
   }
 
   getDispatcher() {
-    return this.#dispatcher ?? getGlobalDispatcher();
+    if (this.#dispatcher) {
+      return this.#dispatcher;
+    }
+    return this.safeGetGlobalDispatcher();
+  }
+
+  safeGetGlobalDispatcher() {
+    // In a multi-version undici environment
+    // the global dispatcher is the highest version of undici
+    // which will conflict with the maxRedirects field and report an error
+    // so we need to create it that use 5.x version
+    const globalDispatcher = getGlobalDispatcher();
+    // eslint-disable-next-line @typescript-eslint/ban-ts-comment
+    // @ts-ignore
+    // eslint-disable-next-line no-proto
+    const proto = globalDispatcher.__proto__;
+    if (proto !== Agent.prototype && proto !== MockAgent.prototype) {
+      const dispatcher = proto.constructor.name === 'MockAgent' ? new MockAgent() : new Agent();
+      setGlobalDispatcher(dispatcher);
+      return dispatcher;
+    }
+    return globalDispatcher;
   }
 
   setDispatcher(dispatcher: Dispatcher) {
@@ -415,7 +438,7 @@ export class HttpClient extends EventEmitter {
         headers,
         bodyTimeout,
         opaque: internalOpaque,
-        dispatcher: args.dispatcher ?? this.#dispatcher,
+        dispatcher: args.dispatcher ?? this.getDispatcher(),
         signal: args.signal,
       };
       if (typeof args.highWaterMark === 'number') {

test/diagnostics_channel.test.ts

@@ -1,8 +1,9 @@
 import { strict as assert } from 'node:assert';
 import diagnosticsChannel from 'node:diagnostics_channel';
 import { describe, it, beforeEach, afterEach } from 'vitest';
-import urllib from '../src';
-import type {
+import urllib, { getGlobalDispatcher, setGlobalDispatcher } from '../src';
+import {
+  MockAgent,
   RequestDiagnosticsMessage,
   ResponseDiagnosticsMessage,
 } from '../src';
@@ -13,14 +14,22 @@ import { sleep } from './utils';
 describe('diagnostics_channel.test.ts', () => {
   let close: any;
   let _url: string;
+
+  let mockAgent: MockAgent;
+  const globalAgent = getGlobalDispatcher();
+
   beforeEach(async () => {
     const { closeServer, url } = await startServer();
     close = closeServer;
     _url = url;
+    mockAgent = new MockAgent();
+    setGlobalDispatcher(mockAgent);
   });
 
   afterEach(async () => {
     await close();
+    setGlobalDispatcher(globalAgent);
+    await mockAgent.close();
   });
 
   it('should support trace socket info by undici:client:sendHeaders and undici:request:trailers', async () => {

test/options.retry.test.ts

@@ -1,7 +1,7 @@
 import { strict as assert } from 'node:assert';
 import { createWriteStream, createReadStream } from 'node:fs';
 import { describe, it, beforeAll, afterAll, beforeEach, afterEach } from 'vitest';
-import urllib from '../src';
+import urllib, { getGlobalDispatcher, MockAgent, setGlobalDispatcher } from '../src';
 import { startServer } from './fixtures/server';
 import { readableToString, createTempfile } from './utils';
 
@@ -11,6 +11,9 @@ describe('options.retry.test.ts', () => {
   let tmpfile: string;
   let cleanup: any;
 
+  let mockAgent: MockAgent;
+  const globalAgent = getGlobalDispatcher();
+
   beforeAll(async () => {
     const { closeServer, url } = await startServer();
     close = closeServer;
@@ -24,9 +27,13 @@ describe('options.retry.test.ts', () => {
     const item = await createTempfile();
     tmpfile = item.tmpfile;
     cleanup = item.cleanup;
+    mockAgent = new MockAgent();
+    setGlobalDispatcher(mockAgent);
   });
   afterEach(async () => {
     await cleanup();
+    setGlobalDispatcher(globalAgent);
+    await mockAgent.close();
   });
 
   it('should not retry on 400', async () => {

test/options.timing.test.ts

@@ -1,21 +1,28 @@
 import { strict as assert } from 'node:assert';
 import { describe, it, beforeAll, afterAll } from 'vitest';
-import urllib from '../src';
+import urllib, { getGlobalDispatcher, MockAgent, setGlobalDispatcher } from '../src';
 import { RawResponseWithMeta } from '../src/Response';
 import { startServer } from './fixtures/server';
 import { sleep } from './utils';
 
 describe('options.timing.test.ts', () => {
   let close: any;
   let _url: string;
+
+  let mockAgent: MockAgent;
+  const globalAgent = getGlobalDispatcher();
   beforeAll(async () => {
     const { closeServer, url } = await startServer();
     close = closeServer;
     _url = url;
+    mockAgent = new MockAgent();
+    setGlobalDispatcher(mockAgent);
   });
 
   afterAll(async () => {
     await close();
+    setGlobalDispatcher(globalAgent);
+    await mockAgent.close();
   });
 
   it('should timing = true work', async () => {