refactor: defaultLookup

nxtedition · Aug 10, 2024 · 866794e · 866794e
1 parent 7ac47fd
commit 866794e
Showing 1 changed file with 5 additions and 9 deletions.
diff --git a/lib/index.js b/lib/index.js
@@ -20,6 +20,10 @@ export const interceptors = {
 export { parseHeaders } from './utils.js'
 export { Client, Pool, Agent, getGlobalDispatcher, setGlobalDispatcher } from 'undici'
 
+function defaultLookup(origin, opts, callback) {
+  callback(null, Array.isArray(origin) ? origin[Math.floor(Math.random() * origin.length)] : origin)
+}
+
 function wrapDispatcher(dispatcher) {
   let wrappedDispatcher = dispatcherCache.get(dispatcher)
   if (wrappedDispatcher == null) {
@@ -70,15 +74,7 @@ function wrapDispatcher(dispatcher) {
             logger: opts.logger ?? null,
             dns: opts.dns ?? true,
             connect: opts.connect,
-            lookup:
-              opts.lookup ??
-              ((origin, opts, callback) =>
-                callback(
-                  null,
-                  Array.isArray(origin)
-                    ? origin[Math.floor(Math.random() * origin.length)]
-                    : origin,
-                )),
+            lookup: opts.lookup ?? defaultLookup,
             maxRedirections: 0, // TODO (fix): Ugly hack to disable undici redirections.
           },
           handler,