lib: initialize suppression storage eagerly

Author: DivyanshuX9

lib/diagnostics_channel.js

@@ -35,20 +35,10 @@ const { subscribers: subscriberCounts } = dc_binding;
 
 const { WeakReference } = require('internal/util');
 const { isPromise } = require('internal/util/types');
+const { AsyncLocalStorage } = require('async_hooks');
 
-let suppressionStorage;
+const suppressionStorage = new AsyncLocalStorage();
 
-function getSuppressionsStorage() {
-  if (suppressionStorage === undefined) {
-    const { AsyncLocalStorage } = require('async_hooks');
-    suppressionStorage = new AsyncLocalStorage();
-  }
-  return suppressionStorage;
-}
-
-function withSuppressionsContext(set, fn, thisArg, args) {
-  return getSuppressionsStorage().run(set, fn.bind(thisArg), ...args);
-}
 // Can't delete when weakref count reaches 0 as it could increment again.
 // Only GC can be used as a valid time to clean up the channels map.
 class WeakRefMap extends SafeMap {
@@ -106,7 +96,7 @@ class RunStoresScope {
 
     // Enter stores using withScope
     if (activeChannel._stores) {
-      const activeKeys = getSuppressionsStorage().getStore();
+      const activeKeys = suppressionStorage.getStore();
       for (const entry of activeChannel._stores.entries()) {
         const store = entry[0];
         const { transform, subscriberId = null } = entry[1];
@@ -224,7 +214,7 @@ class ActiveChannel {
 
   publish(data) {
     const subscribers = this._subscribers;
-    const activeKeys = getSuppressionsStorage().getStore();
+    const activeKeys = suppressionStorage.getStore();
     for (let i = 0; i < (subscribers?.length || 0); i++) {
       try {
         const { handler, subscriberId = null } = subscribers[i];
@@ -692,10 +682,10 @@ function suppressed(key, fn, thisArg, ...args) {
     throw new ERR_INVALID_ARG_TYPE('key', ['object', 'symbol', 'function'], key);
   }
 
-  const currentSet = getSuppressionsStorage().getStore();
+  const currentSet = suppressionStorage.getStore();
   const next = currentSet ? new SafeSet(currentSet) : new SafeSet();
   next.add(key);
-  return withSuppressionsContext(next, fn, thisArg, args);
+  return suppressionStorage.run(next, () => ReflectApply(fn, thisArg, args));
 }
 
 module.exports = {