stream: preserve AsyncLocalStorage on finished only when needed

Author: avcribl

lib/internal/streams/end-of-stream.js

@@ -44,6 +44,9 @@ const {
   kIsClosedPromise,
 } = require('internal/streams/utils');
 
+const { getHookArrays } = require('internal/async_hooks');
+const AsyncContextFrame = require('internal/async_context_frame');
+
 // Lazy load
 let AsyncLocalStorage;
 let addAbortListener;
@@ -66,8 +69,13 @@ function eos(stream, options, callback) {
   validateFunction(callback, 'callback');
   validateAbortSignal(options.signal, 'options.signal');
 
-  AsyncLocalStorage ??= require('async_hooks').AsyncLocalStorage;
-  callback = once(AsyncLocalStorage.bind(callback));
+  if (AsyncContextFrame.current() ||
+      getHookArrays()[0].length > 0) {
+    AsyncLocalStorage ??= require('async_hooks').AsyncLocalStorage;
+    callback = once(AsyncLocalStorage.bind(callback));
+  } else {
+    callback = once(callback);
+  }
 
   if (isReadableStream(stream) || isWritableStream(stream)) {
     return eosWeb(stream, options, callback);